package com.ldr.common.filter;
import java.io.IOException;
import java.util.Objects;
import java.util.concurrent.TimeUnit;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.ValueOperations;
import org.springframework.stereotype.Component;
import org.springframework.web.filter.OncePerRequestFilter;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.ldr.common.config.Result;
import jakarta.servlet.FilterChain;
import jakarta.servlet.ServletException;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;
/**
* @summary 【重复提交验证过滤】
* @author 高振中
* @date 2024-01-09 14:20:20
**/
@Slf4j
@Component
public class SumitFilter extends OncePerRequestFilter {
private static final String MESSAGE = "点的太快了.....发生了重复提交";
private static final int CODE = 200;
@Autowired
private RedisTemplate<String, Integer> template;
@Autowired
private ObjectMapper mapper;
@Override
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException {
ValueOperations<String, Integer> opsForValue = template.opsForValue();
response.setCharacterEncoding("utf-8");
response.setHeader("Content-Type", "application/json");
String key = request.getSession().getId() + "-" + request.getServletPath();
if (Objects.isNull(opsForValue.get(key))) {// 如果缓存中有这个url视为重复提交
opsForValue.set(key, 0, 1, TimeUnit.SECONDS);
filterChain.doFilter(request, response);
} else {
log.error("重复提交{}", key);
response.getWriter().write(mapper.writeValueAsString(Result.error(CODE, MESSAGE)));
}
}
}
网友评论