123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203 |
- 'use strict';
- Object.defineProperty(exports, '__esModule', {
- value: true
- });
- exports.default = void 0;
- var _types = require('./types');
- function _defineProperty(obj, key, value) {
- if (key in obj) {
- Object.defineProperty(obj, key, {
- value: value,
- enumerable: true,
- configurable: true,
- writable: true
- });
- } else {
- obj[key] = value;
- }
- return obj;
- }
- class Farm {
- constructor(numOfWorkers, callback, computeWorkerKey) {
- _defineProperty(this, '_computeWorkerKey', void 0);
- _defineProperty(this, '_cacheKeys', void 0);
- _defineProperty(this, '_callback', void 0);
- _defineProperty(this, '_last', void 0);
- _defineProperty(this, '_locks', void 0);
- _defineProperty(this, '_numOfWorkers', void 0);
- _defineProperty(this, '_offset', void 0);
- _defineProperty(this, '_queue', void 0);
- this._cacheKeys = Object.create(null);
- this._callback = callback;
- this._last = [];
- this._locks = [];
- this._numOfWorkers = numOfWorkers;
- this._offset = 0;
- this._queue = [];
- if (computeWorkerKey) {
- this._computeWorkerKey = computeWorkerKey;
- }
- }
- doWork(method, ...args) {
- const customMessageListeners = new Set();
- const addCustomMessageListener = listener => {
- customMessageListeners.add(listener);
- return () => {
- customMessageListeners.delete(listener);
- };
- };
- const onCustomMessage = message => {
- customMessageListeners.forEach(listener => listener(message));
- };
- const promise = new Promise((resolve, reject) => {
- const computeWorkerKey = this._computeWorkerKey;
- const request = [_types.CHILD_MESSAGE_CALL, false, method, args];
- let worker = null;
- let hash = null;
- if (computeWorkerKey) {
- hash = computeWorkerKey.call(this, method, ...args);
- worker = hash == null ? null : this._cacheKeys[hash];
- }
- const onStart = worker => {
- if (hash != null) {
- this._cacheKeys[hash] = worker;
- }
- };
- const onEnd = (error, result) => {
- customMessageListeners.clear();
- if (error) {
- reject(error);
- } else {
- resolve(result);
- }
- };
- const task = {
- onCustomMessage,
- onEnd,
- onStart,
- request
- };
- if (worker) {
- this._enqueue(task, worker.getWorkerId());
- } else {
- this._push(task);
- }
- });
- promise.UNSTABLE_onCustomMessage = addCustomMessageListener;
- return promise;
- }
- _getNextTask(workerId) {
- let queueHead = this._queue[workerId];
- while (queueHead && queueHead.task.request[1]) {
- queueHead = queueHead.next || null;
- }
- this._queue[workerId] = queueHead;
- return queueHead && queueHead.task;
- }
- _process(workerId) {
- if (this._isLocked(workerId)) {
- return this;
- }
- const task = this._getNextTask(workerId);
- if (!task) {
- return this;
- }
- const onEnd = (error, result) => {
- task.onEnd(error, result);
- this._unlock(workerId);
- this._process(workerId);
- };
- task.request[1] = true;
- this._lock(workerId);
- this._callback(
- workerId,
- task.request,
- task.onStart,
- onEnd,
- task.onCustomMessage
- );
- return this;
- }
- _enqueue(task, workerId) {
- const item = {
- next: null,
- task
- };
- if (task.request[1]) {
- return this;
- }
- if (this._queue[workerId]) {
- this._last[workerId].next = item;
- } else {
- this._queue[workerId] = item;
- }
- this._last[workerId] = item;
- this._process(workerId);
- return this;
- }
- _push(task) {
- for (let i = 0; i < this._numOfWorkers; i++) {
- this._enqueue(task, (this._offset + i) % this._numOfWorkers);
- }
- this._offset++;
- return this;
- }
- _lock(workerId) {
- this._locks[workerId] = true;
- }
- _unlock(workerId) {
- this._locks[workerId] = false;
- }
- _isLocked(workerId) {
- return this._locks[workerId];
- }
- }
- exports.default = Farm;
|