1. 程式人生 > >spring boot 使用 filter 過濾請求頭是否包含指定的資訊

spring boot 使用 filter 過濾請求頭是否包含指定的資訊

下面的例子是使用 filter 過濾 request 中是否包含指定的變數 token 。

思路:

在生成自定義的 token 時,把 token 存到快取或者資料庫中,我這裡是存到了 redis 中,使用的 key 是 "asdfKevin"

通過特定介面把 token 反饋到前端,比如登入、授權

前端把token 儲存起來,每次訪問需要校驗的介面都把 token 放置到請求的 header 引數中,我這裡的需要校驗介面規定為“/api/*”

後臺"/api/*"介面都註冊到 filter 中,完成校驗,校驗通過後繼續;不通過則返回錯誤資訊,不再繼續。

過濾器程式碼:


import java.io.IOException;

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 org.springframework.beans.factory.annotation.Autowired;

import com.kevin.crystal.redis.util.RedisUtil;

public class KevinTokenFilter implements Filter{
	
	@Autowired
	private RedisUtil redisUtil;

	@Override
	public void init(FilterConfig filterConfig) throws ServletException {
		// TODO Auto-generated method stub
		System.out.println("------------test token filter init --------------");
		
	}

	@Override
	public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
			throws IOException, ServletException {
		// TODO Auto-generated method stub
		System.out.println("------------test token filter doFilter --------------");
		if(request instanceof HttpServletRequest) {
			HttpServletRequest req = (HttpServletRequest) request;
			String token = req.getHeader("token");
			System.out.println("token = " + token);
			String t = redisUtil.get("asdfKevin");
			System.out.println("t = " + t);
			if(token != null && token.equals(t)) {
				System.out.println("通過了");
			}else {
				System.out.println("沒通過");
				return;
			}
		}
		chain.doFilter(request, response);
		
//		HttpServletRequest request, HttpServletResponse response
		
		
	}

	@Override
	public void destroy() {
		// TODO Auto-generated method stub
		System.out.println("------------test token filter destory--------------");
	}

	
}

 

redisUtil 是我自定義的工具類,比較簡單,不贅述

config java 類中:

    @Bean
    public Filter kevinTokenFilter() {
    	System.out.println("----------------test token filter-----------------");
    	return new KevinTokenFilter();
    }

    @Bean
    public FilterRegistrationBean tokenFilter() {
    	FilterRegistrationBean registration = new FilterRegistrationBean();
    	registration.setFilter(kevinTokenFilter());
    	registration.addUrlPatterns("/api/*");
//    	registration.addInitParameter("", "");
    	registration.setName("tokenFilter");
    	
    	return registration;
    }

前端無論是 ajax 還是 小程式,只要把指定的 token 放到 header 中就可以,如下小程式程式碼:

wx.request({
    url:"https://kevin.kevinkevin.com/api/check",
    header:{
        "token":app.globalData.token
    },
    success:function(res){
        console.log(res);
    }
});

到這裡使用 filter 校驗token 的辦法就梳理完了,有需求留評論吧,後續會提供原始碼包