net.hasor.web.invoker.RootInvokerCreater Maven / Gradle / Ivy
/*
* Copyright 2008-2009 the original author or authors.
*
* 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 net.hasor.web.invoker;
import net.hasor.core.AppContext;
import net.hasor.core.Settings;
import net.hasor.core.setting.SettingNode;
import net.hasor.utils.ClassUtils;
import net.hasor.utils.StringUtils;
import net.hasor.web.Invoker;
import net.hasor.web.InvokerCreator;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* @version : 2017-01-10
* @author 赵永春 ([email protected])
*/
class RootInvokerCreater implements InvokerCreator {
protected Map, InvokerCreator> createrMap = new HashMap<>();
protected Map, Class> extMapping = new HashMap<>();
public RootInvokerCreater(AppContext appContext) throws Exception {
Settings settings = appContext.getEnvironment().getSettings();
ClassLoader classLoader = appContext.getClassLoader();
//
// .寻找InvokerCreater扩展
Map, Class> extBinderMap = new HashMap<>();
SettingNode[] nodeArray = settings.getNodeArray("hasor.invokerCreatorSet.invokerCreator");
if (nodeArray != null && nodeArray.length > 0) {
for (SettingNode atNode : nodeArray) {
if (atNode == null) {
continue;
}
String binderTypeStr = atNode.getSubValue("type");
String binderImplStr = atNode.getValue();
if (StringUtils.isBlank(binderTypeStr) || StringUtils.isBlank(binderImplStr)) {
continue;
}
//
Class binderType = classLoader.loadClass(binderTypeStr);
Class binderImpl = classLoader.loadClass(binderImplStr);
if (!binderType.isInterface()) {
continue;
}
//
extBinderMap.put(binderType, binderImpl);
List> interfaces = ClassUtils.getAllInterfaces(binderType);
for (Class faces : interfaces) {
extBinderMap.put(faces, binderImpl);
}
}
}
// .创建扩展(extMapping来建立映射，避免重复创建InvokerCreater)
for (Map.Entry, Class> ent : extBinderMap.entrySet()) {
if (this.extMapping.containsKey(ent.getKey())) {
continue;
}
Class createrType = ent.getValue();
this.extMapping.put(ent.getKey(), createrType);
if (this.createrMap.containsKey(createrType)) {
continue;
}
InvokerCreator creater = (InvokerCreator) createrType.newInstance();
this.createrMap.put(createrType, creater);
}
}
//
@Override
public Invoker createExt(Invoker dataContext) {
//
Map, Object> extMap = new HashMap<>();
for (Map.Entry, InvokerCreator> ent : this.createrMap.entrySet()) {
Class extType = ent.getKey();
InvokerCreator creater = ent.getValue();
Object extObject = (creater != null) ? creater.createExt(dataContext) : null;
if (extType != null && extObject != null) {
extMap.put(extType, extObject);
}
}
//
Map, Object> supportMap = new HashMap<>();
supportMap.put(Invoker.class, dataContext);
for (Map.Entry, Class> ent : this.extMapping.entrySet()) {
Class key = ent.getKey();
Class value = ent.getValue();
Object obj = extMap.get(value);
if (obj != null) {
supportMap.put(key, obj);
}
}
//
ClassLoader classLoader = dataContext.getAppContext().getClassLoader();
Class[] apiArrays = supportMap.keySet().toArray(new Class[0]);
return (Invoker) Proxy.newProxyInstance(classLoader, apiArrays, new InvokerCreaterInvocationHandler(supportMap));
}
private static class InvokerCreaterInvocationHandler implements InvocationHandler {
private Map, Object> supportMap;
public InvokerCreaterInvocationHandler(Map, Object> supportMap) {
this.supportMap = supportMap;
}
@Override
public Object invoke(final Object proxy, final Method method, final Object[] args) throws Throwable {
if (method.getName().equals("toString")) {
StringBuilder builder = new StringBuilder();
builder = builder.append("count = ").append(this.supportMap.size()).append(" - [");
for (Class face : this.supportMap.keySet()) {
builder = builder.append(face.getName()).append(",");
}
if (builder.charAt(builder.length() - 1) == ',') {
builder = builder.deleteCharAt(builder.length() - 1);
}
builder.append("]");
return builder.toString();
}
//
Class declaringClass = method.getDeclaringClass();
Object target = this.supportMap.get(declaringClass);
if (target == null) {
throw new UnsupportedOperationException("this method is not support -> " + method);
}
//
try {
return method.invoke(target, args);
} catch (InvocationTargetException ex) {
throw ex.getTargetException();
}
}
}
}