輕松實作java攔截器+自定義注解

目錄和概述

概述

• 內容：攔截器的使用很簡單，定義一個自己的攔截器，向配置中添加一下就可以使用，為了方便，之后又引入了注解，本文就將用簡潔的代碼構建一個springboot的攔截器，
• 假設需求：訪問專案的controller是都要進行"token驗證"，除了某些像登錄之類的方法，
• 專案結構：
  • TokenInterceptor.java 自定義攔截器
  • InterceptorConfig.java 添加攔截器進入專案
  • NoNeedToken.java 自定義注解
  • TestController.java 測驗介面
    

目錄

• 1、自定義攔截器
• 2、載入攔截器
• 3、測驗成功
• 4、使用注解（Annotation）進行選擇
• 5、常見問題

1、自定義攔截器

import com.alibaba.fastjson.JSONObject;
import org.springframework.lang.Nullable;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.PrintWriter;
import java.util.Arrays;
import java.util.List;

@Component
public class TokenInterceptor implements HandlerInterceptor {
    // 假設現在的token有如下資料
    List<String> tokenList = Arrays.asList("111", "222", "333");
    // 這個方法是在訪問介面之前執行的，我們只需要在這里寫驗證登陸狀態的業務邏輯，就可以在用戶呼叫指定介面之前驗證登陸狀態了
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        // 設定回傳為json格式，使用UTF-8
        response.setCharacterEncoding("UTF-8");
        response.setContentType("application/json; charset=utf-8");
        String token = request.getHeader("token");
        PrintWriter out;
        // 之后寫你的判斷邏輯：return true是通過攔截器，可以繼續訪問controller，return false是不通過
        if (token == null || !tokenList.contains(token)) {
            // 如果失敗了回傳{state:"false", msg:"token is null or wrong"}
            JSONObject res = new JSONObject();
            res.put("state","false");
            res.put("msg","token is null or wrong");
            out = response.getWriter();
            out.append(res.toString());
            return false;
        }
        // 否則回傳true 進入controller
        return true;
    }
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, @Nullable ModelAndView modelAndView) throws Exception {
    }

    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, @Nullable Exception ex) throws Exception {
    }
}

2、載入攔截器

import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
import javax.annotation.Resource;

@Configuration
public class InterceptorConfig  implements WebMvcConfigurer {
    @Resource
    TokenInterceptor tokenInterceptor;
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        // 設定所有的路徑都要進行攔截，除了/test/login
        registry.addInterceptor(tokenInterceptor).addPathPatterns("/**")
                .excludePathPatterns("/test/login");
    }
}

3、測驗成功

import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.HashMap;
import java.util.Map;

@RestController
// canload用于檢測攔截器是否正常作業
@RequestMapping("/test")
public class TestController {
    @RequestMapping(value = "https://www.cnblogs.com/canload")
    public Map<String, Object> canLoad(){
        Map<String, Object> res = new HashMap<>();
        res.put("state", "ok");
        res.put("msg", "you can load");
        return res;
    }
    // login模擬登錄時token回傳
    @RequestMapping(value = "https://www.cnblogs.com/login")
    public String login(){
        return "111";
    }
}

可以看到如下結果
①直接登錄成功

②直接訪問canload失敗

③使用postman，帶上token進行訪問成功

4、使用注解（Annotation）進行選擇

NoNeedToken.java

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Target({ElementType.METHOD, ElementType.TYPE}) //注解的范圍是類、介面、列舉的方法上
@Retention(RetentionPolicy.RUNTIME)//被虛擬機保存，可用反射機制讀取
public @interface NoNeedToken {
}

回應的更改自定義攔截器的 preHandle方法，和攔截器配置

① 自定義攔截器

public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
    response.setCharacterEncoding("UTF-8");
    response.setContentType("application/json; charset=utf-8");
    String token = request.getHeader("token");
    PrintWriter out;
    // 對于注解的判斷---------------------------------------------------
    HandlerMethod handlerMethod = (HandlerMethod) handler;
    if(handlerMethod.getMethodAnnotation(NoNeedToken.class)!=null || handlerMethod.getBeanType().isAnnotationPresent(NoNeedToken.class)){
        // 如果自己擁有NoNeedToken標注或者所屬的class擁有NoNeedToken 就直接放行
        return true;
    }
    //------------------------------------------------------------------
    if (token == null || !tokenList.contains(token)) {
        JSONObject res = new JSONObject();
        res.put("state","false");
        res.put("msg","token is null or wrong");
        out = response.getWriter();
        out.append(res.toString());
        return false;
    }
    return true;
}

② 去除攔截器配置中的去除項

public void addInterceptors(InterceptorRegistry registry) {
    registry.addInterceptor(tokenInterceptor).addPathPatterns("/**");
            // .excludePathPatterns("/test/login");
}

③ 在測驗的login方法上添加注釋

@RequestMapping(value = "https://www.cnblogs.com/login")
@NoNeedToken
public String login(){
    return "111";
}

修改完成之后效果和上面相同

5、常見問題

1.這個自定義注解的哪些@Target，@Retention，@Documented是什么意思

https://blog.csdn.net/zt15732625878/article/details/100061528，這篇文章寫的很詳細了

2.為什么我的注解加了，還是被攔截了

一般出現這種情況是類與方法的獲取注解方式不同，在本文中作者使用了下面的一句話來表述

handlerMethod.getMethodAnnotation(NoNeedToken.class)!=null || handlerMethod.getBeanType().isAnnotationPresent(NoNeedToken.class)

其中handlerMethod.getMethodAnnotation(NoNeedToken.class)!=null是對于方法的
而handlerMethod.getBeanType().isAnnotationPresent(NoNeedToken.class)是對于類的

結語：希望對大家有幫助

標籤：Java

上一篇：1、初始Java面向物件

下一篇：【AKKA干貨】AKKA-HTTP（JAVA版）踩坑記