利用redis实现分布式请求防重复提交

最新推荐文章于 2024-04-25 18:05:25 发布

weixin_33766805

最新推荐文章于 2024-04-25 18:05:25 发布

阅读量527

点赞数

文章标签：数据库爬虫移动开发

原文链接：https://my.oschina.net/u/2485283/blog/1859323

版权

2019独角兽企业重金招聘Python工程师标准>>>

1.自定义注解类Token

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Token {

    String flag() default "";
}

2.在需要拦截的路径上加自定义注解

@Token
@RequestMapping(value = "/pda/pick-task/list")
public RespJson getPickTask(@RequestParam("whNo") String whNo,
                            @RequestParam(value = "sourceType", required = false) Integer sourceType,
                            @RequestParam(value = "retrieveValue", required = false) String retrieveValue,
                            @RequestParam(value = "realTimePick", required = false, defaultValue = "0") int realTimePick,
                            @RequestParam(value = "taskGroupNo", required = false) String taskGroupNo) {
...

3.利用切面拦截请求

@Aspect
@Component
public class MethodInterceptor {
    private Logger logger = LoggerFactory.getLogger(MethodInterceptor.class);

/**
 * 重复请求拦截
 *
 * @param joinPoint
 * @return
 * @throws Throwable
 */
@Around("@annotation(com.haiziwang.kwms.common.annotation.Token)")
public Object repeatRequestAround(ProceedingJoinPoint joinPoint) throws Throwable {

    MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();
    Method currentMethod = joinPoint.getTarget().getClass().getMethod(methodSignature.getName(), methodSignature.getParameterTypes());

    IKMEMCache cache = KMemServiceImpl.getCache();
    //拼接签名
    StringBuilder signBuffer = new StringBuilder(currentMethod.getAnnotation(RequestMapping.class).value()[0]);
    Object[] args = joinPoint.getArgs();
    for (Object object : args) {
        if (object != null) {
            String str = "";
            try {
                str = JSONObject.toJSONString(object);
            } catch (Exception e) {

            }
            signBuffer.append("^").append(str);
        }
    }
    String tokenKey = signBuffer.toString();

    if (StringUtils.isNotBlank(cache.readFromHash(RedisKeyType.TOKEN.getName(), tokenKey))) {
        if (StringConstants.WCS.equals(currentMethod.getAnnotation(Token.class).flag())) {
            throw new WMS3CheckedException(WMS3ExceptionCode.WCS_REPEAT_REQUEST_EXCEPTION);
        }
        if (currentMethod.getReturnType() == RespJson.class) {
            throw new WMS3CheckedException(WMS3ExceptionCode.REPEAT_REQUEST_EXCEPTION);
        }
        if (currentMethod.getReturnType() == PageRespJson.class) {
            throw new WMS3CheckedException(WMS3ExceptionCode.REPEAT_REQUEST_PAGE_EXCEPTION);
        }
    }
    cache.write4Hash(RedisKeyType.TOKEN.getName(), tokenKey, "token");
    try {
        return joinPoint.proceed();
    } finally {
        cache.deleteHashFields(RedisKeyType.TOKEN.getName(), tokenKey);
    }
}

转载于:https://my.oschina.net/u/2485283/blog/1859323