yangys
2024-10-30 25db770e621f1259b8d5b7fd514207f7481c2d0f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
package com.qianwen.smartman.common.filter;
 
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Stream;
import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import com.qianwen.smartman.common.constant.DictConstant;
 
public class PreviewFilter implements Filter {
    private static final List<String> KEYS = new ArrayList();
 
    static {
        KEYS.add(DictConstant.NOTICE_CODE);
        KEYS.add("process");
        KEYS.add("work");
        KEYS.add("token");
    }
 
    public void init(FilterConfig filterConfig) {
    }
 
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
        
        HttpServletRequest httpServletRequest = (HttpServletRequest)servletRequest;
        String path = httpServletRequest.getServletPath();
        String method = httpServletRequest.getMethod();
        String get = "GET";
        if (method.equals(get) || KEYS.stream().anyMatch(path::contains)) {
          filterChain.doFilter(servletRequest, servletResponse);
        } else {
          throw new RuntimeException("演示环境暂时无法操作!");
        } 
    }
 
    public void destroy() {
    }
}