SpringBoot：基于redis自定义注解实现后端接口防重复提交校验

 

一、添加依赖

 

        <dependency>

 

            <groupId>org.springframework.boot</groupId>

 

            <artifactId>spring-boot-starter-redis</artifactId>

 

            <version>1.4.4.RELEASE</version>

 

        </dependency>

 

        <dependency>

 

            <groupId>redis.clients</groupId>

 

            <artifactId>jedis</artifactId>

 

            <version>3.6.3</version>

 

        </dependency>

 

<dependency>

 

            <groupId>com.alibaba</groupId>

 

            <artifactId>fastjson</artifactId>

 

            <version>1.2.75</version>

 

        </dependency>

 

二、代码实现

 

1.构建可重复读取inputStream的request

 

package com.hl.springbootrepetitionsubmit.servlet;

 

import javax.servlet.*;

 

import javax.servlet.http.*;

 

import java.io.*;

 

import java.nio.charset.StandardCharsets;

 

/**

 

 * 构建可重复读取inputStream的request

 

 */

 

public class RepeatedlyRequestWrapper extends HttpServletRequestWrapper {

 

    private final String body;

 

    public RepeatedlyRequestWrapper（HttpServletRequest request, ServletResponse response） throws IOException {

 

        super（request）;

 

        request.setCharacterEncoding（"UTF-8"）;

 

        response.setCharacterEncoding（"UTF-8"）;

 

        body = getBodyString（request）;

 

    }

 

    @Override

 

    public BufferedReader getReader（） throws IOException {

 

        return new BufferedReader（new InputStreamReader（getInputStream（）））;

 

    }

 

    public String getBody（） {

 

        return body;

 

    }

 

    @Override

 

    public ServletInputStream getInputStream（） throws IOException {

 

        final ByteArrayInputStream bais = new ByteArrayInputStream（body.getBytes（StandardCharsets.UTF_8））;

 

        return new ServletInputStream（） {

 

            @Override

 

            public int read（） throws IOException {

 

                return bais.read（）;

 

            }

 

            @Override

 

            public int available（） throws IOException {

 

                return body.length（）;

 

            }

 

            @Override

 

            public boolean isFinished（） {

 

                return false;

 

            }

 

            @Override

 

            public boolean isReady（） {

 

                return false;

 

            }

 

            @Override

 

            public void setReadListener（ReadListener readListener） {

 

            }

 

        };

 

    }

 

    /**

 

     * 获取Request请求body内容

 

     *

 

     * @param request

 

     * @return

 

     */

 

    private String getBodyString（ServletRequest request） {

 

        StringBuilder sb = new StringBuilder（）;

 

        BufferedReader reader = null;

 

        try （InputStream inputStream = request.getInputStream（）） {

 

            reader = new BufferedReader（new InputStreamReader（inputStream, StandardCharsets.UTF_8））;

 

            String line = "";

 

            while （（line = reader.readLine（）） != null） {

 

                sb.append（line）;

 

            }

 

        } catch （IOException e） {

 

            e.printStackTrace（）;

 

        } finally {

 

            if （reader != null） {

 

                try {

 

                    reader.close（）;

 

                } catch （IOException e） {

 

                    e.printStackTrace（）;

 

                }

 

            }

 

        }

 

        return sb.toString（）;

 

    }

 

}

 

2.重写request

 

package com.hl.springbootrepetitionsubmit.filter;

 

import com.hl.springbootrepetitionsubmit.servlet.RepeatedlyRequestWrapper;

 

import org.springframework.http.MediaType;

 

import org.springframework.util.StringUtils;

 

import javax.servlet.*;

 

import javax.servlet.http.HttpServletRequest;

 

import java.io.IOException;

 

/**

 

 * 过滤器（重写request）

 

 */

 

public class RepeatableFilter implements Filter {

 

    @Override

 

    public void doFilter（ServletRequest request, ServletResponse response, FilterChain chain） throws IOException, ServletException {

 

        ServletRequest requestWrapper = null;

 

        //将原生的request变为可重复读取inputStream的request

 

        if （request instanceof HttpServletRequest

 

                && StringUtils.startsWithIgnoreCase（request.getContentType（）， MediaType.APPLICATION_JSON_VALUE）） {

 

            requestWrapper = new RepeatedlyRequestWrapper（（HttpServletRequest） request, response）;

 

        }

 

        if （null == requestWrapper） {

 

            chain.doFilter（request, response）;

 

        } else {

 

            chain.doFilter（requestWrapper, response）;

 

        }

 

    }

 

}