All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • cn.potato.core.HandlerMethodArgsBuilder Maven / Gradle / Ivy

    Go to download

    以约定优于配置的思想，结合SpringMVC的一些优点(参数自动注入)，实现一个自己的简单MVC框架。

    The newest version!
    package cn.potato.core;

import cn.potato.helper.ConvertHelper;
import javassist.*;
import javassist.bytecode.CodeAttribute;
import javassist.bytecode.LocalVariableAttribute;
import javassist.bytecode.MethodInfo;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method;
import java.util.*;

/**
 * HandlerMethodArgsBuilder 通过反射实现控制器方法参数的注入，由于JDK 1.7 反射无法获取参数名称故采用 javassist。
 */
public class HandlerMethodArgsBuilder {
	private Map requestParamInfo;
	private Map, String> methodParamInfo;
	private List args;
	private static final Logger log = LoggerFactory.getLogger(HandlerMethodArgsBuilder.class);

	public HandlerMethodArgsBuilder(HttpServletRequest request, Method handler,
			Object controller) {
		initRequestParamInfo(request);
		initMethodParamInfo(controller, handler.getName());
	}

	public Object[] build() {
		inject();
		return args.toArray();
	}

	private void initRequestParamInfo(HttpServletRequest request) {
		requestParamInfo = new HashMap();
		Map parameterMap = request.getParameterMap();
		Set keySet = parameterMap.keySet();
		for (String key : keySet)
			if (key.contains(".")) {
				String[] fs = key.split("\\.");
				Map beanMap = new HashMap<>();
				for (String subKey : keySet) {
					if (subKey.startsWith(fs[0])) {
						String[] subFs = subKey.split("\\.");
						beanMap.put(subFs[1],parameterMap.get(subKey)[0]);
					}
				}
				requestParamInfo.put(fs[0], beanMap);
			} else {
				requestParamInfo.put(key, parameterMap.get(key)[0]);
			}
	}

	private void initMethodParamInfo(Object controller, String methodName) {
		methodParamInfo = new LinkedHashMap, String>();
		Class clazz = controller.getClass();
		try {
			ClassPool classPool = ClassPool.getDefault();
			classPool.insertClassPath(new ClassClassPath(clazz));
			CtClass ctClass = classPool.get(clazz.getName());
			CtMethod ctMethod = ctClass.getDeclaredMethod(methodName);

			MethodInfo methodInfo = ctMethod.getMethodInfo();
			CodeAttribute codeAttribute = methodInfo.getCodeAttribute();
			LocalVariableAttribute attr = (LocalVariableAttribute) codeAttribute
					.getAttribute("LocalVariableTable");

			CtClass[] parameterTypes = ctMethod.getParameterTypes();
			int pos = Modifier.isStatic(ctMethod.getModifiers()) ? 0 : 1;
			for (int i = 0; i < parameterTypes.length; i++)
				methodParamInfo.put(Class.forName(parameterTypes[i].getName()),
						attr.variableName(i + pos));
		} catch (Exception e) {
			throw new RuntimeException(e.getMessage());
		}
	}

	private void inject() {
		args = new ArrayList();
		Object object = null;
		Set> paramTypeClasses = methodParamInfo.keySet();
		try {
			for (Class clazz : paramTypeClasses) {
				String key = (String) methodParamInfo.get(clazz);
				if (requestParamInfo.containsKey(key)) {
					Object value = requestParamInfo.get(key);
					if ((value instanceof Map)) {//如果为Map型的转换为Bean
						@SuppressWarnings("unchecked")
						Map map = (Map) value;
						object  = ConvertHelper.mapConvertToBean(map, clazz);
					}else{
						object = ConvertHelper.convert(value, clazz);
					}
				}
				args.add(object);
			}
		} catch (Exception e) {
			log.error("Handler method args inject fail!", e);
		}
	}
}    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy