@effect-ts/system
Version:
Effect-TS is a zero dependency set of libraries to write highly productive, purely functional TypeScript at scale.
47 lines (42 loc) • 2.12 kB
JavaScript
// ets_tracing: off
import * as CL from "../../../../Clock/index.mjs";
import * as T from "../../../../Effect/index.mjs";
import * as CH from "../../Channel/index.mjs";
import * as C from "../core.mjs";
/**
* Throttles the chunks of this stream according to the given bandwidth parameters using the token bucket
* algorithm. Allows for burst in the processing of elements by allowing the token bucket to accumulate
* tokens up to a `units + burst` threshold. Chunks that do not meet the bandwidth constraints are dropped.
* The weight of each chunk is determined by the `costFn` effectful function.
*/
export function throttleEnforceEffect_(self, units, duration, costFn, burst = 0) {
const loop = (tokens, timestamp) => CH.readWith(in_ => CH.unwrap(T.map_(T.zip_(costFn(in_), CL.currentTime), ({
tuple: [weight, current]
}) => {
const elapsed = current - timestamp;
const cycles = elapsed / duration;
const available = (() => {
const sum = Math.floor(tokens + cycles * units);
const max = units + burst < 0 ? Number.MAX_SAFE_INTEGER : units + burst;
return sum < 0 ? max : Math.min(sum, max);
})();
if (weight <= available) {
return CH.zipRight_(CH.write(in_), loop(available - weight, current));
} else {
return loop(available, current);
}
})), e => CH.fail(e), _ => CH.unit);
return new C.Stream(CH.chain_(CH.fromEffect(CL.currentTime), _ => self.channel[">>>"](loop(units, _))));
}
/**
* Throttles the chunks of this stream according to the given bandwidth parameters using the token bucket
* algorithm. Allows for burst in the processing of elements by allowing the token bucket to accumulate
* tokens up to a `units + burst` threshold. Chunks that do not meet the bandwidth constraints are dropped.
* The weight of each chunk is determined by the `costFn` effectful function.
*
* @ets_data_first throttleEnforceEffect_
*/
export function throttleEnforceEffect(units, duration, costFn, burst = 0) {
return self => throttleEnforceEffect_(self, units, duration, costFn, burst);
}
//# sourceMappingURL=throttleEnforceEffect.mjs.map