|
| 1 | +package actors |
| 2 | +import akka.actor._ |
| 3 | +import akka.http.scaladsl.model.ws.{Message, TextMessage, WebSocketRequest} |
| 4 | +import actors.PublishSocketMessageActor.{AddOutActor, PublishMessage, StopMessage} |
| 5 | +import akka.http.scaladsl.Http |
| 6 | +import akka.stream.Materializer |
| 7 | +import akka.stream.scaladsl.{Flow, Keep, Sink, Source} |
| 8 | +import models.EventEnums.EventType |
| 9 | +import models.{EventJsonSupport, RegistryEvent} |
| 10 | +import play.api.Logger |
| 11 | +import spray.json._ |
| 12 | + |
| 13 | +import scala.collection.mutable |
| 14 | +import scala.collection.mutable.ListBuffer |
| 15 | +import scala.concurrent.Promise |
| 16 | + |
| 17 | +object PublishSocketMessageActor { |
| 18 | + def props(irBasePath: String, mat: Materializer, actorSys: ActorSystem):Props = Props(new PublishSocketMessageActor(irBasePath, mat, actorSys)) |
| 19 | + final case class AddOutActor(out: ActorRef, event: EventType) |
| 20 | + final case class PublishMessage(msg: RegistryEvent) |
| 21 | + final case class StopMessage(toStop: ActorRef) |
| 22 | +} |
| 23 | + |
| 24 | +class PublishSocketMessageActor(irBasePath: String, mat: Materializer, actorSys: ActorSystem) extends Actor with EventJsonSupport { |
| 25 | + |
| 26 | + val eventActorMap: mutable.HashMap[EventType, ListBuffer[ActorRef]] = new mutable.HashMap[EventType, ListBuffer[ActorRef]]() |
| 27 | + |
| 28 | + override def preStart() { |
| 29 | + |
| 30 | + Logger.debug("pre start called in publisher" + self) |
| 31 | + val flow: Flow[Message, Message, Promise[Option[Message]]] = |
| 32 | + Flow.fromSinkAndSourceMat( |
| 33 | + Sink.foreach[Message]{ msg => |
| 34 | + self ! msg}, |
| 35 | + Source(List(TextMessage("one"), TextMessage("two"))) |
| 36 | + .concatMat(Source.maybe[Message])(Keep.right))(Keep.right) |
| 37 | + |
| 38 | + |
| 39 | + Http()(actorSys).singleWebSocketRequest( |
| 40 | + WebSocketRequest("ws://" + irBasePath + "/events"), |
| 41 | + flow)(mat) |
| 42 | + |
| 43 | + |
| 44 | + } |
| 45 | + |
| 46 | + override def postStop() { |
| 47 | + Logger.debug("post stop called in publisher" + self) |
| 48 | + } |
| 49 | + |
| 50 | + def receive: PartialFunction[Any, Unit] = { |
| 51 | + |
| 52 | + case StopMessage(toStop) => |
| 53 | + Logger.debug("stop received" + toStop) |
| 54 | + for ((k, v) <- eventActorMap) v -= toStop |
| 55 | + |
| 56 | + case AddOutActor(out, event) => |
| 57 | + Logger.debug("received add out actor" + out) |
| 58 | + if (!eventActorMap.contains(event)){ |
| 59 | + eventActorMap += (event -> new ListBuffer[ActorRef]()) |
| 60 | + } |
| 61 | + eventActorMap(event) += out |
| 62 | + |
| 63 | + |
| 64 | + case TextMessage.Strict(msg) => |
| 65 | + Logger.debug("received something " + msg) |
| 66 | + val registryEvent = msg.parseJson.convertTo[RegistryEvent](eventFormat) |
| 67 | + self ! PublishMessage(registryEvent) |
| 68 | + |
| 69 | + case PublishMessage(msg) => |
| 70 | + Logger.debug("publish message called with message" + msg) |
| 71 | + if(eventActorMap.contains(msg.eventType)){ |
| 72 | + |
| 73 | + val list = eventActorMap(msg.eventType) |
| 74 | + list.foreach(actor => { |
| 75 | + Logger.debug("sending message to actor" + actor) |
| 76 | + actor ! PublishMessage(msg) |
| 77 | + }) |
| 78 | + } |
| 79 | + } |
| 80 | + |
| 81 | +} |
0 commit comments