95 lines
6.5 KiB
ReStructuredText
95 lines
6.5 KiB
ReStructuredText
.. _stream-io-java:
|
||
|
||
#########################
|
||
Working with streaming IO
|
||
#########################
|
||
|
||
Akka Streams provides a way of handling TCP connections with Streams.
|
||
While the general approach is very similar to the `Actor based TCP handling`_ using Akka IO,
|
||
by using Akka Streams you are freed of having to manually react to back-pressure signals,
|
||
as the library does it transparently for you.
|
||
|
||
.. _Actor based TCP handling: http://doc.akka.io/docs/akka/current/java/io-tcp.html
|
||
|
||
Accepting connections: Echo Server
|
||
==================================
|
||
In order to implement a simple EchoServer we ``bind`` to a given address, which returns a ``Source[IncomingConnection]``,
|
||
which will emit an :class:`IncomingConnection` element for each new connection that the Server should handle:
|
||
|
||
.. includecode:: ../../../akka-samples/akka-docs-java-lambda/src/test/java/docs/stream/StreamTcpDocTest.java#echo-server-simple-bind
|
||
|
||
Next, we simply handle *each* incoming connection using a :class:`Flow` which will be used as the processing stage
|
||
to handle and emit ByteStrings from and to the TCP Socket. Since one :class:`ByteString` does not have to necessarily
|
||
correspond to exactly one line of text (the client might be sending the line in chunks) we use the ``parseLines``
|
||
recipe from the :ref:`cookbook-parse-lines-java` Akka Streams Cookbook recipe to chunk the inputs up into actual lines of text.
|
||
In this example we simply add exclamation marks to each incoming text message and push it through the flow:
|
||
|
||
.. includecode:: ../../../akka-samples/akka-docs-java-lambda/src/test/java/docs/stream/StreamTcpDocTest.java#echo-server-simple-handle
|
||
|
||
Notice that while most building blocks in Akka Streams are reusable and freely shareable, this is *not* the case for the
|
||
incoming connection Flow, since it directly corresponds to an existing, already accepted connection its handling can
|
||
only ever be materialized *once*.
|
||
|
||
Closing connections is possible by cancelling the *incoming connection* :class:`Flow` from your server logic (e.g. by
|
||
connecting its downstream to an :class:`CancelledSink` and its upstream to a *completed* :class:`Source`).
|
||
It is also possible to shut down the servers socket by cancelling the ``connections:Source[IncomingConnection]``.
|
||
|
||
We can then test the TCP server by sending data to the TCP Socket using ``netcat``:
|
||
|
||
::
|
||
|
||
$ echo -n "Hello World" | netcat 127.0.0.1 8889
|
||
Hello World!!!
|
||
|
||
Connecting: REPL Client
|
||
=======================
|
||
In this example we implement a rather naive Read Evaluate Print Loop client over TCP.
|
||
Let's say we know a server has exposed a simple command line interface over TCP,
|
||
and would like to interact with it using Akka Streams over TCP. To open an outgoing connection socket we use
|
||
the ``outgoingConnection`` method:
|
||
|
||
.. includecode:: ../../../akka-samples/akka-docs-java-lambda/src/test/java/docs/stream/StreamTcpDocTest.java#repl-client
|
||
|
||
The ``repl`` flow we use to handle the server interaction first prints the servers response, then awaits on input from
|
||
the command line (this blocking call is used here just for the sake of simplicity) and converts it to a
|
||
:class:`ByteString` which is then sent over the wire to the server. Then we simply connect the TCP pipeline to this
|
||
processing stage–at this point it will be materialized and start processing data once the server responds with
|
||
an *initial message*.
|
||
|
||
A resilient REPL client would be more sophisticated than this, for example it should split out the input reading into
|
||
a separate mapAsync step and have a way to let the server write more data than one ByteString chunk at any given time,
|
||
these improvements however are left as exercise for the reader.
|
||
|
||
Avoiding deadlocks and liveness issues in back-pressured cycles
|
||
===============================================================
|
||
When writing such end-to-end back-pressured systems you may sometimes end up in a situation of a loop,
|
||
in which *either side is waiting for the other one to start the conversation*. One does not need to look far
|
||
to find examples of such back-pressure loops. In the two examples shown previously, we always assumed that the side we
|
||
are connecting to would start the conversation, which effectively means both sides are back-pressured and can not get
|
||
the conversation started. There are multiple ways of dealing with this which are explained in depth in :ref:`graph-cycles-java`,
|
||
however in client-server scenarios it is often the simplest to make either side simply send an initial message.
|
||
|
||
.. note::
|
||
In case of back-pressured cycles (which can occur even between different systems) sometimes you have to decide
|
||
which of the sides has start the conversation in order to kick it off. This can be often done by injecting an
|
||
initial message from one of the sides–a conversation starter.
|
||
|
||
To break this back-pressure cycle we need to inject some initial message, a "conversation starter".
|
||
First, we need to decide which side of the connection should remain passive and which active.
|
||
Thankfully in most situations finding the right spot to start the conversation is rather simple, as it often is inherent
|
||
to the protocol we are trying to implement using Streams. In chat-like applications, which our examples resemble,
|
||
it makes sense to make the Server initiate the conversation by emitting a "hello" message:
|
||
|
||
.. includecode:: ../../../akka-samples/akka-docs-java-lambda/src/test/java/docs/stream/StreamTcpDocTest.java#welcome-banner-chat-server
|
||
|
||
The way we constructed a :class:`Flow` using a :class:`PartialFlowGraph` is explained in detail in
|
||
:ref:`constructing-sources-sinks-flows-from-partial-graphs-java`, however the basic concepts is rather simple–
|
||
we can encapsulate arbitrarily complex logic within a :class:`Flow` as long as it exposes the same interface, which means
|
||
exposing exactly one :class:`UndefinedSink` and exactly one :class:`UndefinedSource` which will be connected to the TCP
|
||
pipeline. In this example we use a :class:`Concat` graph processing stage to inject the initial message, and then
|
||
continue with handling all incoming data using the echo handler. You should use this pattern of encapsulating complex
|
||
logic in Flows and attaching those to :class:`StreamIO` in order to implement your custom and possibly sophisticated TCP servers.
|
||
|
||
In this example both client and server may need to close the stream based on a parsed command command - ``BYE`` in the case
|
||
of the server, and ``q`` in the case of the client. This is implemented by using a custom :class:`PushStage`
|
||
(see :ref:`stream-using-push-pull-stage-java`) which completes the stream once it encounters such command.
|