From b0e407e98fe4893be0b29cae21df6622c09d4e5f Mon Sep 17 00:00:00 2001
From: eight <eight04@gmail.com>
Date: Tue, 25 Sep 2018 14:52:35 +0800
Subject: [PATCH] Add: worker util

---
 js/worker-util.js | 101 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 101 insertions(+)
 create mode 100644 js/worker-util.js

diff --git a/js/worker-util.js b/js/worker-util.js
new file mode 100644
index 00000000..dd19ac6c
--- /dev/null
+++ b/js/worker-util.js
@@ -0,0 +1,101 @@
+/* global importScripts */
+'use strict';
+
+// eslint-disable-next-line no-var
+var workerUtil = (() => {
+  const loadedScripts = new Set();
+  return {createWorker, createAPI, loadScript, cloneError};
+
+  function createWorker({url, lifeTime = 30}) {
+    let worker;
+    let id;
+    let timer;
+    const pendingResponse = new Map();
+
+    return new Proxy({}, {
+      get: (target, prop) =>
+        (...args) => {
+          if (!worker) {
+            init();
+          }
+          return invoke(prop, args);
+        }
+    });
+
+    function init() {
+      id = 0;
+      worker = new Worker(url);
+      worker.onmessage = onMessage;
+    }
+
+    function uninit() {
+      worker.onmessage = null;
+      worker.terminate();
+      worker = null;
+    }
+
+    function onMessage(e) {
+      const message = e.data;
+      pendingResponse.get(message.id)[message.error ? 'reject' : 'resolve'](message.data);
+      pendingResponse.delete(message.id);
+      if (!pendingResponse.size && lifeTime >= 0) {
+        timer = setTimeout(uninit, lifeTime * 1000);
+      }
+    }
+
+    function invoke(action, args) {
+      return new Promise((resolve, reject) => {
+        pendingResponse.set(id, {resolve, reject});
+        clearTimeout(timer);
+        worker.postMessage({
+          id,
+          action,
+          args
+        });
+        id++;
+      });
+    }
+  }
+
+  function createAPI(methods) {
+    self.onmessage = e => {
+      const message = e.data;
+      Promise.resolve()
+        .then(() => methods[message.action](...message.args))
+        .then(result => ({
+          id: message.id,
+          error: false,
+          data: result
+        }))
+        .catch(err => ({
+          id: message.id,
+          error: true,
+          data: cloneError(err)
+        }))
+        .then(data => self.postMessage(data));
+    };
+  }
+
+  function cloneError(err) {
+    return Object.assign({
+      name: err.name,
+      stack: err.stack,
+      message: err.message,
+      lineNumber: err.lineNumber,
+      columnNumber: err.columnNumber,
+      fileName: err.fileName
+    }, err);
+  }
+
+  function loadScript(scripts) {
+    if (!Array.isArray(scripts)) {
+      scripts = [scripts];
+    }
+    const urls = scripts.filter(u => !loadedScripts.has(u));
+    if (!urls.length) {
+      return;
+    }
+    importScripts(...urls);
+    urls.forEach(u => loadedScripts.add(u));
+  }
+})();