add close event

piscinajs · Sep 7, 2023 · a97d391 · a97d391
1 parent 440f54d
commit a97d391
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -1001,6 +1001,7 @@ class ThreadPool {
       this.publicInterface.emit('error', error);
     } finally {
       await this.destroy();
+      this.publicInterface.emit('close');
       this.closingUp = false;
     }
   }

diff --git a/test/pool-close.ts b/test/pool-close.ts
@@ -19,6 +19,7 @@ test('close()', async (t) => {
     setImmediate(() => t.resolves(pool.close(), 'close is resolved when all running tasks are completed'));
 
     await Promise.all([
+      t.resolves(once(pool, 'close'), 'handler is called when pool is closed'),
       t.resolves(task1, 'complete running task'),
       t.resolves(task2, 'complete running task')
     ]);
@@ -46,6 +47,7 @@ test('close({force: true})', async (t) => {
     setImmediate(() => t.resolves(pool.close({ force: true }), 'close is resolved when all running tasks are completed'));
 
     await Promise.all([
+      t.resolves(once(pool, 'close'), 'handler is called when pool is closed'),
       t.resolves(task1, 'complete running task'),
       t.rejects(task2, /The task has been aborted/, 'abort task that are not started yet')
     ]);
@@ -62,6 +64,7 @@ test('close({force: true})', async (t) => {
     setImmediate(() => t.resolves(pool.close({ force: true }), 'close is resolved when all running tasks are completed'));
 
     await Promise.all([
+      t.resolves(once(pool, 'close'), 'handler is called when pool is closed'),
       t.resolves(task1, 'complete running task'),
       t.resolves(task2, 'complete running task'),
       t.rejects(task3, /The task has been aborted/, 'abort task that are not started yet'),