status-im
/
nim-eth
mirror of https://github.com/status-im/nim-eth.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

[P2P DSL] Add a separate genUserHandlerCall helper

This commit is contained in:
Zahary Karadjov 2019-12-09 16:18:22 +02:00
parent 4c6878343c
commit 3f721e7ef6
No known key found for this signature in database
GPG Key ID: C8936F8A3073D609
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
eth/p2p/p2p_protocol_dsl.nim
View File

@ -633,21 +633,23 @@ proc defineThunk*(msg: Message, thunk: NimNode) =
protocol.outRecvProcs.add thunk protocol.outRecvProcs.add thunk
proc genAwaitUserHandler*(msg: Message, receivedMsg: NimNode, proc genUserHandlerCall*(msg: Message, receivedMsg: NimNode,
leadingParams: varargs[NimNode]): NimNode = leadingParams: varargs[NimNode]): NimNode =
if msg.userHandler == nil: if msg.userHandler == nil:
return newStmtList() return newStmtList()
var userHandlerCall = newCall(msg.userHandler.name, leadingParams) result = newCall(msg.userHandler.name, leadingParams)
var params = toSeq(msg.procDef.typedParams(skip = 1)) var params = toSeq(msg.procDef.typedParams(skip = 1))
if params.len == 1 and msg.protocol.useSingleRecordInlining: if params.len == 1 and msg.protocol.useSingleRecordInlining:
userHandlerCall.add receivedMsg result.add receivedMsg
else: else:
for p in params: for p in params:
userHandlerCall.add newDotExpr(receivedMsg, p[0]) result.add newDotExpr(receivedMsg, p[0])
return newCall("await", userHandlerCall) proc genAwaitUserHandler*(msg: Message, receivedMsg: NimNode,
leadingParams: varargs[NimNode]): NimNode =
return newCall("await", msg.genUserHandlerCall(receivedMsg, leadingParams))
proc appendAllParams*(node: NimNode, procDef: NimNode, skipFirst = 0): NimNode = proc appendAllParams*(node: NimNode, procDef: NimNode, skipFirst = 0): NimNode =
result = node result = node