@Test
public void mustBeAbleToRecover() throws Exception {
final ManualProbe<Integer> publisherProbe = TestPublisher.manualProbe(true, system);
final JavaTestKit probe = new JavaTestKit(system);
final Source<Integer, NotUsed> source =
Source.fromPublisher(publisherProbe)
.map(
elem -> {
if (elem == 1) throw new RuntimeException("ex");
else return elem;
})
.recover(new PFBuilder<Throwable, Integer>().matchAny(ex -> 0).build());
final CompletionStage<Done> future =
source.runWith(
Sink.foreach(elem -> probe.getRef().tell(elem, ActorRef.noSender())), materializer);
final PublisherProbeSubscription<Integer> s = publisherProbe.expectSubscription();
s.sendNext(0);
probe.expectMsgEquals(0);
s.sendNext(1);
probe.expectMsgEquals(0);
future.toCompletableFuture().get(200, TimeUnit.MILLISECONDS);
}
public void demonstrateADetachedGraphStage() throws Exception {
// tests:
CompletionStage<Integer> result1 =
Source.from(Arrays.asList(1, 2, 3))
.via(new TwoBuffer<>())
.runFold(0, (acc, n) -> acc + n, mat);
assertEquals(new Integer(6), result1.toCompletableFuture().get(3, TimeUnit.SECONDS));
TestSubscriber.ManualProbe<Integer> subscriber = TestSubscriber.manualProbe(system);
TestPublisher.Probe<Integer> publisher = TestPublisher.probe(0, system);
RunnableGraph<NotUsed> flow2 =
Source.fromPublisher(publisher).via(new TwoBuffer<>()).to(Sink.fromSubscriber(subscriber));
flow2.run(mat);
Subscription sub = subscriber.expectSubscription();
// this happens even though the subscriber has not signalled any demand
publisher.sendNext(1);
publisher.sendNext(2);
sub.cancel();
}
public static void main(String[] args) {
Outlet<Integer> outlet = null;
Outlet<Integer> outlet1 = null;
Outlet<Integer> outlet2 = null;
Inlet<Integer> inlet = null;
Inlet<Integer> inlet1 = null;
Inlet<Integer> inlet2 = null;
Flow<Integer, Integer, BoxedUnit> flow = Flow.of(Integer.class);
Flow<Integer, Integer, BoxedUnit> flow1 = Flow.of(Integer.class);
Flow<Integer, Integer, BoxedUnit> flow2 = Flow.of(Integer.class);
Promise<Option<Integer>> promise = null;
{
Graph<SourceShape<Integer>, BoxedUnit> graphSource = null;
Graph<SinkShape<Integer>, BoxedUnit> graphSink = null;
Graph<FlowShape<Integer, Integer>, BoxedUnit> graphFlow = null;
// #flow-wrap
Source<Integer, BoxedUnit> source = Source.fromGraph(graphSource);
Sink<Integer, BoxedUnit> sink = Sink.fromGraph(graphSink);
Flow<Integer, Integer, BoxedUnit> aflow = Flow.fromGraph(graphFlow);
Flow.fromSinkAndSource(Sink.<Integer>head(), Source.single(0));
Flow.fromSinkAndSourceMat(Sink.<Integer>head(), Source.single(0), Keep.left());
// #flow-wrap
Graph<BidiShape<Integer, Integer, Integer, Integer>, BoxedUnit> bidiGraph = null;
// #bidi-wrap
BidiFlow<Integer, Integer, Integer, Integer, BoxedUnit> bidiFlow =
BidiFlow.fromGraph(bidiGraph);
BidiFlow.fromFlows(flow1, flow2);
BidiFlow.fromFlowsMat(flow1, flow2, Keep.both());
// #bidi-wrap
}
{
// #graph-create
GraphDSL.create(
builder -> {
// ...
return ClosedShape.getInstance();
});
GraphDSL.create(
builder -> {
// ...
return new FlowShape<>(inlet, outlet);
});
// #graph-create
}
{
// #graph-create-2
GraphDSL.create(
builder -> {
// ...
return SourceShape.of(outlet);
});
GraphDSL.create(
builder -> {
// ...
return SinkShape.of(inlet);
});
GraphDSL.create(
builder -> {
// ...
return FlowShape.of(inlet, outlet);
});
GraphDSL.create(
builder -> {
// ...
return BidiShape.of(inlet1, outlet1, inlet2, outlet2);
});
// #graph-create-2
}
{
// #graph-builder
GraphDSL.create(
builder -> {
builder.from(outlet).toInlet(inlet);
builder.from(outlet).via(builder.add(flow)).toInlet(inlet);
builder.from(builder.add(Source.single(0))).to(builder.add(Sink.head()));
// ...
return ClosedShape.getInstance();
});
// #graph-builder
}
// #source-creators
Source<Integer, Promise<Option<Integer>>> src = Source.<Integer>maybe();
// Complete the promise with an empty option to emulate the old lazyEmpty
promise.trySuccess(scala.Option.empty());
final Source<String, Cancellable> ticks =
Source.tick(
FiniteDuration.create(0, TimeUnit.MILLISECONDS),
FiniteDuration.create(200, TimeUnit.MILLISECONDS),
"tick");
final Source<Integer, BoxedUnit> pubSource =
Source.fromPublisher(TestPublisher.<Integer>manualProbe(true, sys));
final Source<Integer, BoxedUnit> futSource = Source.fromFuture(Futures.successful(42));
final Source<Integer, Subscriber<Integer>> subSource = Source.<Integer>asSubscriber();
// #source-creators
// #sink-creators
final Sink<Integer, BoxedUnit> subSink =
Sink.fromSubscriber(TestSubscriber.<Integer>manualProbe(sys));
// #sink-creators
// #sink-as-publisher
final Sink<Integer, Publisher<Integer>> pubSink = Sink.<Integer>asPublisher(false);
final Sink<Integer, Publisher<Integer>> pubSinkFanout = Sink.<Integer>asPublisher(true);
// #sink-as-publisher
// #empty-flow
Flow<Integer, Integer, BoxedUnit> emptyFlow = Flow.<Integer>create();
// or
Flow<Integer, Integer, BoxedUnit> emptyFlow2 = Flow.of(Integer.class);
// #empty-flow
// #flatMapConcat
Flow.<Source<Integer, BoxedUnit>>create()
.<Integer, BoxedUnit>flatMapConcat(
new Function<Source<Integer, BoxedUnit>, Source<Integer, BoxedUnit>>() {
@Override
public Source<Integer, BoxedUnit> apply(Source<Integer, BoxedUnit> param)
throws Exception {
return param;
}
});
// #flatMapConcat
Uri uri = null;
// #raw-query
final akka.japi.Option<String> theRawQueryString = uri.rawQueryString();
// #raw-query
// #query-param
final akka.japi.Option<String> aQueryParam = uri.query().get("a");
// #query-param
// #file-source-sink
final Source<ByteString, Future<Long>> fileSrc = FileIO.fromFile(new File("."));
final Source<ByteString, Future<Long>> otherFileSrc = FileIO.fromFile(new File("."), 1024);
final Sink<ByteString, Future<Long>> fileSink = FileIO.toFile(new File("."));
// #file-source-sink
// #input-output-stream-source-sink
final Source<ByteString, Future<java.lang.Long>> inputStreamSrc =
StreamConverters.fromInputStream(
new Creator<InputStream>() {
public InputStream create() {
return new SomeInputStream();
}
});
final Source<ByteString, Future<java.lang.Long>> otherInputStreamSrc =
StreamConverters.fromInputStream(
new Creator<InputStream>() {
public InputStream create() {
return new SomeInputStream();
}
},
1024);
final Sink<ByteString, Future<java.lang.Long>> outputStreamSink =
StreamConverters.fromOutputStream(
new Creator<OutputStream>() {
public OutputStream create() {
return new SomeOutputStream();
}
});
// #input-output-stream-source-sink
// #output-input-stream-source-sink
final FiniteDuration timeout = FiniteDuration.Zero();
final Source<ByteString, OutputStream> outputStreamSrc = StreamConverters.asOutputStream();
final Source<ByteString, OutputStream> otherOutputStreamSrc =
StreamConverters.asOutputStream(timeout);
final Sink<ByteString, InputStream> someInputStreamSink = StreamConverters.asInputStream();
final Sink<ByteString, InputStream> someOtherInputStreamSink =
StreamConverters.asInputStream(timeout);
// #output-input-stream-source-sink
}