123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188 |
- 'use strict';
- Object.defineProperty(exports, '__esModule', {
- value: true
- });
- exports.default = void 0;
- 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;
- }
- /**
- * Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
- *
- * This source code is licensed under the MIT license found in the
- * LICENSE file in the root directory of this source tree.
- */
- /**
- * Priority queue that processes tasks in natural ordering (lower priority first)
- * accoridng to the priority computed by the function passed in the constructor.
- *
- * FIFO ordering isn't guaranteed for tasks with the same priority.
- *
- * Worker specific tasks with the same priority as a non-worker specific task
- * are always processed first.
- */
- class PriorityQueue {
- constructor(_computePriority) {
- this._computePriority = _computePriority;
- _defineProperty(this, '_queue', []);
- _defineProperty(this, '_sharedQueue', new MinHeap());
- }
- enqueue(task, workerId) {
- if (workerId == null) {
- this._enqueue(task, this._sharedQueue);
- } else {
- const queue = this._getWorkerQueue(workerId);
- this._enqueue(task, queue);
- }
- }
- _enqueue(task, queue) {
- const item = {
- priority: this._computePriority(task.request[2], ...task.request[3]),
- task
- };
- queue.add(item);
- }
- dequeue(workerId) {
- const workerQueue = this._getWorkerQueue(workerId);
- const workerTop = workerQueue.peek();
- const sharedTop = this._sharedQueue.peek(); // use the task from the worker queue if there's no task in the shared queue
- // or if the priority of the worker queue is smaller or equal to the
- // priority of the top task in the shared queue. The tasks of the
- // worker specific queue are preferred because no other worker can pick this
- // specific task up.
- if (
- sharedTop == null ||
- (workerTop != null && workerTop.priority <= sharedTop.priority)
- ) {
- var _workerQueue$poll$tas, _workerQueue$poll;
- return (_workerQueue$poll$tas =
- (_workerQueue$poll = workerQueue.poll()) === null ||
- _workerQueue$poll === void 0
- ? void 0
- : _workerQueue$poll.task) !== null && _workerQueue$poll$tas !== void 0
- ? _workerQueue$poll$tas
- : null;
- }
- return this._sharedQueue.poll().task;
- }
- _getWorkerQueue(workerId) {
- let queue = this._queue[workerId];
- if (queue == null) {
- queue = this._queue[workerId] = new MinHeap();
- }
- return queue;
- }
- }
- exports.default = PriorityQueue;
- class MinHeap {
- constructor() {
- _defineProperty(this, '_heap', []);
- }
- peek() {
- var _this$_heap$;
- return (_this$_heap$ = this._heap[0]) !== null && _this$_heap$ !== void 0
- ? _this$_heap$
- : null;
- }
- add(item) {
- const nodes = this._heap;
- nodes.push(item);
- if (nodes.length === 1) {
- return;
- }
- let currentIndex = nodes.length - 1; // Bubble up the added node as long as the parent is bigger
- while (currentIndex > 0) {
- const parentIndex = Math.floor((currentIndex + 1) / 2) - 1;
- const parent = nodes[parentIndex];
- if (parent.priority <= item.priority) {
- break;
- }
- nodes[currentIndex] = parent;
- nodes[parentIndex] = item;
- currentIndex = parentIndex;
- }
- }
- poll() {
- const nodes = this._heap;
- const result = nodes[0];
- const lastElement = nodes.pop(); // heap was empty or removed the last element
- if (result == null || nodes.length === 0) {
- return result !== null && result !== void 0 ? result : null;
- }
- let index = 0;
- nodes[0] =
- lastElement !== null && lastElement !== void 0 ? lastElement : null;
- const element = nodes[0];
- while (true) {
- let swapIndex = null;
- const rightChildIndex = (index + 1) * 2;
- const leftChildIndex = rightChildIndex - 1;
- const rightChild = nodes[rightChildIndex];
- const leftChild = nodes[leftChildIndex]; // if the left child is smaller, swap with the left
- if (leftChild != null && leftChild.priority < element.priority) {
- swapIndex = leftChildIndex;
- } // If the right child is smaller or the right child is smaller than the left
- // then swap with the right child
- if (
- rightChild != null &&
- rightChild.priority < (swapIndex == null ? element : leftChild).priority
- ) {
- swapIndex = rightChildIndex;
- }
- if (swapIndex == null) {
- break;
- }
- nodes[index] = nodes[swapIndex];
- nodes[swapIndex] = element;
- index = swapIndex;
- }
- return result;
- }
- }
|