effect
Version:
The missing standard library for TypeScript, for writing production-grade software.
303 lines (301 loc) • 8.41 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.withScheduler = exports.timerBatched = exports.timer = exports.makeMatrix = exports.makeBatched = exports.make = exports.defaultShouldYield = exports.defaultScheduler = exports.currentScheduler = exports.SyncScheduler = exports.PriorityBuckets = exports.MixedScheduler = exports.ControlledScheduler = void 0;
var _Function = require("./Function.js");
var _GlobalValue = require("./GlobalValue.js");
var core = _interopRequireWildcard(require("./internal/core.js"));
function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }
function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != typeof e && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && {}.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
/**
* @since 2.0.0
*/
/**
* @since 2.0.0
* @category utils
*/
class PriorityBuckets {
/**
* @since 2.0.0
*/
buckets = [];
/**
* @since 2.0.0
*/
scheduleTask(task, priority) {
const length = this.buckets.length;
let bucket = undefined;
let index = 0;
for (; index < length; index++) {
if (this.buckets[index][0] <= priority) {
bucket = this.buckets[index];
} else {
break;
}
}
if (bucket && bucket[0] === priority) {
bucket[1].push(task);
} else if (index === length) {
this.buckets.push([priority, [task]]);
} else {
this.buckets.splice(index, 0, [priority, [task]]);
}
}
}
/**
* @since 2.0.0
* @category constructors
*/
exports.PriorityBuckets = PriorityBuckets;
class MixedScheduler {
maxNextTickBeforeTimer;
/**
* @since 2.0.0
*/
running = false;
/**
* @since 2.0.0
*/
tasks = /*#__PURE__*/new PriorityBuckets();
constructor(
/**
* @since 2.0.0
*/
maxNextTickBeforeTimer) {
this.maxNextTickBeforeTimer = maxNextTickBeforeTimer;
}
/**
* @since 2.0.0
*/
starveInternal(depth) {
const tasks = this.tasks.buckets;
this.tasks.buckets = [];
for (const [_, toRun] of tasks) {
for (let i = 0; i < toRun.length; i++) {
toRun[i]();
}
}
if (this.tasks.buckets.length === 0) {
this.running = false;
} else {
this.starve(depth);
}
}
/**
* @since 2.0.0
*/
starve(depth = 0) {
if (depth >= this.maxNextTickBeforeTimer) {
setTimeout(() => this.starveInternal(0), 0);
} else {
Promise.resolve(void 0).then(() => this.starveInternal(depth + 1));
}
}
/**
* @since 2.0.0
*/
shouldYield(fiber) {
return fiber.currentOpCount > fiber.getFiberRef(core.currentMaxOpsBeforeYield) ? fiber.getFiberRef(core.currentSchedulingPriority) : false;
}
/**
* @since 2.0.0
*/
scheduleTask(task, priority) {
this.tasks.scheduleTask(task, priority);
if (!this.running) {
this.running = true;
this.starve();
}
}
}
/**
* @since 2.0.0
* @category schedulers
*/
exports.MixedScheduler = MixedScheduler;
const defaultScheduler = exports.defaultScheduler = /*#__PURE__*/(0, _GlobalValue.globalValue)( /*#__PURE__*/Symbol.for("effect/Scheduler/defaultScheduler"), () => new MixedScheduler(2048));
/**
* @since 2.0.0
* @category constructors
*/
class SyncScheduler {
/**
* @since 2.0.0
*/
tasks = /*#__PURE__*/new PriorityBuckets();
/**
* @since 2.0.0
*/
deferred = false;
/**
* @since 2.0.0
*/
scheduleTask(task, priority) {
if (this.deferred) {
defaultScheduler.scheduleTask(task, priority);
} else {
this.tasks.scheduleTask(task, priority);
}
}
/**
* @since 2.0.0
*/
shouldYield(fiber) {
return fiber.currentOpCount > fiber.getFiberRef(core.currentMaxOpsBeforeYield) ? fiber.getFiberRef(core.currentSchedulingPriority) : false;
}
/**
* @since 2.0.0
*/
flush() {
while (this.tasks.buckets.length > 0) {
const tasks = this.tasks.buckets;
this.tasks.buckets = [];
for (const [_, toRun] of tasks) {
for (let i = 0; i < toRun.length; i++) {
toRun[i]();
}
}
}
this.deferred = true;
}
}
/**
* @since 2.0.0
* @category constructors
*/
exports.SyncScheduler = SyncScheduler;
class ControlledScheduler {
/**
* @since 2.0.0
*/
tasks = /*#__PURE__*/new PriorityBuckets();
/**
* @since 2.0.0
*/
deferred = false;
/**
* @since 2.0.0
*/
scheduleTask(task, priority) {
if (this.deferred) {
defaultScheduler.scheduleTask(task, priority);
} else {
this.tasks.scheduleTask(task, priority);
}
}
/**
* @since 2.0.0
*/
shouldYield(fiber) {
return fiber.currentOpCount > fiber.getFiberRef(core.currentMaxOpsBeforeYield) ? fiber.getFiberRef(core.currentSchedulingPriority) : false;
}
/**
* @since 2.0.0
*/
step() {
const tasks = this.tasks.buckets;
this.tasks.buckets = [];
for (const [_, toRun] of tasks) {
for (let i = 0; i < toRun.length; i++) {
toRun[i]();
}
}
}
}
/**
* @since 2.0.0
* @category constructors
*/
exports.ControlledScheduler = ControlledScheduler;
const makeMatrix = (...record) => {
const index = record.sort(([p0], [p1]) => p0 < p1 ? -1 : p0 > p1 ? 1 : 0);
return {
shouldYield(fiber) {
for (const scheduler of record) {
const priority = scheduler[1].shouldYield(fiber);
if (priority !== false) {
return priority;
}
}
return false;
},
scheduleTask(task, priority) {
let scheduler = undefined;
for (const i of index) {
if (priority >= i[0]) {
scheduler = i[1];
} else {
return (scheduler ?? defaultScheduler).scheduleTask(task, priority);
}
}
return (scheduler ?? defaultScheduler).scheduleTask(task, priority);
}
};
};
/**
* @since 2.0.0
* @category utilities
*/
exports.makeMatrix = makeMatrix;
const defaultShouldYield = fiber => {
return fiber.currentOpCount > fiber.getFiberRef(core.currentMaxOpsBeforeYield) ? fiber.getFiberRef(core.currentSchedulingPriority) : false;
};
/**
* @since 2.0.0
* @category constructors
*/
exports.defaultShouldYield = defaultShouldYield;
const make = (scheduleTask, shouldYield = defaultShouldYield) => ({
scheduleTask,
shouldYield
});
/**
* @since 2.0.0
* @category constructors
*/
exports.make = make;
const makeBatched = (callback, shouldYield = defaultShouldYield) => {
let running = false;
const tasks = new PriorityBuckets();
const starveInternal = () => {
const tasksToRun = tasks.buckets;
tasks.buckets = [];
for (const [_, toRun] of tasksToRun) {
for (let i = 0; i < toRun.length; i++) {
toRun[i]();
}
}
if (tasks.buckets.length === 0) {
running = false;
} else {
starve();
}
};
const starve = () => callback(starveInternal);
return make((task, priority) => {
tasks.scheduleTask(task, priority);
if (!running) {
running = true;
starve();
}
}, shouldYield);
};
/**
* @since 2.0.0
* @category constructors
*/
exports.makeBatched = makeBatched;
const timer = (ms, shouldYield = defaultShouldYield) => make(task => setTimeout(task, ms), shouldYield);
/**
* @since 2.0.0
* @category constructors
*/
exports.timer = timer;
const timerBatched = (ms, shouldYield = defaultShouldYield) => makeBatched(task => setTimeout(task, ms), shouldYield);
/** @internal */
exports.timerBatched = timerBatched;
const currentScheduler = exports.currentScheduler = /*#__PURE__*/(0, _GlobalValue.globalValue)( /*#__PURE__*/Symbol.for("effect/FiberRef/currentScheduler"), () => core.fiberRefUnsafeMake(defaultScheduler));
/** @internal */
const withScheduler = exports.withScheduler = /*#__PURE__*/(0, _Function.dual)(2, (self, scheduler) => core.fiberRefLocally(self, currentScheduler, scheduler));
//# sourceMappingURL=Scheduler.js.map