org.jboss.ejb.protocol.remote.ProtocolV3ObjectTable Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of jboss-ejb-client Show documentation
Show all versions of jboss-ejb-client Show documentation
Client library for EJB applications working against Wildfly - Jakarta EE Variant
/*
* JBoss, Home of Professional Open Source.
* Copyright 2017 Red Hat, Inc., and individual contributors
* as indicated by the @author tags.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.jboss.ejb.protocol.remote;
import java.io.IOException;
import java.io.InvalidObjectException;
import java.rmi.RemoteException;
import java.util.Collections;
import java.util.HashMap;
import java.util.IdentityHashMap;
import java.util.Map;
import java.util.concurrent.Future;
import jakarta.ejb.AsyncResult;
import jakarta.ejb.CreateException;
import jakarta.ejb.DuplicateKeyException;
import jakarta.ejb.EJBAccessException;
import jakarta.ejb.EJBException;
import jakarta.ejb.EJBHome;
import jakarta.ejb.EJBMetaData;
import jakarta.ejb.EJBObject;
import jakarta.ejb.EJBTransactionRequiredException;
import jakarta.ejb.EJBTransactionRolledbackException;
import jakarta.ejb.FinderException;
import jakarta.ejb.Handle;
import jakarta.ejb.HomeHandle;
import jakarta.ejb.NoSuchEJBException;
import jakarta.ejb.NoSuchEntityException;
import jakarta.ejb.ObjectNotFoundException;
import jakarta.ejb.RemoveException;
import jakarta.transaction.InvalidTransactionException;
import jakarta.transaction.NotSupportedException;
import jakarta.transaction.RollbackException;
import jakarta.transaction.SystemException;
import jakarta.transaction.TransactionRequiredException;
import jakarta.transaction.TransactionRolledbackException;
import org.jboss.ejb.client.AbstractEJBMetaData;
import org.jboss.ejb.client.Affinity;
import org.jboss.ejb.client.AttachmentKey;
import org.jboss.ejb.client.AttachmentKeys;
import org.jboss.ejb.client.BasicSessionID;
import org.jboss.ejb.client.ClusterAffinity;
import org.jboss.ejb.client.EJBClientInvocationContext;
import org.jboss.ejb.client.EJBClientPermission;
import org.jboss.ejb.client.EJBHandle;
import org.jboss.ejb.client.EJBHomeHandle;
import org.jboss.ejb.client.EJBHomeLocator;
import org.jboss.ejb.client.EJBIdentifier;
import org.jboss.ejb.client.EJBLocator;
import org.jboss.ejb.client.EJBMethodLocator;
import org.jboss.ejb.client.EJBModuleIdentifier;
import org.jboss.ejb.client.EntityEJBLocator;
import org.jboss.ejb.client.EntityEJBMetaData;
import org.jboss.ejb.client.NodeAffinity;
import org.jboss.ejb.client.SerializedEJBInvocationHandler;
import org.jboss.ejb.client.SessionID;
import org.jboss.ejb.client.StatefulEJBLocator;
import org.jboss.ejb.client.StatefulEJBMetaData;
import org.jboss.ejb.client.StatelessEJBLocator;
import org.jboss.ejb.client.StatelessEJBMetaData;
import org.jboss.ejb.client.TransactionID;
import org.jboss.ejb.client.URIAffinity;
import org.jboss.ejb.client.UnknownSessionID;
import org.jboss.ejb.client.UserTransactionID;
import org.jboss.ejb.client.XidTransactionID;
import org.jboss.marshalling.ObjectTable;
import org.jboss.marshalling.Unmarshaller;
/**
* @author David M. Lloyd
*/
@SuppressWarnings("serial")
final class ProtocolV3ObjectTable implements ObjectTable {
static final ProtocolV3ObjectTable INSTANCE = new ProtocolV3ObjectTable();
private static final Map objectWriters;
private static final Map stringWriters;
private static final Map, AbstractWritingExternalizer> classWriters;
private static final AbstractWritingExternalizer[] extById;
static {
final Object[] simpleObjects = {
TransactionID.PRIVATE_DATA_KEY,
Affinity.NONE,
Affinity.WEAK_AFFINITY_CONTEXT_KEY,
EJBClientInvocationContext.PRIVATE_ATTACHMENTS_KEY,
AttachmentKeys.TRANSACTION_ID_KEY,
AttachmentKeys.WEAK_AFFINITY,
AttachmentKeys.COMPRESS_RESPONSE,
AttachmentKeys.RESPONSE_COMPRESSION_LEVEL,
AttachmentKeys.TRANSACTION_KEY,
AttachmentKeys.HINTS_DISABLED,
AttachmentKeys.VIEW_CLASS_DATA_COMPRESSION_HINT_ATTACHMENT_KEY,
AttachmentKeys.VIEW_METHOD_DATA_COMPRESSION_HINT_ATTACHMENT_KEY,
Throwable.class.getName(),
Exception.class.getName(),
RuntimeException.class.getName(),
EJBLocator.class.getName(),
EJBHomeLocator.class.getName(),
StatelessEJBLocator.class.getName(),
StatefulEJBLocator.class.getName(),
EntityEJBLocator.class.getName(),
EJBHandle.class.getName(),
EJBHomeHandle.class.getName(),
SerializedEJBInvocationHandler.class.getName(),
SessionID.class.getName(),
UnknownSessionID.class.getName(),
BasicSessionID.class.getName(),
UserTransactionID.class.getName(),
XidTransactionID.class.getName(),
EJBHome.class.getName(),
EJBObject.class.getName(),
Handle.class.getName(),
HomeHandle.class.getName(),
EJBMetaData.class.getName(),
RemoteException.class.getName(),
NoSuchEJBException.class.getName(),
NoSuchEntityException.class.getName(),
CreateException.class.getName(),
DuplicateKeyException.class.getName(),
EJBAccessException.class.getName(),
EJBException.class.getName(),
EJBTransactionRequiredException.class.getName(),
EJBTransactionRolledbackException.class.getName(),
FinderException.class.getName(),
RemoveException.class.getName(),
ObjectNotFoundException.class.getName(),
Future.class.getName(),
SystemException.class.getName(),
RollbackException.class.getName(),
TransactionRequiredException.class.getName(),
TransactionRolledbackException.class.getName(),
NotSupportedException.class.getName(),
InvalidTransactionException.class.getName(),
StackTraceElement.class.getName(),
SessionID.Serialized.class.getName(),
TransactionID.class.getName(),
TransactionID.Serialized.class.getName(),
Affinity.class.getName(),
NodeAffinity.class.getName(),
ClusterAffinity.class.getName(),
URIAffinity.class.getName(),
EJBMethodLocator.class.getName(),
AbstractEJBMetaData.class.getName(),
StatelessEJBMetaData.class.getName(),
StatefulEJBMetaData.class.getName(),
EntityEJBMetaData.class.getName(),
AttachmentKey.class.getName(),
EJBClientPermission.class.getName(),
AsyncResult.class.getName(),
EJBModuleIdentifier.class.getName(),
EJBIdentifier.class.getName(),
"detailMessage",
"cause",
"stackTrace",
"value",
"suppressedExceptions",
"ejbCreate",
"ejbRemove",
"ejbHome",
"remove",
"ejbActivate",
"ejbPassivate",
"ejbLoad",
"ejbStore",
"appName",
"moduleName",
"distinctName",
"methodName",
"parameterTypeNames",
"viewType",
"beanName",
"moduleIdentifier",
"clusterName",
"identifier",
"affinity",
"classLoaderName",
"moduleName",
"moduleVersion",
"declaringClass",
"methodName",
"fileName",
"lineNumber",
"format",
"java.base",
"java.naming",
"java.sql",
};
final AbstractWritingExternalizer[] extByIdTmp = new AbstractWritingExternalizer[simpleObjects.length];
final Map objMap = new IdentityHashMap<>();
final Map stringMap = new HashMap<>();
for (int i = 0, simpleObjectsLength = simpleObjects.length; i < simpleObjectsLength; i++) {
ByteExternalizer ext = new ByteExternalizer(simpleObjects[i], i);
extByIdTmp[i] = ext;
if(ext.getObject() instanceof String) {
stringMap.put(ext.getObject(), ext);
} else {
objMap.put(ext.getObject(), ext);
}
}
// TODO: add class-based ext for LocalTransaction
// TODO: add class-based ext for RemoteTransaction
// TODO: add class-based ext for InputStream
// TODO: add class-based ext for OutputStream
extById = extByIdTmp;
//we have a separate map for string objects, as we don't want to call hashCode on arbitrary objects
//There are tests we need to pass that involve serializing a Vector that contains itself, which will
//fail if we call hashCode
objectWriters = objMap;
stringWriters = stringMap;
classWriters = Collections.emptyMap();
}
public Writer getObjectWriter(final Object object) throws IOException {
Writer writer;
if(object instanceof String) {
writer = stringWriters.get(object);
} else {
writer = objectWriters.get(object);
}
if (writer == null) {
writer = classWriters.get(object.getClass());
}
return writer;
}
public Object readObject(final Unmarshaller unmarshaller) throws IOException, ClassNotFoundException {
int idx = unmarshaller.readUnsignedByte();
if (idx >= extById.length) {
throw new InvalidObjectException("ObjectTable " + this.getClass().getName() + " cannot find an object for object index " + idx);
}
return extById[idx].createExternal(Object.class, unmarshaller);
}
}