refactor: Use workerData to initialise Worker

near · Sep 15, 2023 · f621411 · f621411
1 parent 843a402
commit f621411
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/runner/src/index.ts b/runner/src/index.ts
@@ -31,8 +31,9 @@ void (async function main () {
           return;
         }
 
-        const worker = new Worker('./dist/worker.js');
-        worker.postMessage({ streamKey });
+        const worker = new Worker('./dist/worker.js', {
+          workerData: { streamKey },
+        });
 
         worker.on('message', (message: Metric) => {
           METRICS[message.type].labels(message.labels).set(message.value);

diff --git a/runner/src/worker.ts b/runner/src/worker.ts
@@ -1,17 +1,21 @@
-import { parentPort } from 'worker_threads';
+import { parentPort, workerData } from 'worker_threads';
 
 import Indexer from './indexer';
 import RedisClient from './redis-client';
 
 const indexer = new Indexer('mainnet');
 const redisClient = new RedisClient();
 
-// eslint-disable-next-line
-parentPort?.on('message', async ({ streamKey }) => {
+if (parentPort === null) {
+  throw new Error('No parentPort');
+}
+
+void (async function main () {
+  const { streamKey } = workerData;
+
   console.log('Started processing stream: ', streamKey);
 
   let indexerName = '';
-
   while (true) {
     try {
       const startTime = performance.now();
@@ -61,4 +65,4 @@ parentPort?.on('message', async ({ streamKey }) => {
       console.log(`Failed: ${indexerName}`, err);
     }
   }
-});
+})();