Store/Legacy store close connections

Author: NagyZoltanPeter

waku/waku_store/client.nim

@@ -25,6 +25,9 @@ proc sendStoreRequest(
 ): Future[StoreQueryResult] {.async, gcsafe.} =
   var req = request
 
+  defer:
+    await connection.closeWithEof()
+
   if req.requestId == "":
     req.requestId = generateRequestId(self.rng)
 

waku/waku_store/protocol.nim

@@ -92,6 +92,10 @@ proc initProtocolHandler(self: WakuStore) =
     var successfulQuery = false ## only consider the correct queries in metrics
     var resBuf: StoreResp
     var queryDuration: float
+
+    defer:
+      await conn.closeWithEof()
+
     self.requestRateLimiter.checkUsageLimit(WakuStoreCodec, conn):
       let readRes = catch:
         await conn.readLp(DefaultMaxRpcSize.int)

waku/waku_store_legacy/client.nim

@@ -43,6 +43,9 @@ proc sendHistoryQueryRPC(
 
   let connection = connOpt.get()
 
+  defer:
+    await connection.closeWithEof()
+
   let requestId =
     if req.requestId != "":
       req.requestId

waku/waku_store_legacy/protocol.nim

@@ -114,6 +114,10 @@ proc initProtocolHandler(ws: WakuStore) =
     var successfulQuery = false ## only consider the correct queries in metrics
     var resBuf: StoreResp
     var queryDuration: float
+
+    defer:
+      await conn.closeWithEof()
+
     ws.requestRateLimiter.checkUsageLimit(WakuLegacyStoreCodec, conn):
       let readRes = catch:
         await conn.readLp(DefaultMaxRpcSize.int)