codeceptjs 4.0.0-beta.2 → 4.0.0-beta.21

package/lib/workerStorage.js

@@ -1,44 +1,47 @@
-import { isMainThread, parentPort } from 'worker_threads';
+import { isMainThread, parentPort } from 'worker_threads'
+import Container from './container.js'
 
-const workerObjects = {};
-const shareEvent = 'share';
+const workerObjects = {}
+const shareEvent = 'share'
 
-export const invokeWorkerListeners = (workerObj) => {
-  const { threadId } = workerObj;
-  workerObj.on('message', (messageData) => {
+const invokeWorkerListeners = workerObj => {
+  const { threadId } = workerObj
+  workerObj.on('message', messageData => {
     if (messageData.event === shareEvent) {
-      share(messageData.data);
+      Container.share(messageData.data)
     }
-  });
+  })
   workerObj.on('exit', () => {
-    delete workerObjects[threadId];
-  });
-};
-
-/**
- * Add worker object
- *
- * @api
- * @param {Worker} workerObj
- */
-export function addWorker(workerObj) {
-  // @ts-ignore
-  workerObjects[workerObj.threadId] = workerObj;
-  invokeWorkerListeners(workerObj);
+    delete workerObjects[threadId]
+  })
 }
 
-/**
- * Share data across workers
- *
- * @param {*} data
- */
-export function share(data) {
-  if (isMainThread) {
-    for (const workerObj of Object.values(workerObjects)) {
-      workerObj.postMessage({ data });
+class WorkerStorage {
+  /**
+   * Add worker object
+   *
+   * @api
+   * @param {Worker} workerObj
+   */
+  static addWorker(workerObj) {
+    workerObjects[workerObj.threadId] = workerObj
+    invokeWorkerListeners(workerObj)
+  }
+
+  /**
+   * Share data across workers
+   *
+   * @param {*} data
+   */
+  static share(data) {
+    if (isMainThread) {
+      for (const workerObj of Object.values(workerObjects)) {
+        workerObj.postMessage({ data })
+      }
+    } else {
+      parentPort.postMessage({ data, event: shareEvent })
     }
-  } else {
-    // @ts-ignore
-    parentPort.postMessage({ data, event: shareEvent });
   }
 }
+
+export default WorkerStorage