pl.touk.nussknacker.engine.testmode.TestServiceInvocationCollector.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of nussknacker-scenario-compiler_2.12 Show documentation

nussknacker-scenario-compiler

The newest version!

package pl.touk.nussknacker.engine.testmode

import cats.Monad
import cats.implicits._
import pl.touk.nussknacker.engine.api.test.InvocationCollectors.{CollectableAction, ToCollect, TransmissionNames}
import pl.touk.nussknacker.engine.api.{Context, ContextId, NodeId}
import pl.touk.nussknacker.engine.resultcollector.ResultCollector

import scala.language.higherKinds

class TestServiceInvocationCollector(resultsCollectingListener: ResultsCollectingListener[_]) extends ResultCollector {

  override def collectWithResponse[A, F[_]: Monad](
      contextId: ContextId,
      nodeId: NodeId,
      serviceRef: String,
      request: => ToCollect,
      mockValue: Option[A],
      action: => F[CollectableAction[A]],
      names: TransmissionNames
  ): F[A] = {
    mockValue match {
      case Some(mockVal) =>
        ResultsCollectingListenerHolder.updateResults(
          resultsCollectingListener.runId,
          _.updateExternalInvocationResult(
            nodeId.id,
            contextId,
            serviceRef,
            request,
            resultsCollectingListener.variableEncoder
          )
        )
        mockVal.pure[F]
      case None =>
        action.map { case CollectableAction(resultToCollect, result) =>
          val invocationResult = Map("request" -> request, "response" -> resultToCollect())
          ResultsCollectingListenerHolder.updateResults(
            resultsCollectingListener.runId,
            _.updateExternalInvocationResult(
              nodeId.id,
              contextId,
              serviceRef,
              invocationResult,
              resultsCollectingListener.variableEncoder
            )
          )
          result
        }
    }
  }

  def createSinkInvocationCollector(nodeId: String, ref: String): SinkInvocationCollector =
    new SinkInvocationCollector(resultsCollectingListener, nodeId, ref)

}

//TODO: this should be somehow expressed via ResultCollector/TestServiceInvocationCollector
final class SinkInvocationCollector(
    resultsCollectingListener: ResultsCollectingListener[_],
    nodeId: String,
    ref: String
) extends Serializable {

  def collect(context: Context, result: Any): Unit = {
    ResultsCollectingListenerHolder.updateResults(
      resultsCollectingListener.runId,
      _.updateExternalInvocationResult(
        nodeId,
        ContextId(context.id),
        ref,
        result,
        resultsCollectingListener.variableEncoder
      )
    )
  }

}