@effect-ts/system/_src/Experimental/Stream/_internal/api/toQueueOfElements.ts

Effect-TS is a zero dependency set of libraries to write highly productive, purely functional TypeScript at scale.

// ets_tracing: off

import * as T from "../../../../Effect/index.js"
import type * as Ex from "../../../../Exit/index.js"
import { pipe } from "../../../../Function/index.js"
import * as M from "../../../../Managed/index.js"
import type * as O from "../../../../Option/index.js"
import * as Q from "../../../../Queue/index.js"
import type * as C from "../core.js"
import * as RunIntoElementsManaged from "./runIntoElementsManaged.js"

/**
 * Converts the stream to a managed queue of elements. After the managed queue is used,
 * the queue will never again produce values and should be discarded.
 */
export function toQueueOfElements_<R, E, A>(
  self: C.Stream<R, E, A>,
  capacity = 2
): M.RIO<R, Q.Queue<Ex.Exit<O.Option<E>, A>>> {
  return pipe(
    M.do,
    M.bind("queue", () =>
      T.toManagedRelease_(Q.makeBounded<Ex.Exit<O.Option<E>, A>>(capacity), Q.shutdown)
    ),
    M.tap(({ queue }) =>
      M.fork(RunIntoElementsManaged.runIntoElementsManaged_(self, queue))
    ),
    M.map(({ queue }) => queue)
  )
}

/**
 * Converts the stream to a managed queue of elements. After the managed queue is used,
 * the queue will never again produce values and should be discarded.
 *
 * @ets_data_first toQueueOfElements_
 */
export function toQueueOfElements(capacity = 2) {
  return <R, E, A>(self: C.Stream<R, E, A>) => toQueueOfElements_(self, capacity)
}