94 lines
4 KiB
Scala
94 lines
4 KiB
Scala
|
/*
|
||
|
|
* Copyright (C) 2014-2019 Lightbend Inc. <https://www.lightbend.com>
|
||
|
|
*/
|
||
|
|
|
||
|
|
package akka.stream.javadsl
|
||
|
|
|
||
|
|
import akka.annotation.ApiMayChange
|
||
|
|
import akka.japi.{ Pair, Util, function }
|
||
|
|
import akka.stream._
|
||
|
|
import akka.stream.impl.LinearTraversalBuilder
|
||
|
|
|
||
|
|
import scala.annotation.unchecked.uncheckedVariance
|
||
|
|
import scala.collection.JavaConverters._
|
||
|
|
import scala.collection.immutable
|
||
|
|
import java.util.concurrent.CompletionStage
|
||
|
|
|
||
|
|
import scala.compat.java8.FutureConverters._
|
||
|
|
|
||
|
|
/**
|
||
|
|
* API MAY CHANGE
|
||
|
|
*/
|
||
|
|
@ApiMayChange
|
||
|
|
object SourceWithContext {
|
||
|
|
def from[Out, Mat](underlying: Source[Out, Mat]): SourceWithContext[Out, Out, Mat] = {
|
||
|
|
new SourceWithContext(scaladsl.SourceWithContext(underlying.asScala))
|
||
|
|
}
|
||
|
|
|
||
|
|
def fromPairs[Out, Ctx, Mat](under: Source[Pair[Out, Ctx], Mat]): SourceWithContext[Ctx, Out, Mat] = {
|
||
|
|
new SourceWithContext(scaladsl.SourceWithContext.from(under.asScala.map(_.toScala)))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* API MAY CHANGE
|
||
|
|
*/
|
||
|
|
@ApiMayChange
|
||
|
|
final class SourceWithContext[+Ctx, +Out, +Mat](delegate: scaladsl.SourceWithContext[Ctx, Out, Mat]) extends Graph[SourceShape[(Out, Ctx)], Mat] {
|
||
|
|
override val traversalBuilder: LinearTraversalBuilder = delegate.traversalBuilder
|
||
|
|
override val shape: SourceShape[(Out, Ctx)] = delegate.shape
|
||
|
|
override def withAttributes(attr: Attributes): SourceWithContext[Ctx, Out, Mat] = new SourceWithContext(delegate.withAttributes(attr))
|
||
|
|
|
||
|
|
def mapContext[Ctx2](extractContext: function.Function[Ctx, Ctx2]): SourceWithContext[Ctx2, Out, Mat] = {
|
||
|
|
new SourceWithContext(delegate.mapContext(extractContext.apply))
|
||
|
|
}
|
||
|
|
|
||
|
|
def via[Ctx2, Out2, Mat2](viaFlow: Graph[FlowShape[Pair[Out @uncheckedVariance, Ctx @uncheckedVariance], Pair[Out2, Ctx2]], Mat2]): SourceWithContext[Ctx2, Out2, Mat] = {
|
||
|
|
val under = endContextPropagation().via(viaFlow)
|
||
|
|
SourceWithContext.fromPairs(under)
|
||
|
|
}
|
||
|
|
|
||
|
|
def to[Mat2](sink: Graph[SinkShape[Pair[Out @uncheckedVariance, Ctx @uncheckedVariance]], Mat2]): RunnableGraph[Mat] =
|
||
|
|
endContextPropagation().toMat(sink, Keep.left)
|
||
|
|
|
||
|
|
def toMat[Mat2, Mat3](sink: Graph[SinkShape[Pair[Out @uncheckedVariance, Ctx @uncheckedVariance]], Mat2], combine: function.Function2[Mat, Mat2, Mat3]): RunnableGraph[Mat3] =
|
||
|
|
endContextPropagation().toMat(sink, combine)
|
||
|
|
|
||
|
|
def endContextPropagation(): Source[Pair[Out @uncheckedVariance, Ctx @uncheckedVariance], Mat @uncheckedVariance] =
|
||
|
|
delegate.endContextPropagation.map { case (o, c) ⇒ Pair(o, c) }.asJava
|
||
|
|
|
||
|
|
def map[Out2](f: function.Function[Out, Out2]): SourceWithContext[Ctx, Out2, Mat] =
|
||
|
|
new SourceWithContext(delegate.map(f.apply))
|
||
|
|
|
||
|
|
def mapAsync[Out2](parallelism: Int, f: function.Function[Out, CompletionStage[Out2]]): SourceWithContext[Ctx, Out2, Mat] =
|
||
|
|
new SourceWithContext(delegate.mapAsync[Out2](parallelism)(o ⇒ f.apply(o).toScala))
|
||
|
|
|
||
|
|
def collect[Out2](pf: PartialFunction[Out, Out2]): SourceWithContext[Ctx, Out2, Mat] =
|
||
|
|
new SourceWithContext(delegate.collect(pf))
|
||
|
|
|
||
|
|
def filter(p: function.Predicate[Out]): SourceWithContext[Ctx, Out, Mat] =
|
||
|
|
new SourceWithContext(delegate.filter(p.test))
|
||
|
|
|
||
|
|
def filterNot(p: function.Predicate[Out]): SourceWithContext[Ctx, Out, Mat] =
|
||
|
|
new SourceWithContext(delegate.filterNot(p.test))
|
||
|
|
|
||
|
|
def grouped(n: Int): SourceWithContext[java.util.List[Ctx @uncheckedVariance], java.util.List[Out @uncheckedVariance], Mat] = {
|
||
|
|
val f = new function.Function[immutable.Seq[Ctx], java.util.List[Ctx]] {
|
||
|
|
def apply(ctxs: immutable.Seq[Ctx]) = ctxs.asJava
|
||
|
|
}
|
||
|
|
|
||
|
|
new SourceWithContext(delegate.grouped(n).map(_.asJava)).mapContext(f)
|
||
|
|
}
|
||
|
|
|
||
|
|
def mapConcat[Out2](f: function.Function[Out, _ <: java.lang.Iterable[Out2]]): SourceWithContext[Ctx, Out2, Mat] =
|
||
|
|
new SourceWithContext(delegate.mapConcat(elem ⇒ Util.immutableSeq(f.apply(elem))))
|
||
|
|
|
||
|
|
def statefulMapConcat[Out2](f: function.Creator[function.Function[Out, java.lang.Iterable[Out2]]]): SourceWithContext[Ctx, Out2, Mat] =
|
||
|
|
new SourceWithContext(delegate.statefulMapConcat { () ⇒
|
||
|
|
val fun = f.create()
|
||
|
|
elem ⇒ Util.immutableSeq(fun(elem))
|
||
|
|
})
|
||
|
|
|
||
|
|
def asScala = delegate
|
||
|
|
}
|