手動實現 SpringMVC

先看整體架構

1712167-20190617103913942-187681576.png

因爲我們這個 MVC 框架要依賴 IOC/DI 容器,所以我們在 pom 文件裏要將自己的 Spring 框架引入進來。

1712167-20190617103928654-2103977636.png

實現 MVC 的整體功能

首先我們定義兩個 MVC 專用的註解,RequestMapping 用來做 url 匹配,RequestParam 做參數轉換:

package com.paul.annotation;import java.lang.annotation.*;

@Target({ElementType.TYPE,ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface RequestMapping {    String value() default "";
}
package com.paul.annotation;import java.lang.annotation.*;@Target(ElementType.PARAMETER)@Retention(RetentionPolicy.RUNTIME)@Documentedpublic @interface RequestParam {    String value() default "";
}

我們知道 SpringMVC 的核心是 DispatcherServlet,用來做核心路由控制,我們也定義這樣一個類,並且在初始化方法裏初始化一個 IOC/DI 容器,看過前面文章的同學應該知道,初始化容器後我們已經將 Bean 放到容器中而且完成了依賴注入。

package com.paul.servlet;import com.paul.annotation.RequestMapping;import com.paul.annotation.RequestParam;import org.springframework.ioc.annotation.Controller;import org.springframework.ioc.bean.AnnotationApplicationContext;import org.springframework.ioc.bean.ApplicationContext;import javax.servlet.ServletConfig;import javax.servlet.ServletException;import javax.servlet.ServletRequest;import javax.servlet.ServletResponse;import javax.servlet.http.HttpServlet;import javax.servlet.http.HttpServletRequest;import javax.servlet.http.HttpServletResponse;import java.io.IOException;import java.lang.annotation.Annotation;import java.lang.reflect.InvocationTargetException;import java.lang.reflect.Method;import java.util.Map;import java.util.concurrent.ConcurrentHashMap;//實現 servlet,和我們以前使用 servlet 一樣。public class DispatcherServlet extends HttpServlet {    // 完整路徑和 方法的 mapping
    private Map<String,Object> handleMapping = new ConcurrentHashMap<>();    // 類路徑和controller 的 mapping
    private Map<String,Object> controllerMapping = new ConcurrentHashMap<>();
    
    private Map<String,Object> beanFactory = new ConcurrentHashMap<>();

    @Override
    public void init(ServletConfig config) throws ServletException {        super.init(config);        //實例化 IOC 容器
        ApplicationContext ctx = new AnnotationApplicationContext("applicationContext.xml");        this.beanFactory = ((AnnotationApplicationContext) ctx).beanFactory;        //上一步已經完成了 Controller,service,respostry,autowired 等註解的掃描和注入
        //遍歷容器,將 requestmapping 註解的路徑和對應的方法以及 contoller 實例對應起來

        for(Map.Entry<String,Object> entry:beanFactory.entrySet()){            Object instance = entry.getValue();
            Class<?> clazz = instance.getClass();            if(clazz.isAnnotationPresent(Controller.class)){
                RequestMapping requestMapping = clazz.getAnnotation(RequestMapping.class);                String classPath = requestMapping.value();
                Method[] methods = clazz.getMethods();                for(Method method:methods){                    if(method.isAnnotationPresent(RequestMapping.class)){
                        RequestMapping requestMapping2 = method.getAnnotation(RequestMapping.class);                        String methodPath = requestMapping2.value();                        String requestPath = classPath + methodPath;
                        handleMapping.put(requestPath,method);
                        controllerMapping.put(requestPath,instance);
                    }else{                        continue;
                    }

                }
            }else{                continue;
            }
        }
    }


    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {        this.doPost(req, resp);
    }    //根據上一步獲取到的 mapping,根據 url 找到對應的 controller 和方法去執行。
    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {        String uri = req.getRequestURI();   //   /paul-mvc/com.paul.controller/method-com.paul.controller
        String context = req.getContextPath();  // /paul-vmc
        String path = uri.replace(context,"");  // /com.paul.controller/method-com.paul.controller
        Method m = (Method) handleMapping.get(path);        //從容器裏拿到controller 實例
        Object instance = controllerMapping.get(path);        Object[] args =  handle(req,resp,m);        for (Object a:args){
            System.out.println("Object:"+a);
        }        try {
            m.invoke(instance,args);
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        } catch (InvocationTargetException e) {
            e.printStackTrace();
        }


    }


    private static Object[] handle(HttpServletRequest req, HttpServletResponse resp,Method method){        //拿到當前執行的方法有哪些參數
        Class<?>[] paramClazzs = method.getParameterTypes();        //根據參數的個數,new 一個參數的數據
        Object[] args = new Object[paramClazzs.length];

        int args_i = 0;
        int index = 0;        for(Class<?> paramClazz:paramClazzs){            if(ServletRequest.class.isAssignableFrom(paramClazz)){
                args[args_i++] = req;
            }            if(ServletResponse.class.isAssignableFrom(paramClazz)){
                args[args_i++] = resp;
            }            //判斷requestParam  註解
            Annotation[] paramAns = method.getParameterAnnotations()[index];            if(paramAns.length > 0){
                System.out.println("my");                for(Annotation paramAn:paramAns){                    if(RequestParam.class.isAssignableFrom(paramAn.getClass())){
                        System.out.println("13mj");
                        RequestParam rp = (RequestParam) paramAn;
                        args[args_i++] = req.getParameter(rp.value());
                    }
                }
            }
            index ++;
        }        return  args;
    }

}

測試代碼

新建一個 WebDemo,Maven web 項目。

先來看我們需要測試的 Controller 和 Service。

@Controller@RequestMapping("/user")public class UserController {    @Autowired
    private UserService userService;    @RequestMapping("/query")    public void get(HttpServletRequest request, HttpServletResponse response, @RequestParam("name") String name,@RequestParam("age") String age){        try {
            PrintWriter pw = response.getWriter();
            String res = userService.query(name,age);
            pw.write(res);
        } catch (IOException e) {
            e.printStackTrace();
        }

    }
}
import org.springframework.ioc.annotation.Service;@Service("userService")public class UserServiceImpl implements UserService {    @Override
    public String query(String name, String age) {        return "name="+name+"age="+age;
    }
}

在 resources 目錄我們需要寫一個名字爲 applicationContext 的配置文件來指明掃描包路徑。

<?xml version="1.0" encoding="UTF-8"?><beans>
    <package-scan component-scan="com.paul.demo" /></beans>

在 web.xml 中配置我們自己的 DispatcherServlet。

<!DOCTYPE web-app PUBLIC
        "-//Sun Microsystems, Inc.//DTD Web Application 2.3//EN"
        "http://java.sun.com/dtd/web-app_2_3.dtd" ><web-app>
  <display-name>Archetype Created Web Application</display-name>
  <servlet>
    <servlet-name>DispatcherServlet</servlet-name>
    <servlet-class>com.paul.servlet.DispatcherServlet</servlet-class>
    <load-on-startup>0</load-on-startup>
  </servlet>

  <servlet-mapping>
    <servlet-name>DispatcherServlet</servlet-name>
    <url-pattern>/</url-pattern>
  </servlet-mapping></web-app>

在瀏覽器中測試結果:

1712167-20190617103959962-2013309167.png

結果和我們想的一樣。


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章