Java结合redis实现接口防重复提交

@PostMapping ( "/user/update" )

@ApiOperation (value = "修改用户信息" , notes = "修改用户信息" , tags = "user module" )

@AvoidReSubmit (expireTime = 1000 * 3 )

public void update( @RequestBody User user){

  userMapper.updateById(user);

}

// 定义自定义注解，设置注解参数默认值

package top.withu.gaof.freehope.annotate;

import java.lang.annotation.*;

/**

  * @author Gaofan

  * @date 2019年10月12日 下午2:54:45

  * @describe 防止重复提交

  */

@Target ({ ElementType.METHOD, ElementType.TYPE })

@Retention (RetentionPolicy.RUNTIME)

@Documented

public @interface AvoidReSubmit {

     /**

      * 失效时间，即可以第二次提交间隔时长

      * @return

      */

     long expireTime() default 30 * 1000L;

}

// 定义切面进行处理

package top.withu.gaof.freehope.aspect;

import org.aspectj.lang.JoinPoint;

import org.aspectj.lang.ProceedingJoinPoint;

import org.aspectj.lang.annotation.*;

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

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

import org.springframework.stereotype.Component;

import top.withu.gaof.freehope.annotate.AvoidReSubmit;

import javax.annotation.Resource;

import java.security.MessageDigest;

import java.security.NoSuchAlgorithmException;

import java.util.concurrent.TimeUnit;

/**

  * @Description: TODO

  * @Author: gaofan

  * @Date: 2019/10/12 16:10

  * @Copyright: 2019 HdhCmsTestblog.freehope.top Inc. All rights reserved.

  **/

@Aspect

@Component

public class AvoidResumitAspect {

     @Resource

     private RedisTemplate redisTemplate;

     /**

      * 定义匹配规则，以便于后续拦截直接拦截submit方法，不用重复表达式

      */

     @Pointcut (value = "@annotation(top.withu.gaof.freehope.annotate.AvoidReSubmit)" )

     public void submit() {

     }

     @Before ( "submit()&&@annotation(avoidReSubmit)" )

     public void doBefore(JoinPoint joinPoint, AvoidReSubmit avoidReSubmit) {

         // 拼装参数

         StringBuffer sb = new StringBuffer();

         for (Object object : joinPoint.getArgs()){

             sb.append(object);

         }

         String key = md5(sb.toString());

         long expireTime = avoidReSubmit.expireTime();

         ValueOperations valueOperations = redisTemplate.opsForValue();

         Object object = valueOperations.get(key);

         if ( null != object){

             throw new RuntimeException( "您已经提交了请求，请不要重复提交哦！" );

         }

         valueOperations.set(key, 1 , expireTime, TimeUnit.MILLISECONDS);

     }

     @Around ( "submit()&&@annotation(avoidReSubmit)" )

     public Object doAround(ProceedingJoinPoint proceedingJoinPoint, AvoidReSubmit avoidReSubmit) throws Throwable {

         System.out.println( "环绕通知：" );

         Object result = null ;

         result = proceedingJoinPoint.proceed();

         return result;

     }

     @After ( "submit()" )

     public void doAfter() {

         System.out.println( "******拦截后的逻辑******" );

     }

     private String md5(String str){

         if (str == null || str.length() == 0 ) {

             throw new IllegalArgumentException( "String to encript cannot be null or zero length" );

         }

         StringBuffer hexString = new StringBuffer();

         try {

             MessageDigest md = MessageDigest.getInstance( "MD5" );

             md.update(str.getBytes());

             byte [] hash = md.digest();

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

                 if (( 0xff & hash[i]) < 0x10 ) {

                     hexString.append( "0" + Integer.toHexString(( 0xFF & hash[i])));

                 } else {

                     hexString.append(Integer.toHexString( 0xFF & hash[i]));

                 }

             }

         } catch (NoSuchAlgorithmException e) {

             e.printStackTrace();

         }

         return hexString.toString();

     }

}