//设置package包名称以及导入依赖的类
package com.flipkart.connekt.busybees.clients

import akka.actor.{Actor, Props, Terminated}
import akka.routing.{ActorRefRoutee, RoundRobinRoutingLogic, Router}
import com.flipkart.connekt.commons.iomodels.XmppRequest

class XMPPChannelSupervisor(maxConnections: Int) extends Actor {
  var router = {
    val xmppChannelHandlers = Vector.fill(maxConnections) {
      val r = context.actorOf(Props[XMPPChannelHandler])
      context watch r
      ActorRefRoutee(r)
    }
    Router(RoundRobinRoutingLogic(), xmppChannelHandlers)
  }

  override def receive: Receive = {
    case d: XmppRequest =>
      router.route(d, self)

    case CreateNewXMPPChannelHandler =>
      val r = context.actorOf(Props[XMPPChannelHandler])
      r ! Configure
      context watch r
      router = router.addRoutee(r)

    case Terminated(a) =>
      router = router.removeRoutee(a)
      val r = context.actorOf(Props[XMPPChannelHandler])
      r ! Configure
      context watch r
      router = router.addRoutee(r)
  }
}

case object CreateNewXMPPChannelHandler 

//设置package包名称以及导入依赖的类
package name.mikulskibartosz.demo.akka.router

import akka.actor.SupervisorStrategy.Restart
import akka.actor.{OneForOneStrategy, Actor, Props}
import akka.routing.{ActorRefRoutee, RoundRobinRoutingLogic, Router}
import name.mikulskibartosz.demo.akka.actors.QueryExecutor
import name.mikulskibartosz.demo.akka.messages.{ExecuteQuery, ServiceUnavailable}
import name.mikulskibartosz.demo.akka.service.Service


class MessageRouter(private val service: Service) extends Actor {
  var router = {
    val routees = Vector.fill(5) {
      val r = context.actorOf(Props(new QueryExecutor(service)))
      context watch r
      ActorRefRoutee(r)
    }
    Router(RoundRobinRoutingLogic(), routees)
  }

  override def supervisorStrategy = OneForOneStrategy() {
    case _ => Restart
  }

  def receive = {
    case ExecuteQuery => router.route(ExecuteQuery, sender())
    case ServiceUnavailable => context.parent ! ServiceUnavailable
  }
}