结合SpringBoot模仿WebServlet中的Filter实现责任链模式

责任链模式

实现对一个对象进行多阶段的处理在调用链中可以随时决定是否调用下一个过滤器从而提前结束调用链

定义一个过滤链对象

@Component
/**
这里定义为多实例是为了保证多线程调用时不会初始化其他线程正在调用的过滤链偏移量
*/
@Scope(value = "prototype")
public class LogFilterChain {
    private List<LogFilter> filters = new ArrayList<>();
    /**
     * The int which is used to maintain the current position in the filter chain.
     */
    private int pos = 0;
    /**
     * The int which gives the current number of filters in the chain.
     */
    private int n = 0;

    public void doFilter( String key) {
        if (pos < n) {
            LogFilter logFilter = filters.get(pos++);
            logFilter.doFilter( key, this);
        }
    }

    public List<LogFilter> getFilters() {
        return filters;
    }

    @Autowired
    public void setFilters(List<LogFilter> filters) {
        this.filters = filters;
        this.n = filters.size();
    }

    public void release() {
        this.n = filters.size();
        this.pos = 0;
    }
}

定义一个接口来表示处理某个对象的方法

public interface LogFilter {
     void doFilter(String key, LogFilterChain chain);
}

定义实现类实现具体的处理方式

这里测试在不同的实现类中用不同的logger来输出字符串

• 使用warn级别输出日志

@Slf4j
@Component
public class WarnLogFilter  implements LogFilter {
    @Override
    public void doFilter( String key, LogFilterChain chain) {
    	// 如果是warn开头的字符串那就输出key 并退出调用链 否则调用下一个责任链中的处理类
        if(key.startsWith("{warn}")){
            log.warn(key);
        }else{
         // 调用下一个责任链中的处理类
          chain.doFilter(key,chain);
        }
    }
}

• 使用info级别输出字符串

@Slf4j
@Component
public class InfoLogFilter  implements LogFilter {
    @Override
    public void doFilter( String key, LogFilterChain chain) {
    	// 如果是info开头的字符串那就输出key 并退出调用链 否则调用下一个责任链中的处理类
        if(key.startsWith("{info}")){
            log.info(key);
        }else{
        // 调用下一个责任链中的处理类
            chain.doFilter(key,chain);
        }
    }
}

测试过滤器模式

@SpringBootTest
public class OaSearchApplicationTests {


    @Autowired
    private LogFilterChain chain;

    @Test
    @DisplayName("测试责任链模式")
    void contextLoads() throws IOException{
        ArrayList<String> strings = Lists.newArrayList("{warn}warn输出了", "{info}info输出了");
        for (String string : strings) {
            chain.doFilter(string);
            chain.release();
        }
        
    }
}

注意! 在SpringMVC中Controller是单例对象使用过滤器对象会有线程安全问题所以应该使用下面的方式使我们获取的chain是多实例的 或者 把mvc的注入方式修改成多例模式

@RestController
@RequestMapping("/api/test")
@RequiredArgsConstructor
public class SearchController {
	// 不能使用该方式获取bean
   //  private final LogFilterChain chain;
    public void test(List<String> key){
    // 为了保证chain在多线程中是线程安全的
	 LogFilterChain chain =
	  SpringConextHoler.getBean(LogFilterChain.class);
	 for (String string : strings) {
            chain.doFilter(string);
            chain.release();
        }
	}
}

参考 tomcat中的 ApplicationFilterChain这个类