xuml.tools.model.compiler.runtime.actor.RootActor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of xuml-model-compiler-runtime Show documentation

Runtime classes for use with classes generated by xuml-model-compiler

There is a newer version: 0.6

Show newest version

package xuml.tools.model.compiler.runtime.actor;

import java.util.HashMap;

import javax.persistence.EntityManagerFactory;

import com.google.common.collect.Maps;

import akka.actor.ActorRef;
import akka.actor.Props;
import akka.actor.UntypedActor;
import akka.event.Logging;
import akka.event.LoggingAdapter;
import xuml.tools.model.compiler.runtime.SignalProcessorListenerFactory;
import xuml.tools.model.compiler.runtime.message.CloseEntityActor;
import xuml.tools.model.compiler.runtime.message.Signal;
import xuml.tools.model.compiler.runtime.message.StopEntityActor;

public class RootActor extends UntypedActor {

    private EntityManagerFactory emf;
    private final HashMap actors = Maps.newHashMap();
    private final LoggingAdapter log;
    private SignalProcessorListenerFactory listenerFactory;

    public RootActor() {
        log = Logging.getLogger(getContext().system(), this);
    }

    @Override
    public void onReceive(Object message) throws Exception {
        log.debug("received message {}", message.getClass().getName());
        if (message instanceof EntityManagerFactory)
            handleMessage((EntityManagerFactory) message);
        else if (message instanceof SignalProcessorListenerFactory)
            listenerFactory = (SignalProcessorListenerFactory) message;
        else if (message instanceof Signal)
            handleMessage((Signal) message);
        else if (message instanceof CloseEntityActor)
            handleMessage((CloseEntityActor) message);
    }

    private void handleMessage(CloseEntityActor message) {
        String key = message.getEntityUniqueId();
        ActorInfo info = actors.remove(key);
        if (info.counter > 1) {
            actors.put(key, info.decrement());
        } else {
            info.actor.tell(new StopEntityActor(), getSelf());
        }
    }

    private void handleMessage(EntityManagerFactory message) {
        emf = message;
    }

    private void handleMessage(Signal message) {
        String key = message.getEntityUniqueId();
        ActorRef actor = getActor(key);
        actor.tell(message, getSelf());
    }

    private ActorRef getActor(String key) {
        ActorInfo info = actors.get(key);
        if (info == null) {
            ActorRef actor = createActor();
            actors.put(key, new ActorInfo(actor, 1));
            actor.tell(emf, getSelf());
            if (listenerFactory != null)
                actor.tell(listenerFactory.create(key), getSelf());
        } else {
            actors.put(key, info.increment());
        }
        return actors.get(key).actor;
    }

    private ActorRef createActor() {
        return getContext().actorOf(Props.create(EntityActor.class));
    }

    private static final class ActorInfo {

        final ActorRef actor;
        final long counter;

        ActorInfo(ActorRef actor, long counter) {
            this.actor = actor;
            this.counter = counter;
        }

        ActorInfo increment() {
            return new ActorInfo(actor, counter + 1);
        }

        ActorInfo decrement() {
            return new ActorInfo(actor, counter - 1);
        }

    }
}