codex-storage
/
nim-json-rpc
mirror of https://github.com/codex-storage/nim-json-rpc.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add rng param to rpcWebsocketServer constructor

This commit is contained in:
jangko 2022-07-27 09:46:36 +07:00
parent 0ef05ae7f8
commit 5ccdaed0ad
No known key found for this signature in database
GPG Key ID: 31702AE10541E6B9
1 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
json_rpc/servers/websocketserver.nim
View File

@ -77,25 +77,27 @@ proc handleRequest(rpc: RpcWebSocketServer, request: HttpRequest) {.async.} =
error "WebSocket error:", exception = exc.msg
proc initWebsocket(rpc: RpcWebSocketServer, compression: bool,
authHooks: seq[WsAuthHook]) =
authHooks: seq[WsAuthHook],
rng: Rng) =
if compression:
let deflateFactory = deflateFactory()
rpc.wsserver = WSServer.new(factories = [deflateFactory])
rpc.wsserver = WSServer.new(factories = [deflateFactory], rng = rng)
else:
rpc.wsserver = WSServer.new()
rpc.wsserver = WSServer.new(rng = rng)
rpc.authHooks = authHooks
proc newRpcWebSocketServer*(
address: TransportAddress,
compression: bool = false,
flags: set[ServerFlags] = {ServerFlags.TcpNoDelay,ServerFlags.ReuseAddr},
authHooks: seq[WsAuthHook] = @[]): RpcWebSocketServer =
authHooks: seq[WsAuthHook] = @[],
rng = newRng()): RpcWebSocketServer =
var server = new(RpcWebSocketServer)
proc processCallback(request: HttpRequest): Future[void] =
handleRequest(server, request)
server.initWebsocket(compression, authHooks)
server.initWebsocket(compression, authHooks, rng)
server.server = HttpServer.create(
address,
processCallback,
@ -109,13 +111,15 @@ proc newRpcWebSocketServer*(
port: Port,
compression: bool = false,
flags: set[ServerFlags] = {ServerFlags.TcpNoDelay, ServerFlags.ReuseAddr},
authHooks: seq[WsAuthHook] = @[]): RpcWebSocketServer =
authHooks: seq[WsAuthHook] = @[],
rng = newRng()): RpcWebSocketServer =
newRpcWebSocketServer(
initTAddress(host, port),
compression,
flags,
authHooks
authHooks,
rng
)
proc newRpcWebSocketServer*(
@ -128,13 +132,14 @@ proc newRpcWebSocketServer*(
tlsFlags: set[TLSFlags] = {},
tlsMinVersion = TLSVersion.TLS12,
tlsMaxVersion = TLSVersion.TLS12,
authHooks: seq[WsAuthHook] = @[]): RpcWebSocketServer =
authHooks: seq[WsAuthHook] = @[],
rng = newRng()): RpcWebSocketServer =
var server = new(RpcWebSocketServer)
proc processCallback(request: HttpRequest): Future[void] =
handleRequest(server, request)
server.initWebsocket(compression, authHooks)
server.initWebsocket(compression, authHooks, rng)
server.server = TlsHttpServer.create(
address,
tlsPrivateKey,
@ -159,7 +164,8 @@ proc newRpcWebSocketServer*(
tlsFlags: set[TLSFlags] = {},
tlsMinVersion = TLSVersion.TLS12,
tlsMaxVersion = TLSVersion.TLS12,
authHooks: seq[WsAuthHook] = @[]): RpcWebSocketServer =
authHooks: seq[WsAuthHook] = @[],
rng = newRng()): RpcWebSocketServer =
newRpcWebSocketServer(
initTAddress(host, port),
@ -170,7 +176,8 @@ proc newRpcWebSocketServer*(
tlsFlags,
tlsMinVersion,
tlsMaxVersion,
authHooks
authHooks,
rng
)
proc start*(server: RpcWebSocketServer) =