packageorg.nf.framework.web;importorg.nf.framework.annotation.UrlMapping;importorg.nf.framework.utils.ScanUtil;importjavax.servlet.ServletConfig;importjavax.servlet.ServletException;importjavax.servlet.http.HttpServlet;importjavax.servlet.http.HttpServletRequest;importjavax.servlet.http.HttpServletResponse;importjava.io.IOException;importjava.lang.reflect.InvocationTargetException;importjava.lang.reflect.Method;importjava.util.HashMap;importjava.util.List;importjava.util.Map;/*** 描述:
* 总控制器
*
*@authorlance
* @create 2018-08-31 9:48
实现思路
1.进行全局扫描:将项目下所标识有注解的方法进行收集和解析(注意:扫描是一个
非常消耗性能的操作,这种操作应该只需要进行一次
)
2.进行类的解析
目的:找到哪些类中有哪些方法是标识有注解的,那么这些方法就是用于请求处理的方法,
解析的过程也是同样消耗性能的,思考一下,是否每次请求都要进行一次相同的解析操作
3.将解析的结果进行缓存
这一步非常重要,因为解析的结果将直接为后续的请求服务,在整个项目的运行同期是不能销毁的,
所以我们需要将这些信息缓存再存入到服务器中*/
public class DispatcherServlet extendsHttpServlet {private static final Map mapper = new HashMap<>();
@Overridepublic void init(ServletConfig config) throwsServletException {//1.全局的扫描(空代表全局,也可以给某个包的值)
List className = scan("");//2.解析类中有注解的方法
resolve(className);
System.out.println("DispatcherServlet初始化完成。。。。");
}
@Overrideprotected void service(HttpServletRequest req, HttpServletResponse resp) throwsServletException, IOException {//获取请求的url,这个path其实就对应了mapper中的key
String path =req.getServletPath();//根据key取出相应的method并调用处理请求//根据相应的key取出相应method,并调用方法
invokeTargetMethod(req,resp,mapper.get(path));
}/*** 进行类扫描
**/
private Listscan(String packageName){returnScanUtil.scanPackage(packageName);
}/*** 解析类中的注解方法
**/
private void resolve(ListclassName){for(String s : className) {try{//进行类加载
Class> clazz=Class.forName(s);//获得类中的所有的方法
Method[] methods =clazz.getMethods();for(Method method : methods) {if(method.isAnnotationPresent(UrlMapping.class)){//获得所有注解方法的上value值,也就是请求url的值
String url = method.getAnnotation(UrlMapping.class).value();//缓存到map集合中
mapper.put(url,method);
}
}
}catch(ClassNotFoundException e) {throw newRuntimeException(e.getMessage());
}
}
}/***调用具体某目标的方法
**/
private voidinvokeTargetMethod(HttpServletRequest request,HttpServletResponse response,Method method){try{//先获得该类中的实例
Object instance =method.getDeclaringClass().newInstance();//回调方法
method.invoke(instance,request,response);
}catch(InstantiationException e) {throw newRuntimeException(e.getMessage());
}catch(IllegalAccessException e) {throw newRuntimeException(e.getMessage());
}catch(InvocationTargetException e) {throw newRuntimeException(e.getMessage());
}
}
}