c9b3f1de6d
* Adds internal access to materializer before initialization (#25324) * Implements new actorRef source based on graph stage (#25324) * Removes obsolete actorRef source (#25324) * Improves backwards compatibility with old implementation (#25324) * Removes dedicated new subclass for materializer access again (#25324) * Improves implementation (#25324) * Finalizes implementation (#25324) * Small improvements to API and documentation (#25324) * Completion strategy as a replacement for poison pill (#25324) * Adding more tests and updating the documentation (#25324)
110 lines
4 KiB
Scala
110 lines
4 KiB
Scala
/*
|
|
* Copyright (C) 2015-2019 Lightbend Inc. <https://www.lightbend.com>
|
|
*/
|
|
|
|
package akka.stream.impl
|
|
|
|
import akka.NotUsed
|
|
import akka.actor._
|
|
import akka.annotation.{ DoNotInherit, InternalApi }
|
|
import akka.stream._
|
|
import akka.stream.impl.StreamLayout.AtomicModule
|
|
import org.reactivestreams._
|
|
|
|
import scala.annotation.unchecked.uncheckedVariance
|
|
import akka.event.Logging
|
|
|
|
/**
|
|
* INTERNAL API
|
|
*/
|
|
@DoNotInherit private[akka] abstract class SourceModule[+Out, +Mat](val shape: SourceShape[Out])
|
|
extends AtomicModule[SourceShape[Out], Mat] {
|
|
|
|
protected def label: String = Logging.simpleName(this)
|
|
final override def toString: String = f"$label [${System.identityHashCode(this)}%08x]"
|
|
|
|
def create(context: MaterializationContext): (Publisher[Out] @uncheckedVariance, Mat)
|
|
|
|
// TODO: Remove this, no longer needed?
|
|
protected def newInstance(shape: SourceShape[Out] @uncheckedVariance): SourceModule[Out, Mat]
|
|
|
|
// TODO: Amendshape changed the name of ports. Is it needed anymore?
|
|
|
|
def attributes: Attributes
|
|
|
|
protected def amendShape(attr: Attributes): SourceShape[Out] = {
|
|
val thisN = traversalBuilder.attributes.nameOrDefault(null)
|
|
val thatN = attr.nameOrDefault(null)
|
|
|
|
if ((thatN eq null) || thisN == thatN) shape
|
|
else shape.copy(out = Outlet(thatN + ".out"))
|
|
}
|
|
|
|
override private[stream] def traversalBuilder =
|
|
LinearTraversalBuilder.fromModule(this, attributes).makeIsland(SourceModuleIslandTag)
|
|
|
|
}
|
|
|
|
/**
|
|
* INTERNAL API
|
|
* Holds a `Subscriber` representing the input side of the flow.
|
|
* The `Subscriber` can later be connected to an upstream `Publisher`.
|
|
*/
|
|
@InternalApi private[akka] final class SubscriberSource[Out](val attributes: Attributes, shape: SourceShape[Out])
|
|
extends SourceModule[Out, Subscriber[Out]](shape) {
|
|
|
|
override def create(context: MaterializationContext): (Publisher[Out], Subscriber[Out]) = {
|
|
val processor = new VirtualProcessor[Out]
|
|
(processor, processor)
|
|
}
|
|
|
|
override protected def newInstance(shape: SourceShape[Out]): SourceModule[Out, Subscriber[Out]] =
|
|
new SubscriberSource[Out](attributes, shape)
|
|
override def withAttributes(attr: Attributes): SourceModule[Out, Subscriber[Out]] =
|
|
new SubscriberSource[Out](attr, amendShape(attr))
|
|
}
|
|
|
|
/**
|
|
* INTERNAL API
|
|
* Construct a transformation starting with given publisher. The transformation steps
|
|
* are executed by a series of [[org.reactivestreams.Processor]] instances
|
|
* that mediate the flow of elements downstream and the propagation of
|
|
* back-pressure upstream.
|
|
*/
|
|
@InternalApi private[akka] final class PublisherSource[Out](
|
|
p: Publisher[Out],
|
|
val attributes: Attributes,
|
|
shape: SourceShape[Out])
|
|
extends SourceModule[Out, NotUsed](shape) {
|
|
|
|
override protected def label: String = s"PublisherSource($p)"
|
|
|
|
override def create(context: MaterializationContext) = (p, NotUsed)
|
|
|
|
override protected def newInstance(shape: SourceShape[Out]): SourceModule[Out, NotUsed] =
|
|
new PublisherSource[Out](p, attributes, shape)
|
|
override def withAttributes(attr: Attributes): SourceModule[Out, NotUsed] =
|
|
new PublisherSource[Out](p, attr, amendShape(attr))
|
|
}
|
|
|
|
/**
|
|
* INTERNAL API
|
|
* Creates and wraps an actor into [[org.reactivestreams.Publisher]] from the given `props`,
|
|
* which should be [[akka.actor.Props]] for an [[akka.stream.actor.ActorPublisher]].
|
|
*/
|
|
@InternalApi private[akka] final class ActorPublisherSource[Out](
|
|
props: Props,
|
|
val attributes: Attributes,
|
|
shape: SourceShape[Out])
|
|
extends SourceModule[Out, ActorRef](shape) {
|
|
|
|
override def create(context: MaterializationContext) = {
|
|
val publisherRef = ActorMaterializerHelper.downcast(context.materializer).actorOf(context, props)
|
|
(akka.stream.actor.ActorPublisher[Out](publisherRef), publisherRef)
|
|
}
|
|
|
|
override protected def newInstance(shape: SourceShape[Out]): SourceModule[Out, ActorRef] =
|
|
new ActorPublisherSource[Out](props, attributes, shape)
|
|
override def withAttributes(attr: Attributes): SourceModule[Out, ActorRef] =
|
|
new ActorPublisherSource(props, attr, amendShape(attr))
|
|
}
|