栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Java

重构DispatcherServlet模块

Java 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

重构DispatcherServlet模块

package com.jt.dispatcherservlet;

import org.springframework.boot.test.context.SpringBootTest;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.junit.jupiter.api.Test;

class MyMappedInterceptor implements MyHandlerInterceptor{
    boolean before;
    public MyMappedInterceptor(boolean before){
        this.before=before;
    }
    @Override
    public boolean beforeMethod() {
        System.out.println("前置方法执行了");
        return before;
    }
}

interface MyHandlerInterceptor {

    default boolean beforeMethod() {
        return true;
    }

    default void afterMethod() {
        System.out.println("后置方法执行了");
    }
}

interface MyAbstractHandler {
    void invokeMethod();
}

class MyHandler implements MyAbstractHandler {
    String path;
    String msg;

    MyHandler(String path, String msg) {
        this.path = path;
        this.msg = msg;
    }

    void method() {
        System.out.println("我是method方法" + msg);
    }

    ;

    @Override
    public void invokeMethod() {
        method();
    }
}

interface MyHandlerMapping {
    MyHandlerExecutionChain getHandler(MyHttpServletRequest myHttpServletRequest);
}

class MyRequestMappingHandlerMapping extends MyAbstractHandlerMapping {
}

abstract class MyAbstractHandlerMapping implements MyHandlerMapping {
    private final List adaptedInterceptors = new ArrayList();

    public MyHandlerExecutionChain getHandler(MyHttpServletRequest myHttpServletRequest) {
        //模拟假数据
        MyAbstractHandler myhandler = new MyHandler("/resouce", "hello");
        MyHandlerExecutionChain chain = new MyHandlerExecutionChain(myhandler);
        Iterator it = adaptedInterceptors.iterator();
        while (it.hasNext()) {
            chain.addInterceptor(it.next());
        }
        return chain;
    }

    protected void initInterceptors() {
        this.adaptedInterceptors.add(new MyMappedInterceptor(false));
    }
}

class MyHandlerExecutionChain {
    private List interceptorList = new ArrayList<>();
    private final MyAbstractHandler handler;

    public MyHandlerExecutionChain(MyAbstractHandler handler) {
        this.handler = handler;
    }

    public void addInterceptor(MyHandlerInterceptor myInterceptor) {
        this.interceptorList.add(myInterceptor);
    }

    MyAbstractHandler getHandler() {
        return this.handler;
    }

    public boolean applyPreHandle() {
        boolean flag = true;
        Iterator var2 = this.interceptorList.iterator();

        while (var2.hasNext()) {
            boolean result = ((MyHandlerInterceptor) var2.next()).beforeMethod();
            flag = flag && result;
            if (!flag)
                return false;
        }
        return flag;
    }

    public void applyPostHandle() {
        Iterator var2 = this.interceptorList.iterator();

        while (var2.hasNext()) {
            ((MyHandlerInterceptor) var2.next()).afterMethod();
        }
    }
}

class MyHttpServletRequest {
    private String lookupPath;
}

class MyDispatcherServlet {
    private List myhandlerMappings = new ArrayList<>();

    public void initHandlerMappings() {
        //虚假数据
        MyRequestMappingHandlerMapping myRequestMappingHandlerMapping = new MyRequestMappingHandlerMapping();
        myRequestMappingHandlerMapping.initInterceptors();
        this.myhandlerMappings.add(myRequestMappingHandlerMapping);
    }

    public MyHandlerExecutionChain getHandler(MyHttpServletRequest myHttpServletRequest) {
        //遍历myhandlerMappings
        Iterator var2 = this.myhandlerMappings.iterator();

        while (var2.hasNext()) {
            MyHandlerMapping mapping = (MyHandlerMapping) var2.next();
            MyHandlerExecutionChain handler = mapping.getHandler(myHttpServletRequest);
            if (handler != null) {
                return handler;
            }
        }
        return null;
    }

    void doDispatch(MyHttpServletRequest myHttpServletRequest) {
        MyHandlerExecutionChain myHandlerExecutionChain = null;
        myHandlerExecutionChain = this.getHandler(myHttpServletRequest);
        if (!myHandlerExecutionChain.applyPreHandle()) {
            return;
        }
        myHandlerExecutionChain.getHandler().invokeMethod();
        myHandlerExecutionChain.applyPostHandle();
    }
}

@SpringBootTest
public class DispatcherServletTest {
    @Test
    public void method() {
        MyDispatcherServlet myDispatcherServlet = new MyDispatcherServlet();
        myDispatcherServlet.initHandlerMappings();
        myDispatcherServlet.doDispatch(new MyHttpServletRequest());
    }
}

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/424201.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号