"use strict";
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.unsafeRemove = exports.unsafeOffer = exports.unsafeCompleteTakers = exports.unbounded = exports.takeUpTo = exports.takeN = exports.takeBetween = exports.takeAll = exports.take = exports.slidingStrategy = exports.sliding = exports.size = exports.shutdown = exports.poll = exports.offerAll = exports.offer = exports.make = exports.isShutdown = exports.isQueue = exports.isFull = exports.isEnqueue = exports.isEmpty = exports.isDequeue = exports.enqueueVariance = exports.droppingStrategy = exports.dropping = exports.dequeueVariance = exports.capacity = exports.bounded = exports.backingQueueFromMutableQueue = exports.backPressureStrategy = exports.awaitShutdown = exports.QueueStrategyTypeId = exports.EnqueueTypeId = exports.DequeueTypeId = exports.BackingQueueTypeId = exports.BackingQueueFromMutableQueue = void 0;
var Arr = _interopRequireWildcard(require("../Array.js"));
var Chunk = _interopRequireWildcard(require("../Chunk.js"));
var Effectable = _interopRequireWildcard(require("../Effectable.js"));
var _Function = require("../Function.js");
var MutableQueue = _interopRequireWildcard(require("../MutableQueue.js"));
var MutableRef = _interopRequireWildcard(require("../MutableRef.js"));
var Option = _interopRequireWildcard(require("../Option.js"));
var _Pipeable = require("../Pipeable.js");
var _Predicate = require("../Predicate.js");
var core = _interopRequireWildcard(require("./core.js"));
var fiberRuntime = _interopRequireWildcard(require("./fiberRuntime.js"));
function _interopRequireWildcard(e, t) { if ("function" == typeof WeakMap) var r = new WeakMap(), n = new WeakMap(); return (_interopRequireWildcard = function (e, t) { if (!t && e && e.__esModule) return e; var o, i, f = { __proto__: null, default: e }; if (null === e || "object" != typeof e && "function" != typeof e) return f; if (o = t ? n : r) { if (o.has(e)) return o.get(e); o.set(e, f); } for (const t in e) "default" !== t && {}.hasOwnProperty.call(e, t) && ((i = (o = Object.defineProperty) && Object.getOwnPropertyDescriptor(e, t)) && (i.get || i.set) ? o(f, t, i) : f[t] = e[t]); return f; })(e, t); }
/** @internal */
const EnqueueSymbolKey = "effect/QueueEnqueue";
/** @internal */
const EnqueueTypeId = exports.EnqueueTypeId = /*#__PURE__*/Symbol.for(EnqueueSymbolKey);
/** @internal */
const DequeueSymbolKey = "effect/QueueDequeue";
/** @internal */
const DequeueTypeId = exports.DequeueTypeId = /*#__PURE__*/Symbol.for(DequeueSymbolKey);
/** @internal */
const QueueStrategySymbolKey = "effect/QueueStrategy";
/** @internal */
const QueueStrategyTypeId = exports.QueueStrategyTypeId = /*#__PURE__*/Symbol.for(QueueStrategySymbolKey);
/** @internal */
const BackingQueueSymbolKey = "effect/BackingQueue";
/** @internal */
const BackingQueueTypeId = exports.BackingQueueTypeId = /*#__PURE__*/Symbol.for(BackingQueueSymbolKey);
const queueStrategyVariance = {
/* c8 ignore next */
_A: _ => _
};
const backingQueueVariance = {
/* c8 ignore next */
_A: _ => _
};
/** @internal */
const enqueueVariance = exports.enqueueVariance = {
/* c8 ignore next */
_In: _ => _
};
/** @internal */
const dequeueVariance = exports.dequeueVariance = {
/* c8 ignore next */
_Out: _ => _
};
/** @internal */
class QueueImpl extends Effectable.Class {
queue;
takers;
shutdownHook;
shutdownFlag;
strategy;
[EnqueueTypeId] = enqueueVariance;
[DequeueTypeId] = dequeueVariance;
constructor(/** @internal */
queue, /** @internal */
takers, /** @internal */
shutdownHook, /** @internal */
shutdownFlag, /** @internal */
strategy) {
super();
this.queue = queue;
this.takers = takers;
this.shutdownHook = shutdownHook;
this.shutdownFlag = shutdownFlag;
this.strategy = strategy;
}
pipe() {
return (0, _Pipeable.pipeArguments)(this, arguments);
}
commit() {
return this.take;
}
capacity() {
return this.queue.capacity();
}
get size() {
return core.suspend(() => core.catchAll(this.unsafeSize(), () => core.interrupt));
}
unsafeSize() {
if (MutableRef.get(this.shutdownFlag)) {
return Option.none();
}
return Option.some(this.queue.length() - MutableQueue.length(this.takers) + this.strategy.surplusSize());
}
get isEmpty() {
return core.map(this.size, size => size <= 0);
}
get isFull() {
return core.map(this.size, size => size >= this.capacity());
}
get shutdown() {
return core.uninterruptible(core.withFiberRuntime(state => {
(0, _Function.pipe)(this.shutdownFlag, MutableRef.set(true));
return (0, _Function.pipe)(fiberRuntime.forEachConcurrentDiscard(unsafePollAll(this.takers), d => core.deferredInterruptWith(d, state.id()), false, false), core.zipRight(this.strategy.shutdown), core.whenEffect(core.deferredSucceed(this.shutdownHook, void 0)), core.asVoid);
}));
}
get isShutdown() {
return core.sync(() => MutableRef.get(this.shutdownFlag));
}
get awaitShutdown() {
return core.deferredAwait(this.shutdownHook);
}
isActive() {
return !MutableRef.get(this.shutdownFlag);
}
unsafeOffer(value) {
if (MutableRef.get(this.shutdownFlag)) {
return false;
}
let noRemaining;
if (this.queue.length() === 0) {
const taker = (0, _Function.pipe)(this.takers, MutableQueue.poll(MutableQueue.EmptyMutableQueue));
if (taker !== MutableQueue.EmptyMutableQueue) {
unsafeCompleteDeferred(taker, value);
noRemaining = true;
} else {
noRemaining = false;
}
} else {
noRemaining = false;
}
if (noRemaining) {
return true;
}
// Not enough takers, offer to the queue
const succeeded = this.queue.offer(value);
unsafeCompleteTakers(this.strategy, this.queue, this.takers);
return succeeded;
}
offer(value) {
return core.suspend(() => {
if (MutableRef.get(this.shutdownFlag)) {
return core.interrupt;
}
let noRemaining;
if (this.queue.length() === 0) {
const taker = (0, _Function.pipe)(this.takers, MutableQueue.poll(MutableQueue.EmptyMutableQueue));
if (taker !== MutableQueue.EmptyMutableQueue) {
unsafeCompleteDeferred(taker, value);
noRemaining = true;
} else {
noRemaining = false;
}
} else {
noRemaining = false;
}
if (noRemaining) {
return core.succeed(true);
}
// Not enough takers, offer to the queue
const succeeded = this.queue.offer(value);
unsafeCompleteTakers(this.strategy, this.queue, this.takers);
return succeeded ? core.succeed(true) : this.strategy.handleSurplus([value], this.queue, this.takers, this.shutdownFlag);
});
}
offerAll(iterable) {
return core.suspend(() => {
if (MutableRef.get(this.shutdownFlag)) {
return core.interrupt;
}
const values = Arr.fromIterable(iterable);
const pTakers = this.queue.length() === 0 ? Arr.fromIterable(unsafePollN(this.takers, values.length)) : Arr.empty;
const [forTakers, remaining] = (0, _Function.pipe)(values, Arr.splitAt(pTakers.length));
for (let i = 0; i < pTakers.length; i++) {
const taker = pTakers[i];
const item = forTakers[i];
unsafeCompleteDeferred(taker, item);
}
if (remaining.length === 0) {
return core.succeed(true);
}
// Not enough takers, offer to the queue
const surplus = this.queue.offerAll(remaining);
unsafeCompleteTakers(this.strategy, this.queue, this.takers);
return Chunk.isEmpty(surplus) ? core.succeed(true) : this.strategy.handleSurplus(surplus, this.queue, this.takers, this.shutdownFlag);
});
}
get take() {
return core.withFiberRuntime(state => {
if (MutableRef.get(this.shutdownFlag)) {
return core.interrupt;
}
const item = this.queue.poll(MutableQueue.EmptyMutableQueue);
if (item !== MutableQueue.EmptyMutableQueue) {
this.strategy.unsafeOnQueueEmptySpace(this.queue, this.takers);
return core.succeed(item);
} else {
// Add the deferred to takers, then:
// - Try to take again in case a value was added since
// - Wait for the deferred to be completed
// - Clean up resources in case of interruption
const deferred = core.deferredUnsafeMake(state.id());
return (0, _Function.pipe)(core.suspend(() => {
(0, _Function.pipe)(this.takers, MutableQueue.offer(deferred));
unsafeCompleteTakers(this.strategy, this.queue, this.takers);
return MutableRef.get(this.shutdownFlag) ? core.interrupt : core.deferredAwait(deferred);
}), core.onInterrupt(() => {
return core.sync(() => unsafeRemove(this.takers, deferred));
}));
}
});
}
get takeAll() {
return core.suspend(() => {
return MutableRef.get(this.shutdownFlag) ? core.interrupt : core.sync(() => {
const values = this.queue.pollUpTo(Number.POSITIVE_INFINITY);
this.strategy.unsafeOnQueueEmptySpace(this.queue, this.takers);
return Chunk.fromIterable(values);
});
});
}
takeUpTo(max) {
return core.suspend(() => MutableRef.get(this.shutdownFlag) ? core.interrupt : core.sync(() => {
const values = this.queue.pollUpTo(max);
this.strategy.unsafeOnQueueEmptySpace(this.queue, this.takers);
return Chunk.fromIterable(values);
}));
}
takeBetween(min, max) {
return core.suspend(() => takeRemainderLoop(this, min, max, Chunk.empty()));
}
}
/** @internal */
const takeRemainderLoop = (self, min, max, acc) => {
if (max < min) {
return core.succeed(acc);
}
return (0, _Function.pipe)(takeUpTo(self, max), core.flatMap(bs => {
const remaining = min - bs.length;
if (remaining === 1) {
return (0, _Function.pipe)(take(self), core.map(b => (0, _Function.pipe)(acc, Chunk.appendAll(bs), Chunk.append(b))));
}
if (remaining > 1) {
return (0, _Function.pipe)(take(self), core.flatMap(b => takeRemainderLoop(self, remaining - 1, max - bs.length - 1, (0, _Function.pipe)(acc, Chunk.appendAll(bs), Chunk.append(b)))));
}
return core.succeed((0, _Function.pipe)(acc, Chunk.appendAll(bs)));
}));
};
/** @internal */
const isQueue = u => isEnqueue(u) && isDequeue(u);
/** @internal */
exports.isQueue = isQueue;
const isEnqueue = u => (0, _Predicate.hasProperty)(u, EnqueueTypeId);
/** @internal */
exports.isEnqueue = isEnqueue;
const isDequeue = u => (0, _Predicate.hasProperty)(u, DequeueTypeId);
/** @internal */
exports.isDequeue = isDequeue;
const bounded = requestedCapacity => (0, _Function.pipe)(core.sync(() => MutableQueue.bounded(requestedCapacity)), core.flatMap(queue => make(backingQueueFromMutableQueue(queue), backPressureStrategy())));
/** @internal */
exports.bounded = bounded;
const dropping = requestedCapacity => (0, _Function.pipe)(core.sync(() => MutableQueue.bounded(requestedCapacity)), core.flatMap(queue => make(backingQueueFromMutableQueue(queue), droppingStrategy())));
/** @internal */
exports.dropping = dropping;
const sliding = requestedCapacity => (0, _Function.pipe)(core.sync(() => MutableQueue.bounded(requestedCapacity)), core.flatMap(queue => make(backingQueueFromMutableQueue(queue), slidingStrategy())));
/** @internal */
exports.sliding = sliding;
const unbounded = () => (0, _Function.pipe)(core.sync(() => MutableQueue.unbounded()), core.flatMap(queue => make(backingQueueFromMutableQueue(queue), droppingStrategy())));
/** @internal */
exports.unbounded = unbounded;
const unsafeMake = (queue, takers, shutdownHook, shutdownFlag, strategy) => {
return new QueueImpl(queue, takers, shutdownHook, shutdownFlag, strategy);
};
/** @internal */
const make = (queue, strategy) => (0, _Function.pipe)(core.deferredMake(), core.map(deferred => unsafeMake(queue, MutableQueue.unbounded(), deferred, MutableRef.make(false), strategy)));
/** @internal */
exports.make = make;
class BackingQueueFromMutableQueue {
mutable;
[BackingQueueTypeId] = backingQueueVariance;
constructor(mutable) {
this.mutable = mutable;
}
poll(def) {
return MutableQueue.poll(this.mutable, def);
}
pollUpTo(limit) {
return MutableQueue.pollUpTo(this.mutable, limit);
}
offerAll(elements) {
return MutableQueue.offerAll(this.mutable, elements);
}
offer(element) {
return MutableQueue.offer(this.mutable, element);
}
capacity() {
return MutableQueue.capacity(this.mutable);
}
length() {
return MutableQueue.length(this.mutable);
}
}
/** @internal */
exports.BackingQueueFromMutableQueue = BackingQueueFromMutableQueue;
const backingQueueFromMutableQueue = mutable => new BackingQueueFromMutableQueue(mutable);
/** @internal */
exports.backingQueueFromMutableQueue = backingQueueFromMutableQueue;
const capacity = self => self.capacity();
/** @internal */
exports.capacity = capacity;
const size = self => self.size;
/** @internal */
exports.size = size;
const isFull = self => self.isFull;
/** @internal */
exports.isFull = isFull;
const isEmpty = self => self.isEmpty;
/** @internal */
exports.isEmpty = isEmpty;
const isShutdown = self => self.isShutdown;
/** @internal */
exports.isShutdown = isShutdown;
const awaitShutdown = self => self.awaitShutdown;
/** @internal */
exports.awaitShutdown = awaitShutdown;
const shutdown = self => self.shutdown;
/** @internal */
exports.shutdown = shutdown;
const offer = exports.offer = /*#__PURE__*/(0, _Function.dual)(2, (self, value) => self.offer(value));
/** @internal */
const unsafeOffer = exports.unsafeOffer = /*#__PURE__*/(0, _Function.dual)(2, (self, value) => self.unsafeOffer(value));
/** @internal */
const offerAll = exports.offerAll = /*#__PURE__*/(0, _Function.dual)(2, (self, iterable) => self.offerAll(iterable));
/** @internal */
const poll = self => core.map(self.takeUpTo(1), Chunk.head);
/** @internal */
exports.poll = poll;
const take = self => self.take;
/** @internal */
exports.take = take;
const takeAll = self => self.takeAll;
/** @internal */
exports.takeAll = takeAll;
const takeUpTo = exports.takeUpTo = /*#__PURE__*/(0, _Function.dual)(2, (self, max) => self.takeUpTo(max));
/** @internal */
const takeBetween = exports.takeBetween = /*#__PURE__*/(0, _Function.dual)(3, (self, min, max) => self.takeBetween(min, max));
/** @internal */
const takeN = exports.takeN = /*#__PURE__*/(0, _Function.dual)(2, (self, n) => self.takeBetween(n, n));
// -----------------------------------------------------------------------------
// Strategy
// -----------------------------------------------------------------------------
/** @internal */
const backPressureStrategy = () => new BackPressureStrategy();
/** @internal */
exports.backPressureStrategy = backPressureStrategy;
const droppingStrategy = () => new DroppingStrategy();
/** @internal */
exports.droppingStrategy = droppingStrategy;
const slidingStrategy = () => new SlidingStrategy();
/** @internal */
exports.slidingStrategy = slidingStrategy;
class BackPressureStrategy {
[QueueStrategyTypeId] = queueStrategyVariance;
putters = /*#__PURE__*/MutableQueue.unbounded();
surplusSize() {
return MutableQueue.length(this.putters);
}
onCompleteTakersWithEmptyQueue(takers) {
while (!MutableQueue.isEmpty(this.putters) && !MutableQueue.isEmpty(takers)) {
const taker = MutableQueue.poll(takers, void 0);
const putter = MutableQueue.poll(this.putters, void 0);
if (putter[2]) {
unsafeCompleteDeferred(putter[1], true);
}
unsafeCompleteDeferred(taker, putter[0]);
}
}
get shutdown() {
return (0, _Function.pipe)(core.fiberId, core.flatMap(fiberId => (0, _Function.pipe)(core.sync(() => unsafePollAll(this.putters)), core.flatMap(putters => fiberRuntime.forEachConcurrentDiscard(putters, ([_, deferred, isLastItem]) => isLastItem ? (0, _Function.pipe)(core.deferredInterruptWith(deferred, fiberId), core.asVoid) : core.void, false, false)))));
}
handleSurplus(iterable, queue, takers, isShutdown) {
return core.withFiberRuntime(state => {
const deferred = core.deferredUnsafeMake(state.id());
return (0, _Function.pipe)(core.suspend(() => {
this.unsafeOffer(iterable, deferred);
this.unsafeOnQueueEmptySpace(queue, takers);
unsafeCompleteTakers(this, queue, takers);
return MutableRef.get(isShutdown) ? core.interrupt : core.deferredAwait(deferred);
}), core.onInterrupt(() => core.sync(() => this.unsafeRemove(deferred))));
});
}
unsafeOnQueueEmptySpace(queue, takers) {
let keepPolling = true;
while (keepPolling && (queue.capacity() === Number.POSITIVE_INFINITY || queue.length() < queue.capacity())) {
const putter = (0, _Function.pipe)(this.putters, MutableQueue.poll(MutableQueue.EmptyMutableQueue));
if (putter === MutableQueue.EmptyMutableQueue) {
keepPolling = false;
} else {
const offered = queue.offer(putter[0]);
if (offered && putter[2]) {
unsafeCompleteDeferred(putter[1], true);
} else if (!offered) {
unsafeOfferAll(this.putters, (0, _Function.pipe)(unsafePollAll(this.putters), Chunk.prepend(putter)));
}
unsafeCompleteTakers(this, queue, takers);
}
}
}
unsafeOffer(iterable, deferred) {
const stuff = Arr.fromIterable(iterable);
for (let i = 0; i < stuff.length; i++) {
const value = stuff[i];
if (i === stuff.length - 1) {
(0, _Function.pipe)(this.putters, MutableQueue.offer([value, deferred, true]));
} else {
(0, _Function.pipe)(this.putters, MutableQueue.offer([value, deferred, false]));
}
}
}
unsafeRemove(deferred) {
unsafeOfferAll(this.putters, (0, _Function.pipe)(unsafePollAll(this.putters), Chunk.filter(([, _]) => _ !== deferred)));
}
}
/** @internal */
class DroppingStrategy {
[QueueStrategyTypeId] = queueStrategyVariance;
surplusSize() {
return 0;
}
get shutdown() {
return core.void;
}
onCompleteTakersWithEmptyQueue() {}
handleSurplus(_iterable, _queue, _takers, _isShutdown) {
return core.succeed(false);
}
unsafeOnQueueEmptySpace(_queue, _takers) {
//
}
}
/** @internal */
class SlidingStrategy {
[QueueStrategyTypeId] = queueStrategyVariance;
surplusSize() {
return 0;
}
get shutdown() {
return core.void;
}
onCompleteTakersWithEmptyQueue() {}
handleSurplus(iterable, queue, takers, _isShutdown) {
return core.sync(() => {
this.unsafeOffer(queue, iterable);
unsafeCompleteTakers(this, queue, takers);
return true;
});
}
unsafeOnQueueEmptySpace(_queue, _takers) {
//
}
unsafeOffer(queue, iterable) {
const iterator = iterable[Symbol.iterator]();
let next;
let offering = true;
while (!(next = iterator.next()).done && offering) {
if (queue.capacity() === 0) {
return;
}
// Poll 1 and retry
queue.poll(MutableQueue.EmptyMutableQueue);
offering = queue.offer(next.value);
}
}
}
/** @internal */
const unsafeCompleteDeferred = (deferred, a) => {
return core.deferredUnsafeDone(deferred, core.succeed(a));
};
/** @internal */
const unsafeOfferAll = (queue, as) => {
return (0, _Function.pipe)(queue, MutableQueue.offerAll(as));
};
/** @internal */
const unsafePollAll = queue => {
return (0, _Function.pipe)(queue, MutableQueue.pollUpTo(Number.POSITIVE_INFINITY));
};
/** @internal */
const unsafePollN = (queue, max) => {
return (0, _Function.pipe)(queue, MutableQueue.pollUpTo(max));
};
/** @internal */
const unsafeRemove = (queue, a) => {
unsafeOfferAll(queue, (0, _Function.pipe)(unsafePollAll(queue), Chunk.filter(b => a !== b)));
};
/** @internal */
exports.unsafeRemove = unsafeRemove;
const unsafeCompleteTakers = (strategy, queue, takers) => {
// Check both a taker and an item are in the queue, starting with the taker
let keepPolling = true;
while (keepPolling && queue.length() !== 0) {
const taker = (0, _Function.pipe)(takers, MutableQueue.poll(MutableQueue.EmptyMutableQueue));
if (taker !== MutableQueue.EmptyMutableQueue) {
const element = queue.poll(MutableQueue.EmptyMutableQueue);
if (element !== MutableQueue.EmptyMutableQueue) {
unsafeCompleteDeferred(taker, element);
strategy.unsafeOnQueueEmptySpace(queue, takers);
} else {
unsafeOfferAll(takers, (0, _Function.pipe)(unsafePollAll(takers), Chunk.prepend(taker)));
}
keepPolling = true;
} else {
keepPolling = false;
}
}
if (keepPolling && queue.length() === 0 && !MutableQueue.isEmpty(takers)) {
strategy.onCompleteTakersWithEmptyQueue(takers);
}
};
exports.unsafeCompleteTakers = unsafeCompleteTakers;
//# sourceMappingURL=queue.js.map