美文网首页
java防止重复提交、防止重复点击

java防止重复提交、防止重复点击

作者: __简单点__ | 来源:发表于2022-09-06 11:54 被阅读0次

    1.引入redis

    <dependency> 

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

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

    </dependency>

    2.yml配置

    spring: 

     redis: 

         host: localhost

         port: 6379    

         password: 123456

    3.自定义注解

    import java.lang.annotation.*;

    @Target({ElementType.METHOD})

    @Retention(RetentionPolicy.RUNTIME)

    @Documented

    @Inherited

    public @interface RedisLock {

        int expire() default 5;

    }

    4. 定义切面类

    import com.alibaba.fastjson.JSONObject;

    import com.google.common.collect.Lists;

    import com.heshu.sz.blockchain.utonhsbs.common.utils.MyMD5Util;

    import com.heshu.sz.blockchain.utonhsbs.common.utils.SecurityUtils;

    import com.heshu.sz.blockchain.utonhsbs.common.utils.ip.IpUtils;

    import com.heshu.sz.blockchain.utonhsbs.framework.interceptor.annotation.RedisLock;

    import com.heshu.sz.blockchain.utonhsbs.framework.system.domain.BaseResult;

    import lombok.extern.slf4j.Slf4j;

    import org.apache.commons.lang3.StringUtils;

    import org.aspectj.lang.ProceedingJoinPoint;

    import org.aspectj.lang.annotation.Around;

    import org.aspectj.lang.annotation.Aspect;

    import org.aspectj.lang.annotation.Pointcut;

    import org.aspectj.lang.reflect.MethodSignature;

    import org.springframework.beans.factory.annotation.Autowired;

    import org.springframework.beans.factory.annotation.Value;

    import org.springframework.context.annotation.Configuration;

    import org.springframework.data.redis.core.StringRedisTemplate;

    import org.springframework.data.redis.core.script.DefaultRedisScript;

    import org.springframework.data.redis.core.script.RedisScript;

    import org.springframework.web.context.request.RequestAttributes;

    import org.springframework.web.context.request.RequestContextHolder;

    import org.springframework.web.context.request.ServletRequestAttributes;

    import javax.servlet.http.HttpServletRequest;

    import java.lang.reflect.Method;

    import java.security.NoSuchAlgorithmException;

    import java.util.ArrayList;

    import java.util.List;

    /**

    * @author :xx

    * @description:

    * @date : 2022/7/1 9:41

    */

    @Slf4j

    @Aspect

    @Configuration

    public class RedisLockMethodInterceptor {

        @Value("${spring.profiles.active}")

        private String springProfilesActive;

        @Value("${spring.application.name}")

        private String springApplicationName;

        @Autowired

        private StringRedisTemplate stringRedisTemplate;

        @Pointcut("@annotation(com.heshu.sz.blockchain.utonhsbs.framework.interceptor.annotation.RedisLock)")

        public void point() {

        }

        @Around("point()")

        public Object doaround(ProceedingJoinPoint joinPoint) {

            MethodSignature signature = (MethodSignature) joinPoint.getSignature();

            Method method = signature.getMethod();

            RedisLock localLock = method.getAnnotation(RedisLock.class);

            try {

                String lockUniqueKey = getLockUniqueKey(signature, joinPoint.getArgs());

                Integer expire = localLock.expire();

                if (expire < 0) {

                    expire = 5;

                }

                ArrayList<String> keys = Lists.newArrayList(lockUniqueKey);

                String result = stringRedisTemplate.execute(setNxWithExpireTime, keys, expire.toString());

                if (!"ok".equalsIgnoreCase(result)) {//不存在

                    return BaseResult.error("不允许重复提交,请稍后再试");

                }

                return joinPoint.proceed();

            } catch (Throwable throwable) {

                throw new RuntimeException(throwable.getMessage());

            }

        }

        /**

        * lua脚本

        */

        private RedisScript<String> setNxWithExpireTime = new DefaultRedisScript<>(

                "return redis.call('set', KEYS[1], 1, 'ex', ARGV[1], 'nx');",

                String.class

        );

        /**

        * 获取唯一标识key

        *

        * @param methodSignature

        * @param args

        * @return

        */

        private String getLockUniqueKey(MethodSignature methodSignature, Object[] args) throws NoSuchAlgorithmException {

            //请求uri, 获取类名称,方法名称

            RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();

            ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) requestAttributes;

            HttpServletRequest request = servletRequestAttributes.getRequest();

    //        HttpServletResponse responese = servletRequestAttributes.getResponse();

            //获取用户信息

            String userMsg = SecurityUtils.getUsername(); //获取登录用户名称

            //1.判断用户是否登录

            if (StringUtils.isEmpty(userMsg)) { //未登录用户获取真实ip

                userMsg = IpUtils.getIpAddr(request);

            }

            String hash = "";

            List list = new ArrayList();

            if (args.length > 0) {

                String[] parameterNames = methodSignature.getParameterNames();

                for (int i = 0; i < parameterNames.length; i++) {

                    Object obj = args[i];

                    list.add(obj);

                }

                String param = JSONObject.toJSONString(list);

                hash = MyMD5Util.getMD5(param);

            }

            //项目名称 + 环境编码 + 获取类名称 + 加密参数

            String key = "lock:" + springApplicationName + ":" + springProfilesActive + ":" + userMsg + ":" + request.getRequestURI();

            if (StringUtils.isNotEmpty(key)) {

                key = key + ":" + hash;

            }

            return key;

        }

    5.调用

    @RedisLock 

     public void save(@RequestBody User user) {

        }

    文章来源:https://blog.csdn.net/qq_33454058/article/details/125516310?spm=1001.2014.3001.5506

    相关文章

      网友评论

          本文标题:java防止重复提交、防止重复点击

          本文链接:https://www.haomeiwen.com/subject/pfrgnrtx.html