/*
* Quasar: lightweight threads and actors for the JVM.
* Copyright (C) 2013, Parallel Universe Software Co. All rights reserved.
*
* This program and the accompanying materials are dual-licensed under
* either the terms of the Eclipse Public License v1.0 as published by
* the Eclipse Foundation
*
* or (per the licensee's choosing)
*
* under the terms of the GNU Lesser General Public License version 3.0
* as published by the Free Software Foundation.
*/
package co.paralleluniverse.remote.galaxy;
import co.paralleluniverse.actors.ActorImpl;
import co.paralleluniverse.actors.LifecycleListener;
import co.paralleluniverse.actors.LifecycleListenerProxy;
import co.paralleluniverse.actors.RemoteActor;
import co.paralleluniverse.galaxy.cluster.NodeChangeListener;
import co.paralleluniverse.galaxy.quasar.Grid;
import java.util.Collections;
import java.util.Iterator;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
*
* @author pron
*/
public class GlxLifecycleListenerProxy extends LifecycleListenerProxy {
private static final Logger LOG = LoggerFactory.getLogger(GlxLifecycleListenerProxy.class);
// private static final ReferenceQueue<LifecycleListener> oldlistenerRefQueue = new ReferenceQueue<>();
private final Grid grid;
// private static final Map<Short, Set<RAPhantomReference>> map = new ConcurrentHashMap<>();
private final static Set<RegistryRecord> listenerRegistry = Collections.newSetFromMap(new ConcurrentHashMap<RegistryRecord, Boolean>());
public GlxLifecycleListenerProxy() {
try {
grid = new Grid(co.paralleluniverse.galaxy.Grid.getInstance());
grid.cluster().addNodeChangeListener(new NodeChangeListener() {
@Override
public void nodeAdded(short id) {
}
@Override
public void nodeSwitched(short id) {
}
@Override
public void nodeRemoved(short id) {
for (Iterator<RegistryRecord> it = listenerRegistry.iterator(); it.hasNext();) {
RegistryRecord registryRecord = it.next();
if (registryRecord.getOwnerNodeId()==id) {
registryRecord.listener.dead(registryRecord.actor, new Throwable("cluster node removed"));
//TODO: remove listeners
it.remove();
}
}
}
});
} catch (InterruptedException ex) {
throw new RuntimeException(ex);
}
}
@Override
public void addLifecycleListener(final RemoteActor actor, final LifecycleListener listener) {
final GlxRemoteActor glxActor = (GlxRemoteActor) actor;
final short nodeId = glxActor.getOwnerNodeId();
if (!grid.cluster().getNodes().contains(nodeId)) {
listener.dead(actor, null);
return;
}
super.addLifecycleListener(actor, listener);
listenerRegistry.add(new RegistryRecord(listener, glxActor));
}
@Override
public void removeLifecycleListener(RemoteActor actor, LifecycleListener listener) {
super.removeLifecycleListener(actor, listener);
listenerRegistry.remove(new RegistryRecord(listener, (GlxRemoteActor) actor));
}
@Override
public void removeLifecycleListeners(RemoteActor actor, ActorImpl observer) {
super.removeLifecycleListeners(actor, observer);
for (Iterator<RegistryRecord> it = listenerRegistry.iterator(); it.hasNext();) {
RegistryRecord registryRecord = it.next();
if (registryRecord.actor.equals(actor))
it.remove();
}
}
class RegistryRecord {
final LifecycleListener listener;
final GlxRemoteActor actor;
public RegistryRecord(LifecycleListener listener, GlxRemoteActor actor) {
this.listener = listener;
this.actor = actor;
}
short getOwnerNodeId() {
return actor.getOwnerNodeId();
}
}
}