fix: 🐛 fix worker error in nextjs

abelce · Oct 17, 2023 · 8bf17cb · 8bf17cb
1 parent 0e4fb17
commit 8bf17cb
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 29 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "let-hooks",
-  "version": "0.0.8",
+  "version": "0.0.9",
   "description": "A react hooks library",
   "keywords": [
     "react",

diff --git a/src/utils/work-timer/timer.worker.ts b/src/utils/work-timer/timer.worker.ts
@@ -1,34 +1,38 @@
+import isBrowser from '../isBrowser';
 import { MsgType } from './types';
 
-addEventListener('message', (message: MessageEvent) => {
-  const data = message.data;
-  switch (data.type) {
-    case MsgType.Create: {
-      const $timer = setTimeout(() => {
-        postMessage({
-          taskId: data.taskId,
-          type: MsgType.Execute,
-        });
+if (isBrowser()) {
+  const webWorker = self;
 
-        clearTimeout($timer);
-      }, data.wait);
+  webWorker.addEventListener('message', (message: MessageEvent) => {
+    const data = message.data;
+    switch (data.type) {
+      case MsgType.Create: {
+        const $timer = setTimeout(() => {
+          webWorker.postMessage({
+            taskId: data.taskId,
+            type: MsgType.Execute,
+          });
 
-      postMessage({
-        taskId: data.taskId,
-        timer: $timer,
-        type: MsgType.Create,
-      });
-      break;
-    }
-    case MsgType.Clear: {
-      clearTimeout(data.timerId);
-      postMessage({
-        taskId: data.taskId,
-        type: MsgType.Clear,
-      });
-      break;
-    }
-  }
-});
+          clearTimeout($timer);
+        }, data.wait);
 
+        webWorker.postMessage({
+          taskId: data.taskId,
+          timer: $timer,
+          type: MsgType.Create,
+        });
+        break;
+      }
+      case MsgType.Clear: {
+        clearTimeout(data.timerId);
+        webWorker.postMessage({
+          taskId: data.taskId,
+          type: MsgType.Clear,
+        });
+        break;
+      }
+    }
+  });
+}
 export default null as any;