#47 persistent context

__tests__/persistent-ctx.ts

@@ -0,0 +1,14 @@
+import { job, stop, start } from '../src/job'
+
+beforeAll(async () => await start())
+afterAll(async () => await stop())
+
+describe('Persistent ctx testing', () => {
+  it('should return persistent ctx after save', async () => {
+    await job(() => {}, {persistentCtx: {test: true}})
+    const persistentCtx = await job(() => {
+      return persistentCtx
+    })
+    expect(persistentCtx).toEqual({test: true})
+  })
+})

src/interfaces.ts

@@ -1,6 +1,7 @@
 export interface Config<T = {}, U = {}> {
   ctx?: T
   data?: U
+  persistentCtx?: any
 }
 
 export interface Task {

src/worker-pool.ts

@@ -94,15 +94,18 @@ class WorkerPool {
       // @ts-ignore
       const dataSerialized = v8.serialize(config.data)
       const dataStr = JSON.stringify(dataSerialized)
+      const persistentCtxStr = JSON.stringify(config.persistentCtx || {})
       const workerStr = `
-      async function __executor__() {
+      async function __executor__(persistentCtx) {
         const v8 = require('v8')
         ${variables}
         const dataParsed = JSON.parse('${dataStr}')
         const dataBuffer = Buffer.from(dataParsed.data)
         const dataDeserialized = v8.deserialize(dataBuffer)
         return await (${handler.toString()})(dataDeserialized)
       }
+
+      __executor__.persistentCtx = JSON.parse('${persistentCtxStr}')
       `
 
       // @ts-ignore

src/worker.js

@@ -1,5 +1,7 @@
 const { parentPort } = require('worker_threads')
 
+let persistentCtx = {}
+
 parentPort.on('message', async worker => {
   const response = {
     error: null,
@@ -9,7 +11,8 @@ parentPort.on('message', async worker => {
   try {
     eval(worker)
     // __executor__ is defined in worker
-    response.data = await __executor__()
+    Object.assign(persistentCtx, __executor__.persistentCtx)
+    response.data = await __executor__(persistentCtx)
     parentPort.postMessage(response)
   } catch (err) {
     response.data = null