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

Add combinedMat method to Source (#23809)

Browse source 
* Add combinedMat method to Source

* Fix formatting
This commit is contained in:
Richard Imaoka 2017-11-02 10:34:40 +09:00 • committed by Konrad `ktoso` Malawski
parent 0988933fac
commit a50df1c575
4 changed files with 87 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

24
akka-stream-tests/src/test/java/akka/stream/javadsl/SourceTest.java
View file

@ -11,6 +11,8 @@ import akka.japi.Pair;
import akka.japi.function.*;
import akka.japi.pf.PFBuilder;
import akka.stream.*;
import akka.stream.testkit.TestSubscriber;
import akka.stream.testkit.javadsl.TestSink;
import akka.util.ConstantFun;
import akka.stream.stage.*;
import akka.testkit.AkkaSpec;
@ -655,6 +657,28 @@ public class SourceTest extends StreamTest {
future.toCompletableFuture().get(3, TimeUnit.SECONDS);
}
@Test
public void mustBeAbleToCombineMat() throws Exception {
final TestKit probe = new TestKit(system);
final Source<Integer, SourceQueueWithComplete<Integer>> source1 = Source.queue(1, OverflowStrategy.dropNew());
final Source<Integer, NotUsed> source2 = Source.from(Arrays.asList(2, 3));
// compiler to check the correct materialized value of type = SourceQueueWithComplete<Integer> available
final Source<Integer, SourceQueueWithComplete<Integer>> combined = Source.combineMat(
source1, source2, width -> Concat.<Integer> create(width), Keep.left()); //Keep.left() (i.e. preserve queueSource's materialized value)
SourceQueueWithComplete<Integer> queue = combined
.toMat(Sink.foreach(elem -> probe.getRef().tell(elem, ActorRef.noSender())), Keep.left())
.run(materializer);
queue.offer(0);
queue.offer(1);
queue.complete(); //complete queueSource so that combined with `Concat` pulls elements from queueSource
// elements from source1 (i.e. first of combined source) come first, then source2 elements, due to `Concat`
probe.expectMsgAllOf(0, 1, 2, 3);
}
@Test
public void mustBeAbleToZipN() throws Exception {
final TestKit probe = new TestKit(system);