mirror of
https://github.com/waku-org/nwaku.git
synced 2025-01-24 05:39:21 +00:00
37 lines
938 B
Nim
37 lines
938 B
Nim
|
import json, macros
|
||
|
|
import chronos, router, chronicles
|
||
|
|
import jsonmarshal
|
||
|
|
|
||
|
|
export chronos, json, jsonmarshal, router, chronicles
|
||
|
|
|
||
|
|
type
|
||
|
|
RpcServer* = ref object of RootRef
|
||
|
|
router*: RpcRouter
|
||
|
|
|
||
|
|
proc newRpcServer*(): RpcServer =
|
||
|
|
new result
|
||
|
|
result.router = newRpcRouter()
|
||
|
|
|
||
|
|
template rpc*(server: RpcServer, path: string, body: untyped): untyped =
|
||
|
|
server.router.rpc(path, body)
|
||
|
|
|
||
|
|
template hasMethod*(server: RpcServer, methodName: string): bool =
|
||
|
|
server.router.hasMethod(methodName)
|
||
|
|
|
||
|
|
# Wrapper for message processing
|
||
|
|
|
||
|
|
proc route*(server: RpcServer, line: string): Future[string] {.async, gcsafe.} =
|
||
|
|
result = await server.router.route(line)
|
||
|
|
|
||
|
|
# Server registration
|
||
|
|
|
||
|
|
proc register*(server: RpcServer, name: string, rpc: RpcProc) =
|
||
|
|
## Add a name/code pair to the RPC server.
|
||
|
|
server.router.register(name, rpc)
|
||
|
|
|
||
|
|
proc unRegisterAll*(server: RpcServer) =
|
||
|
|
# Remove all remote procedure calls from this server.
|
||
|
|
server.router.clear
|
||
|
|
|
||
|
|
|