package org.jboss.as.ejb3.component.stateful;
import java.io.ObjectStreamException;
import java.io.Serializable;
import java.security.AccessController;
import java.security.PrivilegedAction;
import java.util.Map;
import java.util.concurrent.atomic.AtomicReference;
import org.jboss.as.naming.ManagedReference;
import org.jboss.as.naming.ValueManagedReference;
import org.jboss.as.server.CurrentServiceContainer;
import org.jboss.ejb.client.SessionID;
import org.jboss.invocation.InterceptorFactoryContext;
import org.jboss.invocation.SimpleInterceptorFactoryContext;
import org.jboss.msc.service.ServiceContainer;
import org.jboss.msc.service.ServiceController;
import org.jboss.msc.service.ServiceName;
import org.jboss.msc.value.ImmediateValue;
/**
*
* Serialized form of a SFSB
*
* @author Stuart Douglas
*/
public class SerializedStatefulSessionComponent implements Serializable {
private static final long serialVersionUID = 1L;
private final String serviceName;
private final SessionID sessionID;
private final Map<Object, Object> serializableInterceptors;
private final ManagedReference instance;
public SerializedStatefulSessionComponent(final ManagedReference instance, final SessionID sessionID, final String serviceName, final Map<Object, Object> serializableInterceptors) {
this.instance = instance;
this.sessionID = sessionID;
this.serviceName = serviceName;
this.serializableInterceptors = serializableInterceptors;
}
private Object readResolve() throws ObjectStreamException {
ServiceName name = ServiceName.parse(serviceName);
ServiceController<?> service = currentServiceContainer().getRequiredService(name);
StatefulSessionComponent component = (StatefulSessionComponent) service.getValue();
final InterceptorFactoryContext context = new SimpleInterceptorFactoryContext();
for(final Map.Entry<Object, Object> entry : serializableInterceptors.entrySet()) {
AtomicReference<ManagedReference> referenceReference = new AtomicReference<ManagedReference>(new ValueManagedReference(new ImmediateValue<Object>(entry.getValue())));
context.getContextData().put(entry.getKey(), referenceReference);
}
context.getContextData().put(SessionID.class, sessionID);
return component.constructComponentInstance(instance, false, context);
}
private static ServiceContainer currentServiceContainer() {
return AccessController.doPrivileged(new PrivilegedAction<ServiceContainer>() {
@Override
public ServiceContainer run() {
return CurrentServiceContainer.getServiceContainer();
}
});
}
}