gistfile1.txt

case class SendMessage(actionName: String, messageType: String, content: Any, ttl: Long, guaranteed: Boolean, nberOfDataChanges: Int, checks: List[Check[_, _]], next: ActorRef, session: Session, duration: Duration = Duration.Undefined)

class SendStartActionBuilder(val channelName: String,
                             val actionName: EvaluatableString,
                             val queryString: EvaluatableString,
                             val positionalParameters: List[EvaluatableString],
                             val namedParameters: Map[String, EvaluatableString],
                             val nberOfDataChanges: Int,
                             val checks: List[MttpCheck[_, _]],
                             val duration: Duration = Duration.Undefined,
                             val next: ActorRef = null) extends ActionBuilder {

  def withNext(next: ActorRef): ActionBuilder = new SendStartActionBuilder(channelName, actionName, queryString, positionalParameters, namedParameters, nberOfDataChanges, checks, duration, next)

  def build(registry: ProtocolConfigurationRegistry): ActorRef = system.actorOf(Props(new SendStartAction(channelName, actionName, queryString, positionalParameters, namedParameters, nberOfDataChanges, checks, duration, next)))

}

class SendStartAction(channelName: String,
                      actionName: EvaluatableString,
                      queryString: EvaluatableString,
                      positionalParameters: List[EvaluatableString],
                      namedParameters: Map[String, EvaluatableString],
                      val nberOfDataChanges: Int,
                      checks: List[Check[_, _]],
                      val duration: Duration,
                      val next: ActorRef)
  extends ClientChannelAction(actionName) {

  def execute(aSession: Session) {

    if (aSession.isAttributeDefined(channelName)) {
      val bucket = new Bucket
      val requestId = Buckets.addBucket(bucket)
      
      val query = new Query(requestId,
        queryString(aSession),
        positionalParameters.map(content => content(aSession)).toList.asJava,
        namedParameters.mapValues(value => value(aSession)).toMap.asJava,
        "", -1, Array(1, 0))
      
      val actor = aSession.getAttribute(channelName).asInstanceOf[ActorRef]
      actor ! SendMessage(actionName(aSession), "query.start", query, -1, false, nberOfDataChanges, checks, next, aSession, duration)

    } else {
      next ! aSession

    }
  }
}