/*
* JBoss, Home of Professional Open Source.
* Copyright 2008, Red Hat Middleware LLC, and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.proxy.ejb;
import java.lang.reflect.Method;
import javax.ejb.EJBHome;
import javax.ejb.EJBMetaData;
import javax.ejb.RemoveException;
import javax.ejb.Handle;
import javax.ejb.EJBHome;
import javax.ejb.EJBObject;
import javax.ejb.HomeHandle;
import org.jboss.proxy.ejb.handle.HomeHandleImpl;
import org.jboss.invocation.Invocation;
import org.jboss.invocation.InvocationContext;
import org.jboss.invocation.InvocationKey;
import org.jboss.invocation.InvocationType;
/**
* The client-side proxy for an EJB Home object.
*
* @author <a href="mailto:marc.fleury@jboss.org">Marc Fleury</a>
* @version $Revision: 81030 $
*/
public class HomeInterceptor
extends GenericEJBInterceptor
{
/** Serial Version Identifier. @since 1.6 */
private static final long serialVersionUID = 1333656107035759718L;
// Static --------------------------------------------------------
protected static final Object[] EMPTY_ARGS = {};
/** {@link EJBHome#getEJBMetaData} method reference. */
protected static final Method GET_EJB_META_DATA;
/** {@link EJBHome#getHomeHandle} method reference. */
protected static final Method GET_HOME_HANDLE;
/** {@link EJBHome#remove(Handle)} method reference. */
protected static final Method REMOVE_BY_HANDLE;
/** {@link EJBHome#remove(Object)} method reference. */
protected static final Method REMOVE_BY_PRIMARY_KEY;
/** {@link EJBObject#remove} method reference. */
protected static final Method REMOVE_OBJECT;
static
{
try
{
final Class empty[] = {};
final Class type = EJBHome.class;
GET_EJB_META_DATA = type.getMethod("getEJBMetaData", empty);
GET_HOME_HANDLE = type.getMethod("getHomeHandle", empty);
REMOVE_BY_HANDLE = type.getMethod("remove", new Class[] {
Handle.class
});
REMOVE_BY_PRIMARY_KEY = type.getMethod("remove", new Class[] {
Object.class
});
// Get the "remove" method from the EJBObject
REMOVE_OBJECT = EJBObject.class.getMethod("remove", empty);
}
catch (Exception e) {
e.printStackTrace();
throw new ExceptionInInitializerError(e);
}
}
// Attributes ----------------------------------------------------
// Constructors --------------------------------------------------
/**
* No-argument constructor for externalization.
*/
public HomeInterceptor() {}
// Public --------------------------------------------------------
/**
* InvocationHandler implementation.
*
* @throws Throwable Any exception or error thrown while processing.
*/
public Object invoke(Invocation invocation)
throws Throwable
{
InvocationContext ctx = invocation.getInvocationContext();
Method m = invocation.getMethod();
// Implement local methods
if (m.equals(TO_STRING))
{
return ctx.getValue(InvocationKey.JNDI_NAME).toString() + "Home";
}
else if (m.equals(EQUALS))
{
// equality of the proxy home is based on names...
Object[] args = invocation.getArguments();
String argsString = args[0] != null ? args[0].toString() : "";
String thisString = ctx.getValue(InvocationKey.JNDI_NAME).toString() + "Home";
return new Boolean(thisString.equals(argsString));
}
else if (m.equals(HASH_CODE))
{
return new Integer(this.hashCode());
}
// Implement local EJB calls
else if (m.equals(GET_HOME_HANDLE))
{
return new HomeHandleImpl(
(String)ctx.getValue(InvocationKey.JNDI_NAME));
}
else if (m.equals(GET_EJB_META_DATA))
{
return ctx.getValue(InvocationKey.EJB_METADATA);
}
else if (m.equals(REMOVE_BY_HANDLE))
{
// First get the EJBObject
EJBObject object =
((Handle) invocation.getArguments()[0]).getEJBObject();
// remove the object from here
object.remove();
// Return Void
return Void.TYPE;
}
else if (m.equals(REMOVE_BY_PRIMARY_KEY))
{
// Session beans must throw RemoveException (EJB 1.1, 5.3.2)
if(((EJBMetaData)ctx.getValue(InvocationKey.EJB_METADATA)).isSession())
throw new RemoveException("Session beans cannot be removed " +
"by primary key.");
// The trick is simple we trick the container in believe it
// is a remove() on the instance
Object id = invocation.getArguments()[0];
// Just override the Invocation going out
invocation.setId(id);
invocation.setType(InvocationType.REMOTE);
invocation.setMethod(REMOVE_OBJECT);
invocation.setArguments(EMPTY_ARGS);
return getNext().invoke(invocation);
}
// If not taken care of, go on and call the container
else
{
invocation.setType(InvocationType.HOME);
// Create an Invocation
return getNext().invoke(invocation);
}
}
}