mirror of
https://github.com/logos-storage/nim-json-rpc.git
synced 2026-05-28 05:19:26 +00:00
Add chronicles logging and some custom errors
This commit is contained in:
parent
4daf3353aa
commit
f9d20db511
@ -1,4 +1,4 @@
|
|||||||
import json, tables, strutils, options, macros
|
import json, tables, strutils, options, macros, chronicles
|
||||||
import asyncdispatch2
|
import asyncdispatch2
|
||||||
import jsonmarshal
|
import jsonmarshal
|
||||||
|
|
||||||
@ -20,6 +20,9 @@ type
|
|||||||
code*: int
|
code*: int
|
||||||
data*: JsonNode
|
data*: JsonNode
|
||||||
|
|
||||||
|
RpcBindError* = object of Exception
|
||||||
|
RpcAddressUnresolvableError* = object of Exception
|
||||||
|
|
||||||
const
|
const
|
||||||
JSON_PARSE_ERROR* = -32700
|
JSON_PARSE_ERROR* = -32700
|
||||||
INVALID_REQUEST* = -32600
|
INVALID_REQUEST* = -32600
|
||||||
@ -36,12 +39,9 @@ const
|
|||||||
(INVALID_REQUEST, "No id specified")
|
(INVALID_REQUEST, "No id specified")
|
||||||
]
|
]
|
||||||
|
|
||||||
when not defined(release):
|
# Utility functions
|
||||||
template ifDebug*(actions: untyped): untyped =
|
# TODO: Move outside server
|
||||||
actions
|
func `%`*(p: Port): JsonNode = %(p.int)
|
||||||
else:
|
|
||||||
template ifDebug*(actions: untyped): untyped =
|
|
||||||
discard
|
|
||||||
|
|
||||||
# Json state checking
|
# Json state checking
|
||||||
|
|
||||||
@ -53,6 +53,7 @@ template jsonValid*(jsonString: string, node: var JsonNode): (bool, string) =
|
|||||||
except:
|
except:
|
||||||
valid = false
|
valid = false
|
||||||
msg = getCurrentExceptionMsg()
|
msg = getCurrentExceptionMsg()
|
||||||
|
debug "Cannot process json", json = jsonString, msg = msg
|
||||||
(valid, msg)
|
(valid, msg)
|
||||||
|
|
||||||
proc checkJsonErrors*(line: string,
|
proc checkJsonErrors*(line: string,
|
||||||
@ -80,7 +81,7 @@ proc sendError*(client: StreamTransport, code: int, msg: string, id: JsonNode,
|
|||||||
data: JsonNode = newJNull()) {.async.} =
|
data: JsonNode = newJNull()) {.async.} =
|
||||||
## Send error message to client
|
## Send error message to client
|
||||||
let error = %{"code": %(code), "message": %msg, "data": data}
|
let error = %{"code": %(code), "message": %msg, "data": data}
|
||||||
ifDebug: echo "Send error json: ", wrapReply(newJNull(), error, id)
|
debug "Error generated", error = error, id = id
|
||||||
result = client.write(wrapReply(id, newJNull(), error))
|
result = client.write(wrapReply(id, newJNull(), error))
|
||||||
|
|
||||||
proc sendJsonError*(state: RpcJsonError, client: StreamTransport, id: JsonNode,
|
proc sendJsonError*(state: RpcJsonError, client: StreamTransport, id: JsonNode,
|
||||||
@ -96,6 +97,9 @@ proc processMessage(server: RpcServer, client: StreamTransport,
|
|||||||
node: JsonNode
|
node: JsonNode
|
||||||
# set up node and/or flag errors
|
# set up node and/or flag errors
|
||||||
jsonErrorState = checkJsonErrors(line, node)
|
jsonErrorState = checkJsonErrors(line, node)
|
||||||
|
|
||||||
|
debug "Received line", line = line
|
||||||
|
|
||||||
if jsonErrorState.isSome:
|
if jsonErrorState.isSome:
|
||||||
let errState = jsonErrorState.get
|
let errState = jsonErrorState.get
|
||||||
var id = if errState.err == rjeInvalidJson: newJNull() else: node["id"]
|
var id = if errState.err == rjeInvalidJson: newJNull() else: node["id"]
|
||||||
@ -107,12 +111,12 @@ proc processMessage(server: RpcServer, client: StreamTransport,
|
|||||||
|
|
||||||
if not server.procs.hasKey(methodName):
|
if not server.procs.hasKey(methodName):
|
||||||
await client.sendError(METHOD_NOT_FOUND, "Method not found", id,
|
await client.sendError(METHOD_NOT_FOUND, "Method not found", id,
|
||||||
%(methodName & " is not a registered method."))
|
%(methodName & " is not a registered method."))
|
||||||
else:
|
else:
|
||||||
let callRes = await server.procs[methodName](node["params"])
|
let callRes = await server.procs[methodName](node["params"])
|
||||||
discard await client.write(wrapReply(id, callRes, newJNull()))
|
discard await client.write(wrapReply(id, callRes, newJNull()))
|
||||||
|
|
||||||
proc processClient(server: StreamServer, client: StreamTransport) {.async.} =
|
proc processClient(server: StreamServer, client: StreamTransport) {.async, gcsafe.} =
|
||||||
var rpc = getUserData[RpcServer](server)
|
var rpc = getUserData[RpcServer](server)
|
||||||
while true:
|
while true:
|
||||||
## TODO: We need to put limit here, or server could be easily put out of
|
## TODO: We need to put limit here, or server could be easily put out of
|
||||||
@ -122,7 +126,7 @@ proc processClient(server: StreamServer, client: StreamTransport) {.async.} =
|
|||||||
client.close()
|
client.close()
|
||||||
break
|
break
|
||||||
|
|
||||||
ifDebug: echo "Process client: ", client.remoteAddress()
|
debug "Processing client", addresss = client.remoteAddress()
|
||||||
|
|
||||||
let future = processMessage(rpc, client, line)
|
let future = processMessage(rpc, client, line)
|
||||||
yield future
|
yield future
|
||||||
@ -145,17 +149,17 @@ proc newRpcServer*(addresses: openarray[TransportAddress]): RpcServer =
|
|||||||
|
|
||||||
for item in addresses:
|
for item in addresses:
|
||||||
try:
|
try:
|
||||||
ifDebug: echo "Create server on " & $item
|
info "Creating server on ", address = $item
|
||||||
var server = createStreamServer(item, processClient, {ReuseAddr},
|
var server = createStreamServer(item, processClient, {ReuseAddr},
|
||||||
udata = result)
|
udata = result)
|
||||||
result.servers.add(server)
|
result.servers.add(server)
|
||||||
except:
|
except:
|
||||||
ifDebug: echo "Failed to create server on " & $item
|
error "Failed to create server", address = $item, message = getCurrentExceptionMsg()
|
||||||
|
|
||||||
if len(result.servers) == 0:
|
if len(result.servers) == 0:
|
||||||
# Server was not bound, critical error.
|
# Server was not bound, critical error.
|
||||||
# TODO: Custom RpcException error
|
# TODO: Custom RpcException error
|
||||||
raise newException(ValueError, "Unable to create server!")
|
raise newException(RpcBindError, "Unable to create server!")
|
||||||
|
|
||||||
proc newRpcServer*(addresses: openarray[string]): RpcServer =
|
proc newRpcServer*(addresses: openarray[string]): RpcServer =
|
||||||
## Create new server and assign it to addresses ``addresses``.
|
## Create new server and assign it to addresses ``addresses``.
|
||||||
@ -184,7 +188,7 @@ proc newRpcServer*(addresses: openarray[string]): RpcServer =
|
|||||||
|
|
||||||
if len(baddrs) == 0:
|
if len(baddrs) == 0:
|
||||||
# Addresses could not be resolved, critical error.
|
# Addresses could not be resolved, critical error.
|
||||||
raise newException(ValueError, "Unable to get address!")
|
raise newException(RpcAddressUnresolvableError, "Unable to get address!")
|
||||||
|
|
||||||
result = newRpcServer(baddrs)
|
result = newRpcServer(baddrs)
|
||||||
|
|
||||||
@ -207,8 +211,7 @@ proc newRpcServer*(address = "localhost", port: Port = Port(8545)): RpcServer =
|
|||||||
|
|
||||||
if len(tas4) == 0 and len(tas6) == 0:
|
if len(tas4) == 0 and len(tas6) == 0:
|
||||||
# Address was not resolved, critical error.
|
# Address was not resolved, critical error.
|
||||||
# TODO: Custom RpcException error.
|
raise newException(RpcAddressUnresolvableError,
|
||||||
raise newException(ValueError,
|
|
||||||
"Address " & address & " could not be resolved!")
|
"Address " & address & " could not be resolved!")
|
||||||
|
|
||||||
result = RpcServer()
|
result = RpcServer()
|
||||||
@ -216,26 +219,25 @@ proc newRpcServer*(address = "localhost", port: Port = Port(8545)): RpcServer =
|
|||||||
result.servers = newSeq[StreamServer]()
|
result.servers = newSeq[StreamServer]()
|
||||||
for item in tas4:
|
for item in tas4:
|
||||||
try:
|
try:
|
||||||
ifDebug: echo "Create server on " & $item
|
info "Creating server for address", ip4address = $item
|
||||||
var server = createStreamServer(item, processClient, {ReuseAddr},
|
var server = createStreamServer(item, processClient, {ReuseAddr},
|
||||||
udata = result)
|
udata = result)
|
||||||
result.servers.add(server)
|
result.servers.add(server)
|
||||||
except:
|
except:
|
||||||
ifDebug: echo "Failed to create server on " & $item
|
error "Failed to create server for address", address = $item
|
||||||
|
|
||||||
for item in tas6:
|
for item in tas6:
|
||||||
try:
|
try:
|
||||||
ifDebug: echo "Create server on " & $item
|
info "Server created", ip6address = $item
|
||||||
var server = createStreamServer(item, processClient, {ReuseAddr},
|
var server = createStreamServer(item, processClient, {ReuseAddr},
|
||||||
udata = result)
|
udata = result)
|
||||||
result.servers.add(server)
|
result.servers.add(server)
|
||||||
except:
|
except:
|
||||||
ifDebug: echo "Failed to create server on " & $item
|
error "Failed to create server", address = $item
|
||||||
|
|
||||||
if len(result.servers) == 0:
|
if len(result.servers) == 0:
|
||||||
# Server was not bound, critical error.
|
# Server was not bound, critical error.
|
||||||
# TODO: Custom RpcException error
|
raise newException(RpcBindError,
|
||||||
raise newException(ValueError,
|
|
||||||
"Could not setup server on " & address & ":" & $int(port))
|
"Could not setup server on " & address & ":" & $int(port))
|
||||||
|
|
||||||
proc start*(server: RpcServer) =
|
proc start*(server: RpcServer) =
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user