cn.yangjunda.servlet.DispatcherServlet Maven / Gradle / Ivy

Go to download
package cn.yangjunda.servlet;

import cn.yangjunda.annotation.*;
import cn.yangjunda.multipart.MultipartFile;
import cn.yangjunda.multipart.commons.CommonsMultipartFile;
import cn.yangjunda.util.*;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.IOException;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Parameter;
import java.math.BigDecimal;
import java.util.*;
import java.util.Map.Entry;


/**请求几种处理类
 * @author juanda
 *
 */
public class DispatcherServlet extends HttpServlet {
	private static final long serialVersionUID = 1378531571714153483L;
	
	/** 要扫描的包，只有在这个包下并且加了注解的才会呗扫描到 */
	private static String PACKAGE = null;
	
	private static final String CONTROLLER_KEY = "controller";
	
	private static final String METHOD_KEY = "method";

	private static final String ENTRY_KEY = "entry";
	
	/** 存放Controller中url和方法的对应关系，格式：{url:{controller:实例化后的对象,method:实例化的方法}} */
	private static Map> urlMethodMapping = new HashMap>();

	private Map handlerMapping = new HashMap<>();

	private Set> classes;

	private Configuration configuration;


	// 上传文件存储目录
	private static final String UPLOAD_DIRECTORY = "upload";

	// 上传配置
	private static final int MEMORY_THRESHOLD   = 1024 * 1024 * 3;  // 3MB
	private static final int MAX_FILE_SIZE      = 1024 * 1024 * 40; // 40MB
	private static final int MAX_REQUEST_SIZE   = 1024 * 1024 * 50; // 50MB


	public DispatcherServlet() {  
        super();  
    } 
	
	/**
	 * 初始化方法，用于实例化扫描到的对象，并做注入和url映射（注：该方法逻辑上已经判断了，只执行一次）
	 */
	@Override
	public void init(ServletConfig config) throws ServletException {

		// 只处理一次
		if (urlMethodMapping.size() > 0) {
			return;
		}
		long startTime=System.currentTimeMillis();
		System.err.println(new Date()+" cn.yangjunda.servlet.DispatcherServlet  init");
		System.err.println("INFO: juanda-mvc 初始化开始");
		XMLReader xmlReader = new XMLReader();
		xmlReader.readXML();
		configuration = xmlReader.getConfiguration();
		System.err.println(new Date()+" cn.yangjunda.servlet.DispatcherServlet  init");
		System.err.println("INFO: juanda-mvc 读取XML配置文件");
		PACKAGE = configuration.getPackagePath();
		if(PACKAGE==null){
			System.err.println(new Date()+" cn.yangjunda.servlet.DispatcherServlet  init");
			System.err.println("INFO: juanda-mvc 读取properties配置文件");
			PACKAGE = PropertiesUtil.getProperty("packagePath");
		}
		if(StringUtils.isEmpty(PACKAGE)){
			throw new RuntimeException("配置文件中不存在packagePath");
		}
		// 开始扫描包下全部class文件
		System.err.println(new Date()+" cn.yangjunda.servlet.DispatcherServlet  init");
		System.err.println("INFO: juanda-mvc 开始扫描包");
		classes = ClassTools.getClasses(PACKAGE);
		
		// 存放Controller和Service的Map，格式：{beanName:实例化后的对象} 
		Map instanceNameMap = new HashMap();
		// 存放Service接口类型与接口实例对象的Map，格式：{Service.instance.class:实现类实例化后的对象} 
		Map, Object> instanceTypeMap = new HashMap, Object>();
		
		// 组装instanceMap
		buildInstanceMap(classes, instanceNameMap, instanceTypeMap);
		
		// 开始注入
		doIoc(instanceNameMap, instanceTypeMap);
		
		// 注入完之后开始映射url和method
		buildUrlMethodMapping(instanceNameMap, urlMethodMapping);
		long endTime=System.currentTimeMillis();
		System.err.println(new Date()+" cn.yangjunda.servlet.DispatcherServlet  init");
		System.err.println("INFO: juanda-mvc 初始化成功 持续时间："+(endTime-startTime)+"ms");
    }
	
	@Override  
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {  
        this.doPost(req, resp);  
    }  
  
    @Override  
    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {  
    	// 完整路径
    	String url = req.getRequestURI();
		// 跟路径
    	String path = req.getContextPath();
    	// 计算出method上配置的路径,如果用户写了多个"///"，只保留一个
    	String finallyUrl = url.replace(path, "").replaceAll("/+", "/");
    	
    	// 取出这个url对应的Controller和method
    	Map map = urlMethodMapping.get(finallyUrl);
    	if (map != null) {
			Method method = (Method) map.get(METHOD_KEY);
			Entry entry = (Entry) map.get(ENTRY_KEY);
			Class clazz = entry.getValue().getClass();
			if(!handleHttpRequestMethod(req,resp,method)){
				return;
			}
			Object object;
			try {
				String[] paramNames = ASMGetRealNameUtil.getMethodParameterNamesByAsm4(clazz,method);
				// 封装需要注入的参数
				List