server side error handling

waku-org · Mar 21, 2024 · 672f0b0 · 672f0b0
1 parent 7c6ae58
commit 672f0b0
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/waku/waku_store/protocol.nim b/waku/waku_store/protocol.nim
@@ -76,11 +76,21 @@ proc handleQueryRequest*(
 
 proc initProtocolHandler(self: WakuStore) =
   proc handler(conn: Connection, proto: string) {.async, gcsafe, closure.} =
-    let reqBuf = await conn.readLp(MaxRpcSize.int)
+    let readRes = catch:
+      await conn.readLp(MaxRpcSize.int)
+
+    let reqBuf = readRes.valueOr:
+      error "Connection read error", error = error.msg
+      return
 
     let resBuf = await self.handleQueryRequest(conn.peerId, reqBuf)
 
-    await conn.writeLp(resBuf)
+    let writeRes = catch:
+      await conn.writeLp(resBuf)
+
+    if writeRes.isErr():
+      error "Connection write error", error = writeRes.error.msg
+      return
 
   self.handler = handler
   self.codec = WakuStoreCodec