Skip to content

Commit

Permalink
Support primitives in Flow#collectType (#1490)
Browse files Browse the repository at this point in the history
  • Loading branch information
JD557 authored Sep 20, 2024
1 parent 22a7385 commit 6ceb59b
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ class FlowCollectTypeSpec extends StreamSpec {

"A CollectType" must {

"collectType" in {
"collectType with references" in {
val fruit = Source(List(Orange, Apple, Apple, Orange))

val apples = fruit.collectType[Apple].runWith(Sink.seq).futureValue
Expand All @@ -36,6 +36,17 @@ class FlowCollectTypeSpec extends StreamSpec {
all should equal(List(Orange, Apple, Apple, Orange))
}

"collectType with primitives" in {
val numbers = Source(List[Int](1, 2, 3) ++ List[Double](1.5))

val integers = numbers.collectType[Int].runWith(Sink.seq).futureValue
integers should equal(List(1, 2, 3))
val doubles = numbers.collectType[Double].runWith(Sink.seq).futureValue
doubles should equal(List(1.5))
val all = numbers.collectType[Any].runWith(Sink.seq).futureValue
all should equal(List(1, 2, 3, 1.5))
}

}

}
Original file line number Diff line number Diff line change
Expand Up @@ -1686,7 +1686,7 @@ trait FlowOps[+Out, +Mat] {
* '''Cancels when''' downstream cancels
*/
def collectType[T](implicit tag: ClassTag[T]): Repr[T] =
collect { case c if tag.runtimeClass.isInstance(c) => c.asInstanceOf[T] }
collect { case tag(c) => c }

/**
* Chunk up this stream into groups of the given size, with the last group
Expand Down

0 comments on commit 6ceb59b

Please sign in to comment.