public class MyHttpServletRequestWrapper extends HttpServletRequestWrapper { 
    private final byte[] buff; 
    public MyHttpServletRequestWrapper(HttpServletRequest request) throws IOException {
        super(request);
        InputStream is = request.getInputStream();
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        byte[] b = new byte[1024];
        int len;
        while ((len = is.read(b)) != -1) {
            baos.write(b, 0, len);
        }
        buff = baos.toByteArray();
    }
 
    @Override
    public ServletInputStream getInputStream() throws IOException {
        final ByteArrayInputStream bais = new ByteArrayInputStream(buff);
        return new ServletInputStream() {
            @Override
            public int read() throws IOException {
                return bais.read();
            }
        };
    }
 
    @Override
    public BufferedReader getReader() throws IOException {
        return new BufferedReader(new InputStreamReader(getInputStream()));
    }
}

創(chuàng)建一個(gè)Filter，將該過濾器配置在項(xiàng)目中，為了調(diào)用備份的HttpServletRequestWrapper

public class MyRequestBodyFilter implements Filter { 
    @Override
    public void init(FilterConfig filterConfig) throws ServletException { 
    }
 
    @Override
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
        HttpServletRequest httpServletRequest = (HttpServletRequest) servletRequest;
        servletRequest = new MyHttpServletRequestWrapper(httpServletRequest);
        filterChain.doFilter(servletRequest, servletResponse);
    }
 
    @Override
    public void destroy() { 
    }
}

創(chuàng)建一個(gè)工具類，在攔截器中獲取request的值

public class RequestUtils { 
    public static String getRequestValue(HttpServletRequest request) throws IOException {
        StringBuffer sb = new StringBuffer();
        MyHttpServletRequestWrapper myHttpServletRequestWrapper = new MyHttpServletRequestWrapper(request);
        InputStream is = myHttpServletRequestWrapper.getInputStream();
        BufferedReader br = new BufferedReader(new InputStreamReader(is));
        String str;
        while ((str = br.readLine()) != null) {
            sb.append(str);
        }
        return sb.toString();
    } 
}

使用攔截器獲取Controller方法名和注解信息

在使用SpringMVC進(jìn)行項(xiàng)目的時(shí)候用到了權(quán)限驗(yàn)證。

表分為：

用戶表
角色表
資源表

用戶-角色-資源都是多對(duì)多的關(guān)系，驗(yàn)證無非就是收到請(qǐng)求后，在攔截器循環(huán)判斷用戶是否有權(quán)限執(zhí)行操作。

方法一：通過request獲得用戶的URL

再逐一循環(huán)判斷是否可以操作

只是這種方法很讓人難受。

方法二：通過用戶要訪問的方法來判斷是否有權(quán)限

preHandle方法中handler實(shí)際為HandlerMethod，（看網(wǎng)上說的有時(shí)候不是HandlerMethod），加個(gè)instanceof驗(yàn)證吧

可以得到方法名：h.getMethod().getName()
可以得到RequestMapping注解中的值：h.getMethodAnnotation(RequestMapping.class)
這種方法還是不太方便

方法三：自定義注解

自定義注解代碼：

@Retention(RUNTIME)
@Target(METHOD)
public @interface MyOperation {
    String value() default "";//默認(rèn)為空，因?yàn)槊质莢alue，實(shí)際操作中可以不寫"value="
}

Controller代碼：

@Controller("testController")
public class TestController {
    @MyOperation("用戶修改")//主要看這里
    @RequestMapping("test")
    @ResponseBody
    public String test(String id) {
        return "Hello,2018!"+id;
    }
}

攔截器的代碼：

@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)
        throws Exception {
    System.out.println("進(jìn)入攔截器");
    if(handler instanceof HandlerMethod) {
        HandlerMethod h = (HandlerMethod)handler;
        System.out.println("用戶想執(zhí)行的操作是:"+h.getMethodAnnotation(MyOperation.class).value());
        //判斷后執(zhí)行操作...
    }
    return HandlerInterceptor.super.preHandle(request, response, handler);
}

補(bǔ)充

在每個(gè)方法上面加注解太麻煩啦,可以在類上加注解

@Retention(RUNTIME)
@Target(TYPE)
public @interface MyOperation {
    String value() default "";
}
//攔截器中這樣獲得
h.getMethod().getDeclaringClass().getAnnotation(MyOperation.class);

我可以獲取requestMapping,不用創(chuàng)建自定義注解啊,值得注意的是,不要使用GetMapping等,要使用requestMapping。

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: