@ouim/vectoriadb-server 0.1.2 → 0.1.3

package/README.md

@@ -189,15 +189,6 @@ A: The client SDK stringifies your filter function. The server then reconstructs
 
 ---
 
-## Future Roadmap
-
-- [ ] **JWT Authentication**: Add more robust auth options.
-- [ ] **Data Encryption**: Support for end-to-end encryption of metadata.
-- [ ] **Multi-Instance**: Support for multiple VectoriaDB instances managed by one server.
-- [ ] **Python Client**: A compatible SDK for Python applications.
-
----
-
 ## Credits
 
 This SDK builds on and integrates with [VectoriaDB](https://github.com/agentfront/vectoriadb). See the Vectoria product website at [https://agentfront.dev/vectoria](https://agentfront.dev/vectoria) for more information and additional resources.

package/index.js

@@ -1,23 +1,23 @@
 import VectoriaDBServer from './vectoriadb-server.js'
 import { FileStorageAdapter } from 'vectoriadb'
 
-// If run directly, start a demo server
-if (process.argv[1] && process.argv[1].endsWith('index.js')) {
-  const server = new VectoriaDBServer({
-    port: process.env.PORT ? Number(process.env.PORT) : 3001,
-    host: '0.0.0.0',
-    vectoriadbConfig: {
-      // default storage / model - user should override for production
-      storageAdapter: new FileStorageAdapter({ cacheDir: '/data/vectoriadb', namespace: 'default' }),
-    },
-    cors: ['http://localhost:3000'],
-  })
+// // If run directly, start a demo server
+// if (process.argv[1] && process.argv[1].endsWith('index.js')) {
+//   const server = new VectoriaDBServer({
+//     port: process.env.PORT ? Number(process.env.PORT) : 3001,
+//     host: '0.0.0.0',
+//     vectoriadbConfig: {
+//       // default storage / model - user should override for production
+//       storageAdapter: new FileStorageAdapter({ cacheDir: '/data/vectoriadb', namespace: 'default' }),
+//     },
+//     cors: ['http://localhost:3000'],
+//   })
 
-  server.listen().catch(err => {
-    console.error(err)
-    process.exit(1)
-  })
-}
+//   server.listen().catch(err => {
+//     console.error(err)
+//     process.exit(1)
+//   })
+// }
 
-export { VectoriaDBServer }
+export { VectoriaDBServer, FileStorageAdapter }
 export default VectoriaDBServer

package/listen-race-test.js

@@ -0,0 +1,28 @@
+import VectoriaDBServer from './vectoriadb-server.js'
+import { FileStorageAdapter } from 'vectoriadb'
+
+async function run() {
+  const server = new VectoriaDBServer({
+    port: 3002,
+    vectoriadbConfig: {
+      storageAdapter: new FileStorageAdapter({ cacheDir: './.cache/test-vectoriadb', namespace: 'test' }),
+    },
+  })
+
+  try {
+    // call listen() twice concurrently to reproduce the race if present
+    const results = await Promise.all([server.listen(), server.listen()])
+    console.log('listen() calls resolved, same instance:', results[0] === results[1])
+    console.log('server._started =', server._started)
+  } catch (err) {
+    console.error('concurrent listen failed:', err)
+    process.exitCode = 1
+  } finally {
+    await server.close()
+  }
+}
+
+run().catch(err => {
+  console.error(err)
+  process.exit(1)
+})

package/package.json

@@ -1,12 +1,11 @@
 {
   "name": "@ouim/vectoriadb-server",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "type": "module",
   "main": "index.js",
   "scripts": {
     "start": "node index.js",
-    "demo": "node demo.js",
-    "publish": "npm publish --access public"
+    "demo": "node demo.js"
   },
   "dependencies": {
     "@huggingface/transformers": "^3.8.1",

package/vectoriadb-server.js

@@ -19,63 +19,89 @@ export default class VectoriaDBServer {
     this._vectoria = null
     this._sockets = new Set()
     this._started = false
+    this._startPromise = null
   }
 
   async listen() {
-    if (this._started) return
-    // Initialize VectoriaDB (loads models / storage as configured)
-    this._vectoria = new VectoriaDB(this.vectoriadbConfig)
-    if (typeof this._vectoria.initialize === 'function') {
-      await this._vectoria.initialize()
-    }
-
-    this._http = http.createServer((req, res) => {
-      res.writeHead(200, { 'Content-Type': 'text/plain' })
-      res.end('VectoriaDB Server')
-    })
-
-    this._io = new IOServer(this._http, {
-      cors: { origin: this.cors.length ? this.cors : '*', methods: ['GET', 'POST'] },
-      maxHttpBufferSize: 1e7,
-    })
+    // fast-path: already started
+    if (this._started) return this
 
-    const nsp = this._io.of('/vectoriadb')
+    // if a start is already in progress, return the same promise
+    if (this._startPromise) return this._startPromise
 
-    nsp.use((socket, next) => {
-      // simple API key auth if configured
-      if (this.apiKey) {
-        const provided = socket.handshake.auth?.apiKey || socket.handshake.query?.apiKey
-        if (!provided || provided !== this.apiKey) {
-          return next(new Error('Unauthorized'))
+    this._startPromise = (async () => {
+      try {
+        // Initialize VectoriaDB (loads models / storage as configured)
+        this._vectoria = new VectoriaDB(this.vectoriadbConfig)
+        if (typeof this._vectoria.initialize === 'function') {
+          await this._vectoria.initialize()
         }
-      }
-      next()
-    })
-
-    nsp.on('connection', socket => {
-      this._sockets.add(socket)
-      socket.on('disconnect', () => this._sockets.delete(socket))
 
-      socket.on('request', async payload => {
-        // payload: { id, method, params, collection, timestamp }
+        this._http = http.createServer((req, res) => {
+          res.writeHead(200, { 'Content-Type': 'text/plain' })
+          res.end('VectoriaDB Server')
+        })
+
+        this._io = new IOServer(this._http, {
+          cors: { origin: this.cors.length ? this.cors : '*', methods: ['GET', 'POST'] },
+          maxHttpBufferSize: 1e7,
+        })
+
+        const nsp = this._io.of('/vectoriadb')
+
+        nsp.use((socket, next) => {
+          // simple API key auth if configured
+          if (this.apiKey) {
+            const provided = socket.handshake.auth?.apiKey || socket.handshake.query?.apiKey
+            if (!provided || provided !== this.apiKey) {
+              return next(new Error('Unauthorized'))
+            }
+          }
+          next()
+        })
+
+        nsp.on('connection', socket => {
+          this._sockets.add(socket)
+          socket.on('disconnect', () => this._sockets.delete(socket))
+
+          socket.on('request', async payload => {
+            // payload: { id, method, params, collection, timestamp }
+            try {
+              await this._handleRequest(socket, payload)
+            } catch (err) {
+              socket.emit('response', { id: payload?.id ?? null, result: null, error: { message: err.message }, took: 0 })
+            }
+          })
+
+          // allow ping from client
+          socket.on('health', cb => cb && cb({ ok: true, ts: Date.now() }))
+        })
+
+        await new Promise((res, rej) => {
+          this._http.listen(this.port, this.host, err => (err ? rej(err) : res()))
+        })
+
+        this._started = true
+        console.log(`VectoriaDB Server listening on ${this.host}:${this.port}`)
+        return this
+      } catch (err) {
+        // cleanup partial resources on failure
         try {
-          await this._handleRequest(socket, payload)
-        } catch (err) {
-          socket.emit('response', { id: payload?.id ?? null, result: null, error: { message: err.message }, took: 0 })
-        }
-      })
-
-      // allow ping from client
-      socket.on('health', cb => cb && cb({ ok: true, ts: Date.now() }))
-    })
-
-    await new Promise((res, rej) => {
-      this._http.listen(this.port, this.host, err => (err ? rej(err) : res()))
-    })
+          if (this._io) await this._io.close()
+        } catch (e) {}
+        try {
+          if (this._http) this._http.close(() => {})
+        } catch (e) {}
+        throw err
+      }
+    })()
 
-    this._started = true
-    console.log(`VectoriaDB Server listening on ${this.host}:${this.port}`)
-    return this
+    try {
+      return await this._startPromise
+    } finally {
+      // ensure the in-progress marker is cleared after attempt completes
+      this._startPromise = null
+    }
   }
 
   async _handleRequest(socket, payload) {