UNPKG

effect

Version:

The missing standard library for TypeScript, for writing production-grade software.

1,363 lines 85.4 kB
import * as RA from "../Array.js"; import * as Boolean from "../Boolean.js"; import * as Chunk from "../Chunk.js"; import * as Context from "../Context.js"; import * as Deferred from "../Deferred.js"; import * as Effectable from "../Effectable.js"; import * as ExecutionStrategy from "../ExecutionStrategy.js"; import * as FiberId from "../FiberId.js"; import * as FiberRefs from "../FiberRefs.js"; import * as FiberRefsPatch from "../FiberRefsPatch.js"; import * as FiberStatus from "../FiberStatus.js"; import { dual, identity, pipe } from "../Function.js"; import { globalValue } from "../GlobalValue.js"; import * as HashMap from "../HashMap.js"; import * as HashSet from "../HashSet.js"; import * as Inspectable from "../Inspectable.js"; import * as LogLevel from "../LogLevel.js"; import * as Micro from "../Micro.js"; import * as MRef from "../MutableRef.js"; import * as Option from "../Option.js"; import { pipeArguments } from "../Pipeable.js"; import * as Predicate from "../Predicate.js"; import * as Ref from "../Ref.js"; import * as RuntimeFlagsPatch from "../RuntimeFlagsPatch.js"; import { currentScheduler } from "../Scheduler.js"; import { internalCall, yieldWrapGet } from "../Utils.js"; import * as RequestBlock_ from "./blockedRequests.js"; import * as internalCause from "./cause.js"; import * as clock from "./clock.js"; import { currentRequestMap } from "./completedRequestMap.js"; import * as concurrency from "./concurrency.js"; import { configProviderTag } from "./configProvider.js"; import * as internalEffect from "./core-effect.js"; import * as core from "./core.js"; import * as defaultServices from "./defaultServices.js"; import { consoleTag } from "./defaultServices/console.js"; import * as executionStrategy from "./executionStrategy.js"; import * as internalFiber from "./fiber.js"; import * as FiberMessage from "./fiberMessage.js"; import * as fiberRefs from "./fiberRefs.js"; import * as fiberScope from "./fiberScope.js"; import * as internalLogger from "./logger.js"; import * as metric from "./metric.js"; import * as metricBoundaries from "./metric/boundaries.js"; import * as metricLabel from "./metric/label.js"; import * as OpCodes from "./opCodes/effect.js"; import { randomTag } from "./random.js"; import { complete } from "./request.js"; import * as runtimeFlags_ from "./runtimeFlags.js"; import { OpSupervision } from "./runtimeFlags.js"; import * as supervisor from "./supervisor.js"; import * as SupervisorPatch from "./supervisor/patch.js"; import * as tracer from "./tracer.js"; import * as version from "./version.js"; /** @internal */ export const fiberStarted = /*#__PURE__*/metric.counter("effect_fiber_started", { incremental: true }); /** @internal */ export const fiberActive = /*#__PURE__*/metric.counter("effect_fiber_active"); /** @internal */ export const fiberSuccesses = /*#__PURE__*/metric.counter("effect_fiber_successes", { incremental: true }); /** @internal */ export const fiberFailures = /*#__PURE__*/metric.counter("effect_fiber_failures", { incremental: true }); /** @internal */ export const fiberLifetimes = /*#__PURE__*/metric.tagged( /*#__PURE__*/metric.histogram("effect_fiber_lifetimes", /*#__PURE__*/metricBoundaries.exponential({ start: 0.5, factor: 2, count: 35 })), "time_unit", "milliseconds"); /** @internal */ const EvaluationSignalContinue = "Continue"; /** @internal */ const EvaluationSignalDone = "Done"; /** @internal */ const EvaluationSignalYieldNow = "Yield"; const runtimeFiberVariance = { /* c8 ignore next */ _E: _ => _, /* c8 ignore next */ _A: _ => _ }; const absurd = _ => { throw new Error(`BUG: FiberRuntime - ${Inspectable.toStringUnknown(_)} - please report an issue at https://github.com/Effect-TS/effect/issues`); }; const YieldedOp = /*#__PURE__*/Symbol.for("effect/internal/fiberRuntime/YieldedOp"); const yieldedOpChannel = /*#__PURE__*/globalValue("effect/internal/fiberRuntime/yieldedOpChannel", () => ({ currentOp: null })); const contOpSuccess = { [OpCodes.OP_ON_SUCCESS]: (_, cont, value) => { return internalCall(() => cont.effect_instruction_i1(value)); }, ["OnStep"]: (_, _cont, value) => { return core.exitSucceed(core.exitSucceed(value)); }, [OpCodes.OP_ON_SUCCESS_AND_FAILURE]: (_, cont, value) => { return internalCall(() => cont.effect_instruction_i2(value)); }, [OpCodes.OP_REVERT_FLAGS]: (self, cont, value) => { self.patchRuntimeFlags(self.currentRuntimeFlags, cont.patch); if (runtimeFlags_.interruptible(self.currentRuntimeFlags) && self.isInterrupted()) { return core.exitFailCause(self.getInterruptedCause()); } else { return core.exitSucceed(value); } }, [OpCodes.OP_WHILE]: (self, cont, value) => { internalCall(() => cont.effect_instruction_i2(value)); if (internalCall(() => cont.effect_instruction_i0())) { self.pushStack(cont); return internalCall(() => cont.effect_instruction_i1()); } else { return core.void; } }, [OpCodes.OP_ITERATOR]: (self, cont, value) => { const state = internalCall(() => cont.effect_instruction_i0.next(value)); if (state.done) return core.exitSucceed(state.value); self.pushStack(cont); return yieldWrapGet(state.value); } }; const drainQueueWhileRunningTable = { [FiberMessage.OP_INTERRUPT_SIGNAL]: (self, runtimeFlags, cur, message) => { self.processNewInterruptSignal(message.cause); return runtimeFlags_.interruptible(runtimeFlags) ? core.exitFailCause(message.cause) : cur; }, [FiberMessage.OP_RESUME]: (_self, _runtimeFlags, _cur, _message) => { throw new Error("It is illegal to have multiple concurrent run loops in a single fiber"); }, [FiberMessage.OP_STATEFUL]: (self, runtimeFlags, cur, message) => { message.onFiber(self, FiberStatus.running(runtimeFlags)); return cur; }, [FiberMessage.OP_YIELD_NOW]: (_self, _runtimeFlags, cur, _message) => { return core.flatMap(core.yieldNow(), () => cur); } }; /** * Executes all requests, submitting requests to each data source in parallel. */ const runBlockedRequests = self => core.forEachSequentialDiscard(RequestBlock_.flatten(self), requestsByRequestResolver => forEachConcurrentDiscard(RequestBlock_.sequentialCollectionToChunk(requestsByRequestResolver), ([dataSource, sequential]) => { const map = new Map(); const arr = []; for (const block of sequential) { arr.push(Chunk.toReadonlyArray(block)); for (const entry of block) { map.set(entry.request, entry); } } const flat = arr.flat(); return core.fiberRefLocally(invokeWithInterrupt(dataSource.runAll(arr), flat, () => flat.forEach(entry => { entry.listeners.interrupted = true; })), currentRequestMap, map); }, false, false)); const _version = /*#__PURE__*/version.getCurrentVersion(); /** @internal */ export class FiberRuntime extends Effectable.Class { [internalFiber.FiberTypeId] = internalFiber.fiberVariance; [internalFiber.RuntimeFiberTypeId] = runtimeFiberVariance; _fiberRefs; _fiberId; _queue = /*#__PURE__*/new Array(); _children = null; _observers = /*#__PURE__*/new Array(); _running = false; _stack = []; _asyncInterruptor = null; _asyncBlockingOn = null; _exitValue = null; _steps = []; _isYielding = false; currentRuntimeFlags; currentOpCount = 0; currentSupervisor; currentScheduler; currentTracer; currentSpan; currentContext; currentDefaultServices; constructor(fiberId, fiberRefs0, runtimeFlags0) { super(); this.currentRuntimeFlags = runtimeFlags0; this._fiberId = fiberId; this._fiberRefs = fiberRefs0; if (runtimeFlags_.runtimeMetrics(runtimeFlags0)) { const tags = this.getFiberRef(core.currentMetricLabels); fiberStarted.unsafeUpdate(1, tags); fiberActive.unsafeUpdate(1, tags); } this.refreshRefCache(); } commit() { return internalFiber.join(this); } /** * The identity of the fiber. */ id() { return this._fiberId; } /** * Begins execution of the effect associated with this fiber on in the * background. This can be called to "kick off" execution of a fiber after * it has been created. */ resume(effect) { this.tell(FiberMessage.resume(effect)); } /** * The status of the fiber. */ get status() { return this.ask((_, status) => status); } /** * Gets the fiber runtime flags. */ get runtimeFlags() { return this.ask((state, status) => { if (FiberStatus.isDone(status)) { return state.currentRuntimeFlags; } return status.runtimeFlags; }); } /** * Returns the current `FiberScope` for the fiber. */ scope() { return fiberScope.unsafeMake(this); } /** * Retrieves the immediate children of the fiber. */ get children() { return this.ask(fiber => Array.from(fiber.getChildren())); } /** * Gets the fiber's set of children. */ getChildren() { if (this._children === null) { this._children = new Set(); } return this._children; } /** * Retrieves the interrupted cause of the fiber, which will be `Cause.empty` * if the fiber has not been interrupted. * * **NOTE**: This method is safe to invoke on any fiber, but if not invoked * on this fiber, then values derived from the fiber's state (including the * log annotations and log level) may not be up-to-date. */ getInterruptedCause() { return this.getFiberRef(core.currentInterruptedCause); } /** * Retrieves the whole set of fiber refs. */ fiberRefs() { return this.ask(fiber => fiber.getFiberRefs()); } /** * Returns an effect that will contain information computed from the fiber * state and status while running on the fiber. * * This allows the outside world to interact safely with mutable fiber state * without locks or immutable data. */ ask(f) { return core.suspend(() => { const deferred = core.deferredUnsafeMake(this._fiberId); this.tell(FiberMessage.stateful((fiber, status) => { core.deferredUnsafeDone(deferred, core.sync(() => f(fiber, status))); })); return core.deferredAwait(deferred); }); } /** * Adds a message to be processed by the fiber on the fiber. */ tell(message) { this._queue.push(message); if (!this._running) { this._running = true; this.drainQueueLaterOnExecutor(); } } get await() { return core.async(resume => { const cb = exit => resume(core.succeed(exit)); this.tell(FiberMessage.stateful((fiber, _) => { if (fiber._exitValue !== null) { cb(this._exitValue); } else { fiber.addObserver(cb); } })); return core.sync(() => this.tell(FiberMessage.stateful((fiber, _) => { fiber.removeObserver(cb); }))); }, this.id()); } get inheritAll() { return core.withFiberRuntime((parentFiber, parentStatus) => { const parentFiberId = parentFiber.id(); const parentFiberRefs = parentFiber.getFiberRefs(); const parentRuntimeFlags = parentStatus.runtimeFlags; const childFiberRefs = this.getFiberRefs(); const updatedFiberRefs = fiberRefs.joinAs(parentFiberRefs, parentFiberId, childFiberRefs); parentFiber.setFiberRefs(updatedFiberRefs); const updatedRuntimeFlags = parentFiber.getFiberRef(currentRuntimeFlags); const patch = pipe(runtimeFlags_.diff(parentRuntimeFlags, updatedRuntimeFlags), // Do not inherit WindDown or Interruption! RuntimeFlagsPatch.exclude(runtimeFlags_.Interruption), RuntimeFlagsPatch.exclude(runtimeFlags_.WindDown)); return core.updateRuntimeFlags(patch); }); } /** * Tentatively observes the fiber, but returns immediately if it is not * already done. */ get poll() { return core.sync(() => Option.fromNullable(this._exitValue)); } /** * Unsafely observes the fiber, but returns immediately if it is not * already done. */ unsafePoll() { return this._exitValue; } /** * In the background, interrupts the fiber as if interrupted from the specified fiber. */ interruptAsFork(fiberId) { return core.sync(() => this.tell(FiberMessage.interruptSignal(internalCause.interrupt(fiberId)))); } /** * In the background, interrupts the fiber as if interrupted from the specified fiber. */ unsafeInterruptAsFork(fiberId) { this.tell(FiberMessage.interruptSignal(internalCause.interrupt(fiberId))); } /** * Adds an observer to the list of observers. * * **NOTE**: This method must be invoked by the fiber itself. */ addObserver(observer) { if (this._exitValue !== null) { observer(this._exitValue); } else { this._observers.push(observer); } } /** * Removes the specified observer from the list of observers that will be * notified when the fiber exits. * * **NOTE**: This method must be invoked by the fiber itself. */ removeObserver(observer) { this._observers = this._observers.filter(o => o !== observer); } /** * Retrieves all fiber refs of the fiber. * * **NOTE**: This method is safe to invoke on any fiber, but if not invoked * on this fiber, then values derived from the fiber's state (including the * log annotations and log level) may not be up-to-date. */ getFiberRefs() { this.setFiberRef(currentRuntimeFlags, this.currentRuntimeFlags); return this._fiberRefs; } /** * Deletes the specified fiber ref. * * **NOTE**: This method must be invoked by the fiber itself. */ unsafeDeleteFiberRef(fiberRef) { this._fiberRefs = fiberRefs.delete_(this._fiberRefs, fiberRef); } /** * Retrieves the state of the fiber ref, or else its initial value. * * **NOTE**: This method is safe to invoke on any fiber, but if not invoked * on this fiber, then values derived from the fiber's state (including the * log annotations and log level) may not be up-to-date. */ getFiberRef(fiberRef) { if (this._fiberRefs.locals.has(fiberRef)) { return this._fiberRefs.locals.get(fiberRef)[0][1]; } return fiberRef.initial; } /** * Sets the fiber ref to the specified value. * * **NOTE**: This method must be invoked by the fiber itself. */ setFiberRef(fiberRef, value) { this._fiberRefs = fiberRefs.updateAs(this._fiberRefs, { fiberId: this._fiberId, fiberRef, value }); this.refreshRefCache(); } refreshRefCache() { this.currentDefaultServices = this.getFiberRef(defaultServices.currentServices); this.currentTracer = this.currentDefaultServices.unsafeMap.get(tracer.tracerTag.key); this.currentSupervisor = this.getFiberRef(currentSupervisor); this.currentScheduler = this.getFiberRef(currentScheduler); this.currentContext = this.getFiberRef(core.currentContext); this.currentSpan = this.currentContext.unsafeMap.get(tracer.spanTag.key); } /** * Wholesale replaces all fiber refs of this fiber. * * **NOTE**: This method must be invoked by the fiber itself. */ setFiberRefs(fiberRefs) { this._fiberRefs = fiberRefs; this.refreshRefCache(); } /** * Adds a reference to the specified fiber inside the children set. * * **NOTE**: This method must be invoked by the fiber itself. */ addChild(child) { this.getChildren().add(child); } /** * Removes a reference to the specified fiber inside the children set. * * **NOTE**: This method must be invoked by the fiber itself. */ removeChild(child) { this.getChildren().delete(child); } /** * Transfers all children of this fiber that are currently running to the * specified fiber scope. * * **NOTE**: This method must be invoked by the fiber itself after it has * evaluated the effects but prior to exiting. */ transferChildren(scope) { const children = this._children; // Clear the children of the current fiber this._children = null; if (children !== null && children.size > 0) { for (const child of children) { // If the child is still running, add it to the scope if (child._exitValue === null) { scope.add(this.currentRuntimeFlags, child); } } } } /** * On the current thread, executes all messages in the fiber's inbox. This * method may return before all work is done, in the event the fiber executes * an asynchronous operation. * * **NOTE**: This method must be invoked by the fiber itself. */ drainQueueOnCurrentThread() { let recurse = true; while (recurse) { let evaluationSignal = EvaluationSignalContinue; const prev = globalThis[internalFiber.currentFiberURI]; globalThis[internalFiber.currentFiberURI] = this; try { while (evaluationSignal === EvaluationSignalContinue) { evaluationSignal = this._queue.length === 0 ? EvaluationSignalDone : this.evaluateMessageWhileSuspended(this._queue.splice(0, 1)[0]); } } finally { this._running = false; globalThis[internalFiber.currentFiberURI] = prev; } // Maybe someone added something to the queue between us checking, and us // giving up the drain. If so, we need to restart the draining, but only // if we beat everyone else to the restart: if (this._queue.length > 0 && !this._running) { this._running = true; if (evaluationSignal === EvaluationSignalYieldNow) { this.drainQueueLaterOnExecutor(); recurse = false; } else { recurse = true; } } else { recurse = false; } } } /** * Schedules the execution of all messages in the fiber's inbox. * * This method will return immediately after the scheduling * operation is completed, but potentially before such messages have been * executed. * * **NOTE**: This method must be invoked by the fiber itself. */ drainQueueLaterOnExecutor() { this.currentScheduler.scheduleTask(this.run, this.getFiberRef(core.currentSchedulingPriority)); } /** * Drains the fiber's message queue while the fiber is actively running, * returning the next effect to execute, which may be the input effect if no * additional effect needs to be executed. * * **NOTE**: This method must be invoked by the fiber itself. */ drainQueueWhileRunning(runtimeFlags, cur0) { let cur = cur0; while (this._queue.length > 0) { const message = this._queue.splice(0, 1)[0]; // @ts-expect-error cur = drainQueueWhileRunningTable[message._tag](this, runtimeFlags, cur, message); } return cur; } /** * Determines if the fiber is interrupted. * * **NOTE**: This method is safe to invoke on any fiber, but if not invoked * on this fiber, then values derived from the fiber's state (including the * log annotations and log level) may not be up-to-date. */ isInterrupted() { return !internalCause.isEmpty(this.getFiberRef(core.currentInterruptedCause)); } /** * Adds an interruptor to the set of interruptors that are interrupting this * fiber. * * **NOTE**: This method must be invoked by the fiber itself. */ addInterruptedCause(cause) { const oldSC = this.getFiberRef(core.currentInterruptedCause); this.setFiberRef(core.currentInterruptedCause, internalCause.sequential(oldSC, cause)); } /** * Processes a new incoming interrupt signal. * * **NOTE**: This method must be invoked by the fiber itself. */ processNewInterruptSignal(cause) { this.addInterruptedCause(cause); this.sendInterruptSignalToAllChildren(); } /** * Interrupts all children of the current fiber, returning an effect that will * await the exit of the children. This method will return null if the fiber * has no children. * * **NOTE**: This method must be invoked by the fiber itself. */ sendInterruptSignalToAllChildren() { if (this._children === null || this._children.size === 0) { return false; } let told = false; for (const child of this._children) { child.tell(FiberMessage.interruptSignal(internalCause.interrupt(this.id()))); told = true; } return told; } /** * Interrupts all children of the current fiber, returning an effect that will * await the exit of the children. This method will return null if the fiber * has no children. * * **NOTE**: This method must be invoked by the fiber itself. */ interruptAllChildren() { if (this.sendInterruptSignalToAllChildren()) { const it = this._children.values(); this._children = null; let isDone = false; const body = () => { const next = it.next(); if (!next.done) { return core.asVoid(next.value.await); } else { return core.sync(() => { isDone = true; }); } }; return core.whileLoop({ while: () => !isDone, body, step: () => { // } }); } return null; } reportExitValue(exit) { if (runtimeFlags_.runtimeMetrics(this.currentRuntimeFlags)) { const tags = this.getFiberRef(core.currentMetricLabels); const startTimeMillis = this.id().startTimeMillis; const endTimeMillis = Date.now(); fiberLifetimes.unsafeUpdate(endTimeMillis - startTimeMillis, tags); fiberActive.unsafeUpdate(-1, tags); switch (exit._tag) { case OpCodes.OP_SUCCESS: { fiberSuccesses.unsafeUpdate(1, tags); break; } case OpCodes.OP_FAILURE: { fiberFailures.unsafeUpdate(1, tags); break; } } } if (exit._tag === "Failure") { const level = this.getFiberRef(core.currentUnhandledErrorLogLevel); if (!internalCause.isInterruptedOnly(exit.cause) && level._tag === "Some") { this.log("Fiber terminated with an unhandled error", exit.cause, level); } } } setExitValue(exit) { this._exitValue = exit; this.reportExitValue(exit); for (let i = this._observers.length - 1; i >= 0; i--) { this._observers[i](exit); } this._observers = []; } getLoggers() { return this.getFiberRef(currentLoggers); } log(message, cause, overrideLogLevel) { const logLevel = Option.isSome(overrideLogLevel) ? overrideLogLevel.value : this.getFiberRef(core.currentLogLevel); const minimumLogLevel = this.getFiberRef(currentMinimumLogLevel); if (LogLevel.greaterThan(minimumLogLevel, logLevel)) { return; } const spans = this.getFiberRef(core.currentLogSpan); const annotations = this.getFiberRef(core.currentLogAnnotations); const loggers = this.getLoggers(); const contextMap = this.getFiberRefs(); if (HashSet.size(loggers) > 0) { const clockService = Context.get(this.getFiberRef(defaultServices.currentServices), clock.clockTag); const date = new Date(clockService.unsafeCurrentTimeMillis()); Inspectable.withRedactableContext(contextMap, () => { for (const logger of loggers) { logger.log({ fiberId: this.id(), logLevel, message, cause, context: contextMap, spans, annotations, date }); } }); } } /** * Evaluates a single message on the current thread, while the fiber is * suspended. This method should only be called while evaluation of the * fiber's effect is suspended due to an asynchronous operation. * * **NOTE**: This method must be invoked by the fiber itself. */ evaluateMessageWhileSuspended(message) { switch (message._tag) { case FiberMessage.OP_YIELD_NOW: { return EvaluationSignalYieldNow; } case FiberMessage.OP_INTERRUPT_SIGNAL: { this.processNewInterruptSignal(message.cause); if (this._asyncInterruptor !== null) { this._asyncInterruptor(core.exitFailCause(message.cause)); this._asyncInterruptor = null; } return EvaluationSignalContinue; } case FiberMessage.OP_RESUME: { this._asyncInterruptor = null; this._asyncBlockingOn = null; this.evaluateEffect(message.effect); return EvaluationSignalContinue; } case FiberMessage.OP_STATEFUL: { message.onFiber(this, this._exitValue !== null ? FiberStatus.done : FiberStatus.suspended(this.currentRuntimeFlags, this._asyncBlockingOn)); return EvaluationSignalContinue; } default: { return absurd(message); } } } /** * Evaluates an effect until completion, potentially asynchronously. * * **NOTE**: This method must be invoked by the fiber itself. */ evaluateEffect(effect0) { this.currentSupervisor.onResume(this); try { let effect = runtimeFlags_.interruptible(this.currentRuntimeFlags) && this.isInterrupted() ? core.exitFailCause(this.getInterruptedCause()) : effect0; while (effect !== null) { const eff = effect; const exit = this.runLoop(eff); if (exit === YieldedOp) { const op = yieldedOpChannel.currentOp; yieldedOpChannel.currentOp = null; if (op._op === OpCodes.OP_YIELD) { if (runtimeFlags_.cooperativeYielding(this.currentRuntimeFlags)) { this.tell(FiberMessage.yieldNow()); this.tell(FiberMessage.resume(core.exitVoid)); effect = null; } else { effect = core.exitVoid; } } else if (op._op === OpCodes.OP_ASYNC) { // Terminate this evaluation, async resumption will continue evaluation: effect = null; } } else { this.currentRuntimeFlags = pipe(this.currentRuntimeFlags, runtimeFlags_.enable(runtimeFlags_.WindDown)); const interruption = this.interruptAllChildren(); if (interruption !== null) { effect = core.flatMap(interruption, () => exit); } else { if (this._queue.length === 0) { // No more messages to process, so we will allow the fiber to end life: this.setExitValue(exit); } else { // There are messages, possibly added by the final op executed by // the fiber. To be safe, we should execute those now before we // allow the fiber to end life: this.tell(FiberMessage.resume(exit)); } effect = null; } } } } finally { this.currentSupervisor.onSuspend(this); } } /** * Begins execution of the effect associated with this fiber on the current * thread. This can be called to "kick off" execution of a fiber after it has * been created, in hopes that the effect can be executed synchronously. * * This is not the normal way of starting a fiber, but it is useful when the * express goal of executing the fiber is to synchronously produce its exit. */ start(effect) { if (!this._running) { this._running = true; const prev = globalThis[internalFiber.currentFiberURI]; globalThis[internalFiber.currentFiberURI] = this; try { this.evaluateEffect(effect); } finally { this._running = false; globalThis[internalFiber.currentFiberURI] = prev; // Because we're special casing `start`, we have to be responsible // for spinning up the fiber if there were new messages added to // the queue between the completion of the effect and the transition // to the not running state. if (this._queue.length > 0) { this.drainQueueLaterOnExecutor(); } } } else { this.tell(FiberMessage.resume(effect)); } } /** * Begins execution of the effect associated with this fiber on in the * background, and on the correct thread pool. This can be called to "kick * off" execution of a fiber after it has been created, in hopes that the * effect can be executed synchronously. */ startFork(effect) { this.tell(FiberMessage.resume(effect)); } /** * Takes the current runtime flags, patches them to return the new runtime * flags, and then makes any changes necessary to fiber state based on the * specified patch. * * **NOTE**: This method must be invoked by the fiber itself. */ patchRuntimeFlags(oldRuntimeFlags, patch) { const newRuntimeFlags = runtimeFlags_.patch(oldRuntimeFlags, patch); globalThis[internalFiber.currentFiberURI] = this; this.currentRuntimeFlags = newRuntimeFlags; return newRuntimeFlags; } /** * Initiates an asynchronous operation, by building a callback that will * resume execution, and then feeding that callback to the registration * function, handling error cases and repeated resumptions appropriately. * * **NOTE**: This method must be invoked by the fiber itself. */ initiateAsync(runtimeFlags, asyncRegister) { let alreadyCalled = false; const callback = effect => { if (!alreadyCalled) { alreadyCalled = true; this.tell(FiberMessage.resume(effect)); } }; if (runtimeFlags_.interruptible(runtimeFlags)) { this._asyncInterruptor = callback; } try { asyncRegister(callback); } catch (e) { callback(core.failCause(internalCause.die(e))); } } pushStack(cont) { this._stack.push(cont); if (cont._op === "OnStep") { this._steps.push({ refs: this.getFiberRefs(), flags: this.currentRuntimeFlags }); } } popStack() { const item = this._stack.pop(); if (item) { if (item._op === "OnStep") { this._steps.pop(); } return item; } return; } getNextSuccessCont() { let frame = this.popStack(); while (frame) { if (frame._op !== OpCodes.OP_ON_FAILURE) { return frame; } frame = this.popStack(); } } getNextFailCont() { let frame = this.popStack(); while (frame) { if (frame._op !== OpCodes.OP_ON_SUCCESS && frame._op !== OpCodes.OP_WHILE && frame._op !== OpCodes.OP_ITERATOR) { return frame; } frame = this.popStack(); } } [OpCodes.OP_TAG](op) { return core.sync(() => Context.unsafeGet(this.currentContext, op)); } ["Left"](op) { return core.fail(op.left); } ["None"](_) { return core.fail(new core.NoSuchElementException()); } ["Right"](op) { return core.exitSucceed(op.right); } ["Some"](op) { return core.exitSucceed(op.value); } ["Micro"](op) { return core.unsafeAsync(microResume => { let resume = microResume; const fiber = Micro.runFork(Micro.provideContext(op, this.currentContext)); fiber.addObserver(exit => { if (exit._tag === "Success") { return resume(core.exitSucceed(exit.value)); } switch (exit.cause._tag) { case "Interrupt": { return resume(core.exitFailCause(internalCause.interrupt(FiberId.none))); } case "Fail": { return resume(core.fail(exit.cause.error)); } case "Die": { return resume(core.die(exit.cause.defect)); } } }); return core.unsafeAsync(abortResume => { resume = _ => { abortResume(core.void); }; fiber.unsafeInterrupt(); }); }); } [OpCodes.OP_SYNC](op) { const value = internalCall(() => op.effect_instruction_i0()); const cont = this.getNextSuccessCont(); if (cont !== undefined) { if (!(cont._op in contOpSuccess)) { // @ts-expect-error absurd(cont); } // @ts-expect-error return contOpSuccess[cont._op](this, cont, value); } else { yieldedOpChannel.currentOp = core.exitSucceed(value); return YieldedOp; } } [OpCodes.OP_SUCCESS](op) { const oldCur = op; const cont = this.getNextSuccessCont(); if (cont !== undefined) { if (!(cont._op in contOpSuccess)) { // @ts-expect-error absurd(cont); } // @ts-expect-error return contOpSuccess[cont._op](this, cont, oldCur.effect_instruction_i0); } else { yieldedOpChannel.currentOp = oldCur; return YieldedOp; } } [OpCodes.OP_FAILURE](op) { const cause = op.effect_instruction_i0; const cont = this.getNextFailCont(); if (cont !== undefined) { switch (cont._op) { case OpCodes.OP_ON_FAILURE: case OpCodes.OP_ON_SUCCESS_AND_FAILURE: { if (!(runtimeFlags_.interruptible(this.currentRuntimeFlags) && this.isInterrupted())) { return internalCall(() => cont.effect_instruction_i1(cause)); } else { return core.exitFailCause(internalCause.stripFailures(cause)); } } case "OnStep": { if (!(runtimeFlags_.interruptible(this.currentRuntimeFlags) && this.isInterrupted())) { return core.exitSucceed(core.exitFailCause(cause)); } else { return core.exitFailCause(internalCause.stripFailures(cause)); } } case OpCodes.OP_REVERT_FLAGS: { this.patchRuntimeFlags(this.currentRuntimeFlags, cont.patch); if (runtimeFlags_.interruptible(this.currentRuntimeFlags) && this.isInterrupted()) { return core.exitFailCause(internalCause.sequential(cause, this.getInterruptedCause())); } else { return core.exitFailCause(cause); } } default: { absurd(cont); } } } else { yieldedOpChannel.currentOp = core.exitFailCause(cause); return YieldedOp; } } [OpCodes.OP_WITH_RUNTIME](op) { return internalCall(() => op.effect_instruction_i0(this, FiberStatus.running(this.currentRuntimeFlags))); } ["Blocked"](op) { const refs = this.getFiberRefs(); const flags = this.currentRuntimeFlags; if (this._steps.length > 0) { const frames = []; const snap = this._steps[this._steps.length - 1]; let frame = this.popStack(); while (frame && frame._op !== "OnStep") { frames.push(frame); frame = this.popStack(); } this.setFiberRefs(snap.refs); this.currentRuntimeFlags = snap.flags; const patchRefs = FiberRefsPatch.diff(snap.refs, refs); const patchFlags = runtimeFlags_.diff(snap.flags, flags); return core.exitSucceed(core.blocked(op.effect_instruction_i0, core.withFiberRuntime(newFiber => { while (frames.length > 0) { newFiber.pushStack(frames.pop()); } newFiber.setFiberRefs(FiberRefsPatch.patch(newFiber.id(), newFiber.getFiberRefs())(patchRefs)); newFiber.currentRuntimeFlags = runtimeFlags_.patch(patchFlags)(newFiber.currentRuntimeFlags); return op.effect_instruction_i1; }))); } return core.uninterruptibleMask(restore => core.flatMap(forkDaemon(core.runRequestBlock(op.effect_instruction_i0)), () => restore(op.effect_instruction_i1))); } ["RunBlocked"](op) { return runBlockedRequests(op.effect_instruction_i0); } [OpCodes.OP_UPDATE_RUNTIME_FLAGS](op) { const updateFlags = op.effect_instruction_i0; const oldRuntimeFlags = this.currentRuntimeFlags; const newRuntimeFlags = runtimeFlags_.patch(oldRuntimeFlags, updateFlags); // One more chance to short circuit: if we're immediately going // to interrupt. Interruption will cause immediate reversion of // the flag, so as long as we "peek ahead", there's no need to // set them to begin with. if (runtimeFlags_.interruptible(newRuntimeFlags) && this.isInterrupted()) { return core.exitFailCause(this.getInterruptedCause()); } else { // Impossible to short circuit, so record the changes this.patchRuntimeFlags(this.currentRuntimeFlags, updateFlags); if (op.effect_instruction_i1) { // Since we updated the flags, we need to revert them const revertFlags = runtimeFlags_.diff(newRuntimeFlags, oldRuntimeFlags); this.pushStack(new core.RevertFlags(revertFlags, op)); return internalCall(() => op.effect_instruction_i1(oldRuntimeFlags)); } else { return core.exitVoid; } } } [OpCodes.OP_ON_SUCCESS](op) { this.pushStack(op); return op.effect_instruction_i0; } ["OnStep"](op) { this.pushStack(op); return op.effect_instruction_i0; } [OpCodes.OP_ON_FAILURE](op) { this.pushStack(op); return op.effect_instruction_i0; } [OpCodes.OP_ON_SUCCESS_AND_FAILURE](op) { this.pushStack(op); return op.effect_instruction_i0; } [OpCodes.OP_ASYNC](op) { this._asyncBlockingOn = op.effect_instruction_i1; this.initiateAsync(this.currentRuntimeFlags, op.effect_instruction_i0); yieldedOpChannel.currentOp = op; return YieldedOp; } [OpCodes.OP_YIELD](op) { this._isYielding = false; yieldedOpChannel.currentOp = op; return YieldedOp; } [OpCodes.OP_WHILE](op) { const check = op.effect_instruction_i0; const body = op.effect_instruction_i1; if (check()) { this.pushStack(op); return body(); } else { return core.exitVoid; } } [OpCodes.OP_ITERATOR](op) { return contOpSuccess[OpCodes.OP_ITERATOR](this, op, undefined); } [OpCodes.OP_COMMIT](op) { return internalCall(() => op.commit()); } /** * The main run-loop for evaluating effects. * * **NOTE**: This method must be invoked by the fiber itself. */ runLoop(effect0) { let cur = effect0; this.currentOpCount = 0; while (true) { if ((this.currentRuntimeFlags & OpSupervision) !== 0) { this.currentSupervisor.onEffect(this, cur); } if (this._queue.length > 0) { cur = this.drainQueueWhileRunning(this.currentRuntimeFlags, cur); } if (!this._isYielding) { this.currentOpCount += 1; const shouldYield = this.currentScheduler.shouldYield(this); if (shouldYield !== false) { this._isYielding = true; this.currentOpCount = 0; const oldCur = cur; cur = core.flatMap(core.yieldNow({ priority: shouldYield }), () => oldCur); } } try { // @ts-expect-error cur = this.currentTracer.context(() => { if (_version !== cur[core.EffectTypeId]._V) { return core.dieMessage(`Cannot execute an Effect versioned ${cur[core.EffectTypeId]._V} with a Runtime of version ${version.getCurrentVersion()}`); } // @ts-expect-error return this[cur._op](cur); }, this); if (cur === YieldedOp) { const op = yieldedOpChannel.currentOp; if (op._op === OpCodes.OP_YIELD || op._op === OpCodes.OP_ASYNC) { return YieldedOp; } yieldedOpChannel.currentOp = null; return op._op === OpCodes.OP_SUCCESS || op._op === OpCodes.OP_FAILURE ? op : core.exitFailCause(internalCause.die(op)); } } catch (e) { if (cur !== YieldedOp && !Predicate.hasProperty(cur, "_op") || !(cur._op in this)) { cur = core.dieMessage(`Not a valid effect: ${Inspectable.toStringUnknown(cur)}`); } else if (core.isInterruptedException(e)) { cur = core.exitFailCause(internalCause.sequential(internalCause.die(e), internalCause.interrupt(FiberId.none))); } else { cur = core.die(e); } } } } run = () => { this.drainQueueOnCurrentThread(); }; } // circular with Logger /** @internal */ export const currentMinimumLogLevel = /*#__PURE__*/globalValue("effect/FiberRef/currentMinimumLogLevel", () => core.fiberRefUnsafeMake(LogLevel.fromLiteral("Info"))); /** @internal */ export const loggerWithConsoleLog = self => internalLogger.makeLogger(opts => { const services = FiberRefs.getOrDefault(opts.context, defaultServices.currentServices); Context.get(services, consoleTag).unsafe.log(self.log(opts)); }); /** @internal */ export const loggerWithLeveledLog = self => internalLogger.makeLogger(opts => { const services = FiberRefs.getOrDefault(opts.context, defaultServices.currentServices); const unsafeLogger = Context.get(services, consoleTag).unsafe; switch (opts.logLevel._tag) { case "Debug": return unsafeLogger.debug(self.log(opts)); case "Info": return unsafeLogger.info(self.log(opts)); case "Trace": return unsafeLogger.trace(self.log(opts)); case "Warning": return unsafeLogger.warn(self.log(opts)); case "Error": case "Fatal": return unsafeLogger.error(self.log(opts)); default: return unsafeLogger.log(self.log(opts)); } }); /** @internal */ export const loggerWithConsoleError = self => internalLogger.makeLogger(opts => { const services = FiberRefs.getOrDefault(opts.context, defaultServices.currentServices); Context.get(services, consoleTag).unsafe.error(self.log(opts)); }); /** @internal */ export const defaultLogger = /*#__PURE__*/globalValue( /*#__PURE__*/Symbol.for("effect/Logger/defaultLogger"), () => loggerWithConsoleLog(internalLogger.stringLogger)); /** @internal */ export const jsonLogger = /*#__PURE__*/globalValue( /*#__PURE__*/Symbol.for("effect/Logger/jsonLogger"), () => loggerWithConsoleLog(internalLogger.jsonLogger)); /** @internal */ export const logFmtLogger = /*#__PURE__*/globalValue( /*#__PURE__*/Symbol.for("effect/Logger/logFmtLogger"), () => loggerWithConsoleLog(internalLogger.logfmtLogger)); /** @internal */ export const prettyLogger = /*#__PURE__*/globalValue( /*#__PURE__*/Symbol.for("effect/Logger/prettyLogger"), () => internalLogger.prettyLoggerDefault); /** @internal */ export const structuredLogger = /*#__PURE__*/globalValue( /*#__PURE__*/Symbol.for("effect/Logger/structuredLogger"), () => loggerWithConsoleLog(internalLogger.structuredLogger)); /** @internal */ export const tracerLogger = /*#__PURE__*/globalValue( /*#__PURE__*/Symbol.for("effect/Logger/tracerLogger"), () => internalLogger.makeLogger(({ annotations, cause, context, fiberId, logLevel, message }) => { const span = Context.getOption(fiberRefs.getOrDefault(context, core.currentContext), tracer.spanTag); if (span._tag === "None" || span.value._tag === "ExternalSpan") { return; } const clockService = Context.unsafeGet(fiberRefs.getOrDefault(context, defaultServices.currentServices), clock.clockTag); const attributes = {}; for (const [key, value] of annotations) { attributes[key] = value; } attributes["effect.fiberId"] = FiberId.threadName(fiberId); attributes["effect.logLevel"] = logLevel.label; if (cause !== null && cause._tag !== "Empty") { attributes["effect.cause"] = internalCause.pretty(cause, { renderErrorCause: true }); } span.value.event(Inspectable.toStringUnknown(Array.isArray(message) ? message[0] : message), clockService.unsafeCurrentTimeNanos(), attributes); })); /** @internal */ export const loggerWithSpanAnnotations = self => internalLogger.mapInputOptions(self, options => { const span = Option.flatMap(fiberRefs.get(options.context, core.currentContext), Context.getOption(tracer.spanTag)); if (span._tag === "None") { return options; } return { ...options, annotations: pipe(options.annotations, HashMap.set("effect.traceId", span.value.traceId), HashMap.set("effect.spanId", span.value.spanId), span.value._tag === "Span" ? HashMap.set("effect.spanName", span.value.name) : identity) }; }); /** @internal */ export const currentLoggers = /*#__PURE__*/globalValue( /*#__PURE__*/Symbol.for("effect/FiberRef/currentLoggers"), () => core.fiberRefUnsafeMakeHashSet(HashSet.make(defaultLogger, tracerLogger))); /** @internal */ export const batchedLogger = /*#__PURE__*/dual(3, (self, window, f) => core.flatMap(scope, scope => { let buffer = []; const flush = core.suspend(() => { if (buffer.length === 0) { return core.void; } const arr = buffer; buffer = []; return f(arr); }); return core.uninterruptibleMask(restore => pipe(internalEffect.sleep(window), core.zipRight(flush), internalEffect.forever, restore, forkDaemon, core.flatMap(fiber => core.scopeAddFinalizer(scope, core.interruptFiber(fiber))), core.zipRight(addFinalizer(() => flush)), core.as(internalLogger.makeLogger(options => { buffer.push(self.log(options)); })))); })); export const annotateLogsScoped = function () { if (typeof arguments[0] === "string") { return fiberRefLocallyScopedWith(core.currentLogAnnotations, HashMap.set(arguments[0], arguments[1])); } const entries = Object.entries(arguments[0]); return fiberRefLocallyScopedWith(core.currentLogAnnotations, HashMap.mutate(annotations => { for (let i = 0; i < entries.length; i++) { const [key, value] = entries[i]; HashMap.set(annotations, key, value); } return annotations; })); }; /** @internal */ export const whenLogLevel = /*#__PURE__*/dual(2, (effect, level) => { const requiredLogLevel = typeof level === "string" ? LogLevel.fromLiteral(level) : level; return core.withFiberRuntime(fiberState => { const minimumLogLevel = fiberState.getFiberRef(currentMinimumLogLevel); // Imitate the behaviour of `FiberRuntime.log` if (LogLevel.greaterThan(minimumLogLevel, requiredLogLevel)) { return core.succeed(Option.none()); } return core.map(effect, Option.some); }); }); // circular with Effect /* @internal */ export const acquireRelease = /*#__PURE__*/dual(args => core.isEffect(args[0]), (acquire, release) => core.uninterruptible(core.tap(acquire, a => addFinalizer(exit => release(a, exit))))); /* @internal */ export const acquireReleaseInterruptible = /*#__PURE__*/dual(args => core.isEffect(args[0]), (acquire, release) => ensuring(acquire, addFinalizer(exit => release(exit)))); /* @internal */ export const addFinalizer = finalizer => core.withFiberRuntime(runtime => { const acquireRefs = runtime.getFiberRefs(); const acquireFlags = runtime.currentRuntimeFlags; return core.flatMap(scope, scope => core.scopeAddFinalizerExit(scope, exit => core.withFiberRuntime(runtimeFinalizer => { const preRefs = runtimeFinalizer.getFiberRefs(); const preFlags = runtimeFinalizer.currentRuntimeFlags; const patchRefs = FiberRefsPatch.diff(preRefs, acquireRefs); const patchFlags = runtimeFlags_.diff(preFlags, acquireFlags); const inverseRefs = FiberRefsPatch.diff(acquireRefs, preRefs); runtimeFinalizer.setFiberRefs(FiberRefsPatch.patch(patchRefs, runtimeFinalizer.id(), acquireRefs)); return ensuring(core.withRuntimeFlags(finalizer(exit), patchFlags), core.sync(() => { runtimeFinalizer.setFiberRefs(FiberRefsPatch.patch(inverseRefs, runtimeFinalizer.id(), runtimeFinalizer.getFiberRefs())); })); }))); }); /* @internal */ export const daemonChildren = self => { const forkScope = core.fiberRefLocally(core.currentForkScopeOverride, Option.some(fiberScope.globalScope)); return forkScope(self); }; /** @internal */ const _existsParFound = /*#__PURE__*/Symbol.for("effect/Effect/existsPar/found"); /* @internal */ export const exists = /*#__PURE__*/dual(args => Predicate.isIterable(args[0]) && !core.isEffect(args[0]), (elements, predicate, options) => concurrency.matchSimple(options?.concurrency, () => core.suspend(() => existsLoop(elements[Symbol.iterator](), 0, predicate)), () => core.matchEffect(forEach(elements, (a, i) => core.if_(predicate(a, i), { onTrue: () => core.fail(_existsParFound), onFalse: () => core.void }), options), { onFailure: e => e === _existsParFound ? core.succeed(true) : core.fail(e), onSuccess: () => core.succeed(false) }))); const existsLoop = (iterator, index, f) => { const next = iterator.next(); if (next.done) { return core.succeed(false); } return pipe(core.flatMap(f(next.value, index), b => b ? core.succeed(b) : existsLoop(iterator, index + 1, f))); }; /* @internal */ export const filter = /*#__PURE__*/dual(args => Predicate.isIterable(args[0]) && !core.isEffect(args[0]), (elements, predicate, options) => { const predicate_ = options?.negate ? (a, i) => core.map(predicate(a, i), Boolean.not) : predicate; return concurrency.matchSimple(options?.concurrency, () => core.suspend(() => RA.fromIterable(elements).reduceRight((effect, a, i) => core.zipWith(effect, core.suspend(() => predicate_(a, i)), (list, b) => b ? [a, ...list] : list), core.sync(() => new Array()))), () => core.map(forEach(elements, (a, i) => core.map(predicate_(a, i), b => b ? Option.some(a) : Option.none()), options), RA.getSomes)); }); // === all const allResolveInput = input => { if (Array.isArray(input) || Predicate.isIterable(input)) { return [input, Option.none()]; } const keys = Object.keys(input); const size = keys.length; return [keys.map(k => input[k]), Option.some(values => { const res = {}; for (let i = 0; i < size; i++) { ; res[keys[i]] = values[i]; } return res; })]; }; const allValidate = (effects, reconcile, options) => { const eitherEffects = []; for (const effect of effects) { eitherEffects.push(core.either(effect)); } return core.flatMap(forEach(eitherEffects, identity, { concurrency: options?.concurrency, batching: options?.batching, concurrentFinalizers: options?.concurrentFinalizers }), eithers => { const none = Option.none(); const size = eithers.length; const errors = new Array(size); const successes = new Array(size); let errored = false; for (let i = 0; i < size; i++) { const either = eithers[i]; if (ei