raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 10

chore: Add Flow#flattenMerge operator. (#1045)

Browse source
This commit is contained in:
He-Pin(kerr) 2024-01-27 18:03:23 +08:00 • committed by GitHub
parent 8e7ae52a06
commit 4c20580ff2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 59 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

20
stream/src/main/scala/org/apache/pekko/stream/scaladsl/Flow.scala
View file

@ -2667,7 +2667,7 @@ trait FlowOps[+Out, +Mat] {
/**
* Flattens a stream of `Source` into a single output stream by concatenation,
* fully consuming one `Source` after the other. This function is qquivalent to <code>flatMapConcat(identity)</code>.
* fully consuming one `Source` after the other. This function is equivalent to <code>flatMapConcat(identity)</code>.
*
* '''Emits when''' a currently consumed substream has an element available
*
@ -2697,6 +2697,24 @@ trait FlowOps[+Out, +Mat] {
def flatMapMerge[T, M](breadth: Int, f: Out => Graph[SourceShape[T], M]): Repr[T] =
map(f).via(new FlattenMerge[T, M](breadth))
/**
* Flattens a stream of `Source` into a single output stream by merging,
* where at most `breadth` substreams are being consumed at any given time.
* This function is equivalent to <code>flatMapMerge(breadth, identity)</code>.
*
* '''Emits when''' a currently consumed substream has an element available
*
* '''Backpressures when''' downstream backpressures
*
* '''Completes when''' upstream completes and all consumed substreams complete
*
* '''Cancels when''' downstream cancels
*
* @since 1.1.0
*/
def flattenMerge[T, M](breadth: Int)(implicit ev: Out <:< Graph[SourceShape[T], M]): Repr[T] =
flatMapMerge(breadth, ev)
/**
* If the first element has not passed through this operator before the provided timeout, the stream is failed
* with a [[org.apache.pekko.stream.InitialTimeoutException]].