OncePerRequestFilter自定义效验过滤器

注册效验过滤器

package com.axzhengxin.platform.common.filter;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;


@Slf4j
@Configuration
public class CustomizedFilterConfig {

    private static final String[] CAPTCHA_FILER_URL_PATTERNS = new String[]{"/getPhoneCode"};

    private static final String[] PHONE_CODE_FILTER_URL_PATTERNS = new String[]{"/verifyPhoneCode"};

    @Bean
    public FilterRegistrationBean<CaptchaFilter> captchaFilterFilterRegistration(@Autowired CaptchaFilter captchaFilter) {
        FilterRegistrationBean<CaptchaFilter> registration = new FilterRegistrationBean<>();
        registration.setFilter(captchaFilter);
        registration.addUrlPatterns(CAPTCHA_FILER_URL_PATTERNS);
        registration.setName("captchaFilter");
        registration.setOrder(FilterRegistrationBean.LOWEST_PRECEDENCE);
        log.info("注册图形验证码校验过滤器-成功");
        return registration;
    }

}

效验过滤器实现

package com.axzhengxin.platform.common.filter;

import cn.hutool.extra.servlet.ServletUtil;
import com.axzhengxin.platform.common.service.ICaptchaService;
import com.axzhengxin.platform.train.util.JacksonUtil;
import com.ruoyi.common.constant.Constants;
import com.ruoyi.common.core.domain.AjaxResult;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.filter.OncePerRequestFilter;

import javax.servlet.FilterChain;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.Map;


@Slf4j
@Component
public class CaptchaFilter extends OncePerRequestFilter {

    @Autowired
    private ICaptchaService captchaService;

    @Override
    protected void doFilterInternal(HttpServletRequest request,
                                    HttpServletResponse response,
                                    FilterChain filterChain) throws ServletException, IOException {
        Map<String, String> queryParams = ServletUtil.getParamMap(request);
        if (queryParams.containsKey(Constants.CAPTCHA_UUID_PARAM_NAME)
                && queryParams.containsKey(Constants.CAPTCHA_CODE_PARAM_NAME)) {
            String uuid = queryParams.get(Constants.CAPTCHA_UUID_PARAM_NAME);
            String code = queryParams.get(Constants.CAPTCHA_CODE_PARAM_NAME);
            AjaxResult result = captchaService.verifyImageCode(uuid, code, Boolean.TRUE);
            if (AjaxResult.ifSuccessResult(result)) {
                filterChain.doFilter(request, response);
            } else {
                ServletUtil.write(response, JacksonUtil.toJsonString(result), "application/json;charset=UTF-8");
            }
        } else {
            AjaxResult result = AjaxResult.error("图形验证码不能为空");
            ServletUtil.write(response, JacksonUtil.toJsonString(result), "application/json;charset=UTF-8");
        }
    }
}