Buffers and working with rate

When upstream and downstream rates differ, especially when the throughput has spikes, it can be useful to introduce buffers in a stream. In this chapter we cover how buffers are used in Akka Streams.

Buffers for asynchronous stages

In this section we will discuss internal buffers that are introduced as an optimization when using asynchronous stages.

To run a stage asynchronously it has to be marked explicitly as such using the .async.async() method. Being run asynchronously means that a stage, after handing out an element to its downstream consumer is able to immediately process the next message. To demonstrate what we mean by this, let’s take a look at the following example:

Scala
Source(1 to 3)
  .map { i ⇒ println(s"A: $i"); i }.async
  .map { i ⇒ println(s"B: $i"); i }.async
  .map { i ⇒ println(s"C: $i"); i }.async
  .runWith(Sink.ignore)
Java
Source.from(Arrays.asList(1, 2, 3))
  .map(i -> {System.out.println("A: " + i); return i;}).async()
  .map(i -> {System.out.println("B: " + i); return i;}).async()
  .map(i -> {System.out.println("C: " + i); return i;}).async()
  .runWith(Sink.ignore(), mat);

Running the above example, one of the possible outputs looks like this:

A: 1
A: 2
B: 1
A: 3
B: 2
C: 1
B: 3
C: 2
C: 3

Note that the order is not A:1, B:1, C:1, A:2, B:2, C:2, which would correspond to the normal fused synchronous execution model of flows where an element completely passes through the processing pipeline before the next element enters the flow. The next element is processed by an asynchronous stage as soon as it is emitted the previous one.

While pipelining in general increases throughput, in practice there is a cost of passing an element through the asynchronous (and therefore thread crossing) boundary which is significant. To amortize this cost Akka Streams uses a windowed, batching backpressure strategy internally. It is windowed because as opposed to a Stop-And-Wait protocol multiple elements might be “in-flight” concurrently with requests for elements. It is also batching because a new element is not immediately requested once an element has been drained from the window-buffer but multiple elements are requested after multiple elements have been drained. This batching strategy reduces the communication cost of propagating the backpressure signal through the asynchronous boundary.

While this internal protocol is mostly invisible to the user (apart form its throughput increasing effects) there are situations when these details get exposed. In all of our previous examples we always assumed that the rate of the processing chain is strictly coordinated through the backpressure signal causing all stages to process no faster than the throughput of the connected chain. There are tools in Akka Streams however that enable the rates of different segments of a processing chain to be “detached” or to define the maximum throughput of the stream through external timing sources. These situations are exactly those where the internal batching buffering strategy suddenly becomes non-transparent.

Internal buffers and their effect

As we have explained, for performance reasons Akka Streams introduces a buffer for every asynchronous processing stage. The purpose of these buffers is solely optimization, in fact the size of 1 would be the most natural choice if there would be no need for throughput improvements. Therefore it is recommended to keep these buffer sizes small, and increase them only to a level suitable for the throughput requirements of the application. Default buffer sizes can be set through configuration:

akka.stream.materializer.max-input-buffer-size = 16

Alternatively they can be set by passing a ActorMaterializerSettings to the materializer:

Scala
val materializer = ActorMaterializer(
  ActorMaterializerSettings(system)
    .withInputBuffer(
      initialSize = 64,
      maxSize = 64))
Java
final Materializer materializer = ActorMaterializer.create(
  ActorMaterializerSettings.create(system)
    .withInputBuffer(64, 64), system);

If the buffer size needs to be set for segments of a Flow only, it is possible by defining a separate Flow with these attributes:

Scala
val section = Flow[Int].map(_ * 2).async
  .addAttributes(Attributes.inputBuffer(initial = 1, max = 1)) // the buffer size of this map is 1
val flow = section.via(Flow[Int].map(_ / 2)).async // the buffer size of this map is the default
Java
final Flow<Integer, Integer, NotUsed> flow1 =
  Flow.of(Integer.class)
  .map(elem -> elem * 2).async()
  .addAttributes(Attributes.inputBuffer(1, 1)); // the buffer size of this map is 1
final Flow<Integer, Integer, NotUsed> flow2 =
  flow1.via(
    Flow.of(Integer.class)
    .map(elem -> elem / 2)).async(); // the buffer size of this map is the default

Here is an example of a code that demonstrate some of the issues caused by internal buffers:

Scala
import scala.concurrent.duration._
case class Tick()

RunnableGraph.fromGraph(GraphDSL.create() { implicit b ⇒
  import GraphDSL.Implicits._

  // this is the asynchronous stage in this graph
  val zipper = b.add(ZipWith[Tick, Int, Int]((tick, count) ⇒ count).async)

  Source.tick(initialDelay = 3.second, interval = 3.second, Tick()) ~> zipper.in0

  Source.tick(initialDelay = 1.second, interval = 1.second, "message!")
    .conflateWithSeed(seed = (_) ⇒ 1)((count, _) ⇒ count + 1) ~> zipper.in1

  zipper.out ~> Sink.foreach(println)
  ClosedShape
})
Java
final FiniteDuration oneSecond =
    FiniteDuration.create(1, TimeUnit.SECONDS);
final Source<String, Cancellable> msgSource =
    Source.tick(oneSecond, oneSecond, "message!");
final Source<String, Cancellable> tickSource =
    Source.tick(oneSecond.mul(3), oneSecond.mul(3), "tick");
final Flow<String, Integer, NotUsed> conflate =
    Flow.of(String.class).conflateWithSeed(
        first -> 1, (count, elem) -> count + 1);

RunnableGraph.fromGraph(GraphDSL.create(b -> {
  // this is the asynchronous stage in this graph
  final FanInShape2<String, Integer, Integer> zipper =
      b.add(ZipWith.create((String tick, Integer count) -> count).async());
  b.from(b.add(msgSource)).via(b.add(conflate)).toInlet(zipper.in1());
  b.from(b.add(tickSource)).toInlet(zipper.in0());
  b.from(zipper.out()).to(b.add(Sink.foreach(elem -> System.out.println(elem))));
  return ClosedShape.getInstance();
})).run(mat);

Running the above example one would expect the number 3 to be printed in every 3 seconds (the conflateWithSeed step here is configured so that it counts the number of elements received before the downstream ZipWith consumes them). What is being printed is different though, we will see the number 1. The reason for this is the internal buffer which is by default 16 elements large, and prefetches elements before the ZipWith starts consuming them. It is possible to fix this issue by changing the buffer size of ZipWith (or the whole graph) to 1. We will still see a leading 1 though which is caused by an initial prefetch of the ZipWith element.

Note

In general, when time or rate driven processing stages exhibit strange behavior, one of the first solutions to try should be to decrease the input buffer of the affected elements to 1.

Buffers in Akka Streams

In this section we will discuss explicit user defined buffers that are part of the domain logic of the stream processing pipeline of an application.

The example below will ensure that 1000 jobs (but not more) are dequeued from an external (imaginary) system and stored locally in memory - relieving the external system:

Scala
// Getting a stream of jobs from an imaginary external system as a Source
val jobs: Source[Job, NotUsed] = inboundJobsConnector()
jobs.buffer(1000, OverflowStrategy.backpressure)
Java
// Getting a stream of jobs from an imaginary external system as a Source
final Source<Job, NotUsed> jobs = inboundJobsConnector;
jobs.buffer(1000, OverflowStrategy.backpressure());

The next example will also queue up 1000 jobs locally, but if there are more jobs waiting in the imaginary external systems, it makes space for the new element by dropping one element from the tail of the buffer. Dropping from the tail is a very common strategy but it must be noted that this will drop the youngest waiting job. If some “fairness” is desired in the sense that we want to be nice to jobs that has been waiting for long, then this option can be useful.

Scala
jobs.buffer(1000, OverflowStrategy.dropTail)
Java
jobs.buffer(1000, OverflowStrategy.dropTail());

Instead of dropping the youngest element from the tail of the buffer a new element can be dropped without enqueueing it to the buffer at all.

Scala
jobs.buffer(1000, OverflowStrategy.dropNew)
Java
jobs.buffer(1000, OverflowStrategy.dropNew());

Here is another example with a queue of 1000 jobs, but it makes space for the new element by dropping one element from the head of the buffer. This is the oldest waiting job. This is the preferred strategy if jobs are expected to be resent if not processed in a certain period. The oldest element will be retransmitted soon, (in fact a retransmitted duplicate might be already in the queue!) so it makes sense to drop it first.

Scala
jobs.buffer(1000, OverflowStrategy.dropHead)
Java
jobs.buffer(1000, OverflowStrategy.dropHead());

Compared to the dropping strategies above, dropBuffer drops all the 1000 jobs it has enqueued once the buffer gets full. This aggressive strategy is useful when dropping jobs is preferred to delaying jobs.

Scala
jobs.buffer(1000, OverflowStrategy.dropBuffer)
Java
jobs.buffer(1000, OverflowStrategy.dropBuffer());

If our imaginary external job provider is a client using our API, we might want to enforce that the client cannot have more than 1000 queued jobs otherwise we consider it flooding and terminate the connection. This is easily achievable by the error strategy which simply fails the stream once the buffer gets full.

Scala
jobs.buffer(1000, OverflowStrategy.fail)
Java
jobs.buffer(1000, OverflowStrategy.fail());

Rate transformation

Understanding conflate

When a fast producer can not be informed to slow down by backpressure or some other signal, conflate might be useful to combine elements from a producer until a demand signal comes from a consumer.

Below is an example snippet that summarizes fast stream of elements to a standard deviation, mean and count of elements that have arrived while the stats have been calculated.

Scala
val statsFlow = Flow[Double]
  .conflateWithSeed(Seq(_))(_ :+ _)
  .map { s ⇒
    val μ = s.sum / s.size
    val se = s.map(x ⇒ pow(x - μ, 2))
    val σ = sqrt(se.sum / se.size)
    (σ, μ, s.size)
  }
Java
final Flow<Double, Tuple3<Double, Double, Integer>, NotUsed> statsFlow =
  Flow.of(Double.class)
    .conflateWithSeed(elem -> Collections.singletonList(elem), (acc, elem) -> {
      return Stream
        .concat(acc.stream(), Collections.singletonList(elem).stream())
        .collect(Collectors.toList());
    })
    .map(s -> {
      final Double mean = s.stream().mapToDouble(d -> d).sum() / s.size();
      final DoubleStream se = s.stream().mapToDouble(x -> Math.pow(x - mean, 2));
      final Double stdDev = Math.sqrt(se.sum() / s.size());
      return new Tuple3<>(stdDev, mean, s.size());
    });

This example demonstrates that such flow’s rate is decoupled. The element rate at the start of the flow can be much higher that the element rate at the end of the flow.

Another possible use of conflate is to not consider all elements for summary when producer starts getting too fast. Example below demonstrates how conflate can be used to implement random drop of elements when consumer is not able to keep up with the producer.

Scala
val p = 0.01
val sampleFlow = Flow[Double]
  .conflateWithSeed(Seq(_)) {
    case (acc, elem) if Random.nextDouble < p ⇒ acc :+ elem
    case (acc, _)                             ⇒ acc
  }
  .mapConcat(identity)
Java
final Double p = 0.01;
final Flow<Double, Double, NotUsed> sampleFlow = Flow.of(Double.class)
  .conflateWithSeed(elem -> Collections.singletonList(elem), (acc, elem) -> {
    if (r.nextDouble() < p) {
      return Stream
        .concat(acc.stream(), Collections.singletonList(elem).stream())
        .collect(Collectors.toList());
    }
    return acc;
  })
  .mapConcat(d -> d);

Understanding expand

Expand helps to deal with slow producers which are unable to keep up with the demand coming from consumers. Expand allows to extrapolate a value to be sent as an element to a consumer.

As a simple use of expand here is a flow that sends the same element to consumer when producer does not send any new elements.

Scala
val lastFlow = Flow[Double]
  .expand(Iterator.continually(_))
Java
final Flow<Double, Double, NotUsed> lastFlow = Flow.of(Double.class)
  .expand(in -> Stream.iterate(in, i -> i).iterator());

Expand also allows to keep some state between demand requests from the downstream. Leveraging this, here is a flow that tracks and reports a drift between fast consumer and slow producer.

Scala
val driftFlow = Flow[Double]
  .expand(i ⇒ Iterator.from(0).map(i -> _))
Java
	final Flow<Double, Pair<Double, Integer>, NotUsed> driftFlow = Flow.of(Double.class)
      .expand(d -> Stream.iterate(0, i -> i + 1).map(i -> new Pair<>(d, i)).iterator());

Note that all of the elements coming from upstream will go through expand at least once. This means that the output of this flow is going to report a drift of zero if producer is fast enough, or a larger drift otherwise.

The source code for this page can be found here.