org.jboss.as.ejb3.deployment.ApplicableMethodInformation Maven / Gradle / Ivy
The newest version!
/*
* JBoss, Home of Professional Open Source.
* Copyright 2012, Red Hat, Inc., 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.as.ejb3.deployment;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.jboss.as.ejb3.component.MethodIntf;
import static org.jboss.as.ejb3.EjbMessages.MESSAGES;
/**
* Metadata store for method level information that can be applied to a method via various different deployment
* descriptor styles
*
* @author Stuart Douglas
*/
public class ApplicableMethodInformation {
private final String componentName;
/**
* EJB 3.1 FR 13.3.7, the default transaction attribute is REQUIRED.
*/
private T defaultAttribute;
/**
* applied to all view methods of a given type
*/
private final Map perViewStyle1 = new HashMap();
/**
* Applied to all methods with a given name on a given view type
*/
private final PopulatingMap> perViewStyle2 = new PopulatingMap>() {
@Override
Map populate() {
return new HashMap();
}
};
/**
* Applied to an exact method on a view type
*/
private final PopulatingMap>> perViewStyle3 = new PopulatingMap>>() {
@Override
PopulatingMap> populate() {
return new PopulatingMap>() {
@Override
Map populate() {
return new HashMap();
}
};
}
};
/**
* Map of bean class to attribute
*/
private final Map style1 = new HashMap();
/**
* map of bean method name to attribute
*/
private final Map style2 = new HashMap();
public ApplicableMethodInformation(final String componentName, final T defaultAttribute) {
this.componentName = componentName;
this.defaultAttribute = defaultAttribute;
}
/**
* map of exact bean method to attribute
*/
private final PopulatingMap>> style3 = new PopulatingMap>>() {
@Override
PopulatingMap> populate() {
return new PopulatingMap>() {
@Override
Map populate() {
return new HashMap();
}
};
}
};
public T getAttribute(MethodIntf methodIntf, String className, String methodName, String... methodParams) {
assert methodIntf != null : "methodIntf is null";
assert methodName != null : "methodName is null";
assert methodParams != null : "methodParams is null";
ArrayKey methodParamsKey = new ArrayKey((Object[]) methodParams);
T attr = get(get(get(perViewStyle3, methodIntf), methodName), methodParamsKey);
if (attr != null)
return attr;
attr = get(get(perViewStyle2, methodIntf), methodName);
if (attr != null)
return attr;
attr = get(perViewStyle1, methodIntf);
if (attr != null)
return attr;
attr = get(get(get(style3, className), methodName), methodParamsKey);
if (attr != null)
return attr;
attr = get(style2, methodName);
if (attr != null)
return attr;
attr = get(style1, className);
if (attr != null)
return attr;
return defaultAttribute;
}
public List getAllAttributes(MethodIntf methodIntf, String className, String methodName, String... methodParams) {
assert methodIntf != null : "methodIntf is null";
assert methodName != null : "methodName is null";
assert methodParams != null : "methodParams is null";
final List ret = new ArrayList();
ArrayKey methodParamsKey = new ArrayKey((Object[]) methodParams);
T attr = get(get(get(perViewStyle3, methodIntf), methodName), methodParamsKey);
if (attr != null)
ret.add(attr);
attr = get(get(perViewStyle2, methodIntf), methodName);
if (attr != null)
ret.add(attr);
attr = get(perViewStyle1, methodIntf);
if (attr != null)
ret.add(attr);
attr = get(get(get(style3, className), methodName), methodParamsKey);
if (attr != null)
ret.add(attr);
attr = get(style2, methodName);
if (attr != null)
ret.add(attr);
attr = get(style1, className);
if (attr != null)
ret.add(attr);
return ret;
}
public T getViewAttribute(MethodIntf methodIntf, String methodName, String... methodParams) {
assert methodIntf != null : "methodIntf is null";
assert methodName != null : "methodName is null";
assert methodParams != null : "methodParams is null";
ArrayKey methodParamsKey = new ArrayKey((Object[]) methodParams);
T attr = get(get(get(perViewStyle3, methodIntf), methodName), methodParamsKey);
if (attr != null)
return attr;
attr = get(get(perViewStyle2, methodIntf), methodName);
if (attr != null)
return attr;
attr = get(perViewStyle1, methodIntf);
if (attr != null)
return attr;
return null;
}
/**
* @param className The class name
* @return The attribute that has been applied directly to the given class
*/
public T getClassLevelAttribute(String className) {
return style1.get(className);
}
/**
* Style 1 (13.3.7.2.1 @1)
*
* @param methodIntf the method-intf the annotations apply to or null if EJB class itself
* @param attribute
*/
public void setAttribute(MethodIntf methodIntf, String className, T attribute) {
if (methodIntf != null && className != null)
throw MESSAGES.bothMethodIntAndClassNameSet(componentName);
if (methodIntf == null) {
style1.put(className, attribute);
} else
perViewStyle1.put(methodIntf, attribute);
}
public T getAttributeStyle1(MethodIntf methodIntf, String className) {
if (methodIntf != null && className != null)
throw MESSAGES.bothMethodIntAndClassNameSet(componentName);
if (methodIntf == null) {
return style1.get(className);
} else {
return perViewStyle1.get(methodIntf);
}
}
/**
* Style 2 (13.3.7.2.1 @2)
*
* @param methodIntf the method-intf the annotations apply to or null if EJB class itself
* @param transactionAttribute
* @param methodName
*/
public void setAttribute(MethodIntf methodIntf, T transactionAttribute, String methodName) {
if (methodIntf == null)
style2.put(methodName, transactionAttribute);
else
perViewStyle2.pick(methodIntf).put(methodName, transactionAttribute);
}
public T getAttributeStyle2(MethodIntf methodIntf, String methodName) {
if (methodIntf == null)
return style2.get(methodName);
else
return perViewStyle2.pick(methodIntf).get(methodName);
}
/**
* Style 3 (13.3.7.2.1 @3)
*
* @param methodIntf the method-intf the annotations apply to or null if EJB class itself
* @param transactionAttribute
* @param methodName
* @param methodParams
*/
public void setAttribute(MethodIntf methodIntf, T transactionAttribute, final String className, String methodName, String... methodParams) {
ArrayKey methodParamsKey = new ArrayKey((Object[]) methodParams);
if (methodIntf == null)
style3.pick(className).pick(methodName).put(methodParamsKey, transactionAttribute);
else
perViewStyle3.pick(methodIntf).pick(methodName).put(methodParamsKey, transactionAttribute);
}
public T getAttributeStyle3(MethodIntf methodIntf, final String className, String methodName, String... methodParams) {
ArrayKey methodParamsKey = new ArrayKey((Object[]) methodParams);
if (methodIntf == null)
return style3.pick(className).pick(methodName).get(methodParamsKey);
else
return perViewStyle3.pick(methodIntf).pick(methodName).get(methodParamsKey);
}
public T getDefaultAttribute() {
return defaultAttribute;
}
public void setDefaultAttribute(final T defaultAttribute) {
this.defaultAttribute = defaultAttribute;
}
/**
* Makes an array usable as a key.
*
* @author Carlo de Wolf
*/
private static class ArrayKey {
private final Object[] a;
private final int hashCode;
private transient String s;
ArrayKey(Object... a) {
this.a = a;
this.hashCode = Arrays.hashCode(a);
}
@Override
public boolean equals(Object obj) {
if (!(obj instanceof ArrayKey))
return false;
return Arrays.equals(a, ((ArrayKey) obj).a);
}
@Override
public int hashCode() {
return hashCode;
}
@Override
public String toString() {
if (s == null)
this.s = Arrays.toString(a);
return s;
}
}
/**
* @author Carlo de Wolf
*/
private abstract static class PopulatingMap extends HashMap {
V pick(K key) {
V value = get(key);
if (value == null) {
value = populate();
put(key, value);
}
return value;
}
abstract V populate();
}
private static V get(Map map, K key) {
if (map == null)
return null;
return map.get(key);
}
}