resplite 1.1.10 → 1.1.12

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "resplite",
-  "version": "1.1.10",
+  "version": "1.1.12",
   "description": "A RESP2 server with practical Redis compatibility, backed by SQLite",
   "type": "module",
   "main": "src/index.js",

package/src/embed.js

@@ -34,11 +34,34 @@ export async function createRESPlite({
 } = {}) {
   const db = openDb(dbPath, { pragmaTemplate });
   const engine = createEngine({ db });
-  const server = net.createServer((socket) => handleConnection(socket, engine));
+  const connections = new Set();
+
+  const server = net.createServer((socket) => {
+    connections.add(socket);
+    socket.once('close', () => connections.delete(socket));
+    handleConnection(socket, engine);
+  });
   await new Promise((resolve) => server.listen(port, host, resolve));
+
+  let closePromise = null;
+  const close = () => {
+    if (closePromise) return closePromise;
+    closePromise = new Promise((resolve) => {
+      for (const socket of connections) {
+        socket.destroy();
+      }
+      connections.clear();
+      server.close(() => {
+        db.close();
+        resolve();
+      });
+    });
+    return closePromise;
+  };
+
   return {
     port: server.address().port,
     host,
-    close: () => new Promise((resolve) => server.close(() => { db.close(); resolve(); })),
+    close,
   };
 }