@platformatic/itc 2.0.0-alpha.5 → 2.0.0-alpha.6

package/lib/itc.js

@@ -2,6 +2,7 @@
 
 const { randomUUID } = require('node:crypto')
 const { EventEmitter, once } = require('node:events')
+const { Unpromise } = require('@watchable/unpromise')
 const errors = require('./errors.js')
 
 const PLT_ITC_REQUEST_TYPE = 'PLT_ITC_REQUEST'
@@ -18,7 +19,7 @@ class ITC extends EventEmitter {
   #closePromise
   #closeAfterCurrentRequest
 
-  constructor ({ port }) {
+  constructor ({ port, handlers }) {
     super()
 
     this.port = port
@@ -30,6 +31,13 @@ class ITC extends EventEmitter {
 
     // Make sure the emitter handle a lot of listeners at once before raising a warning
     this.#requestEmitter.setMaxListeners(1e3)
+
+    // Register handlers provided with the constructor
+    if (typeof handlers === 'object') {
+      for (const [name, fn] of Object.entries(handlers)) {
+        this.handle(name, fn)
+      }
+    }
   }
 
   async send (name, message) {
@@ -43,7 +51,7 @@ class ITC extends EventEmitter {
 
     const responsePromise = once(this.#requestEmitter, request.reqId).then(([response]) => response)
 
-    const { error, data } = await Promise.race([responsePromise, this.#closePromise])
+    const { error, data } = await Unpromise.race([responsePromise, this.#closePromise])
 
     if (error !== null) throw error
     return data
@@ -207,7 +215,7 @@ class ITC extends EventEmitter {
       version: PLT_ITC_VERSION,
       reqId: randomUUID(),
       name,
-      data,
+      data
     }
   }
 
@@ -218,7 +226,7 @@ class ITC extends EventEmitter {
       reqId: request.reqId,
       name: request.name,
       error,
-      data,
+      data
     }
   }
 
@@ -227,7 +235,7 @@ class ITC extends EventEmitter {
       type: PLT_ITC_NOTIFICATION_TYPE,
       version: PLT_ITC_VERSION,
       name,
-      data,
+      data
     }
   }
 
@@ -236,7 +244,7 @@ class ITC extends EventEmitter {
       type: PLT_ITC_UNHANDLED_ERROR_TYPE,
       version: PLT_ITC_VERSION,
       error,
-      data: null,
+      data: null
     }
   }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@platformatic/itc",
-  "version": "2.0.0-alpha.5",
+  "version": "2.0.0-alpha.6",
   "description": "",
   "main": "index.js",
   "author": "Matteo Collina <hello@matteocollina.com>",
@@ -22,7 +22,8 @@
     "typescript": "^5.5.4"
   },
   "dependencies": {
-    "@fastify/error": "^3.4.1"
+    "@fastify/error": "^3.4.1",
+    "@watchable/unpromise": "^1.0.2"
   },
   "scripts": {
     "test": "npm run lint && borp --timeout=180000 --concurrency=1 --coverage && tsd",

package/test/itc.test.js

@@ -38,20 +38,26 @@ test('should send a request between threads', async t => {
 test('should support close while replying to a message', async t => {
   const { port1, port2 } = new MessageChannel()
 
-  const itc1 = new ITC({ port: port1 })
-  const itc2 = new ITC({ port: port2 })
-
   const requestName = 'test-command'
   const testRequest = { test: 'test-req-message' }
   const testResponse = { test: 'test-res-message' }
 
   const requests = []
-  itc2.handle(requestName, async request => {
-    requests.push(request)
-    itc2.close()
-    return testResponse
+
+  const itc1 = new ITC({ port: port1 })
+  const itc2 = new ITC({
+    port: port2,
+    handlers: {
+      [requestName] (request) {
+        requests.push(request)
+        itc2.close()
+        return testResponse
+      }
+    }
   })
 
+  itc2.handle()
+
   itc1.listen()
   itc2.listen()
 
@@ -360,8 +366,8 @@ test('should sanitize a request before sending', async t => {
     nested: {
       test: 'test-req-message',
       foo: () => {},
-      bar: Symbol('test'),
-    },
+      bar: Symbol('test')
+    }
   }
   const testResponse = { test: 'test-res-message' }
 
@@ -382,8 +388,8 @@ test('should sanitize a request before sending', async t => {
   assert.deepStrictEqual(requests, [
     {
       test: 'test-req-message',
-      nested: { test: 'test-req-message' },
-    },
+      nested: { test: 'test-req-message' }
+    }
   ])
 })