其他分享
首页 > 其他分享> > 请求防重处理

请求防重处理

作者:互联网

1、自定义注解

@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public  @interface NoRepeatSubmit {

    /**
     * 默认1s钟以内算重复提交
     * @return
     */
    int lockTime() default 1000;
}

2、定义切面类

/**
 * 接口重复调用切面
 * 仅使用单机时
 */
@Aspect
@Component
@Slf4j
public class RepeatSubmitAspect {

     //定义切入点表达式
    @Pointcut("@annotation(noRepeatSubmit)")
    public void pointCut(NoRepeatSubmit noRepeatSubmit) {
    }

    @Around("pointCut(noRepeatSubmit)")
    public Object around(ProceedingJoinPoint pjp, NoRepeatSubmit noRepeatSubmit) throws Throwable {
        int lockMillisecond = noRepeatSubmit.lockTime();
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest();
        Assert.notNull(request, "request can not null");
        //此处可以用或者JSessionId
        String token = request.getHeader("Cookie");
        String path = getPath(request);
        Object[] args = pjp.getArgs();
        String key = getKey(token, path, args);
        boolean isSuccess = LocalCache.getInstance().tryLock(key, lockMillisecond);
        log.info("获取锁 key = [{}], clientId = [{}]", key,token);

        if (isSuccess) {
            log.info("获取锁 成功, key = [{}], clientId = [{}]", key,token);
            // 获取锁成功
            Object result;
            try {
                // 执行进程
                result = pjp.proceed();
            } finally {
                // 解锁
                LocalCache.getInstance().releaseLock(key);
                log.info("释放锁 成功, key = [{}], clientId = [{}]", key,token);
            }
            return result;
        } else {
            // 获取锁失败,认为是重复提交的请求
            log.info("获取锁 失败, key = [{}]", key);
            LayUiResult result = new LayUiResult();
            result.setResultStr("操作太快了,请稍后重试");
            result.setResult(false);
            return result;
        }

    }

    private String getKey(String token, String path, Object... args) {
        return token + "_" + path + "_" + getParam(args);
    }

    public <T> String getNotNullFieldStr(T origin) {
        if (origin == null)
            return "";
        StringBuffer stringBuffer = new StringBuffer();
        Field[] fields = origin.getClass().getDeclaredFields();
        for (int i = 0; i < fields.length; i++) {
            try {
                fields[i].setAccessible(true);
                String name = fields[i].getName();
                Object value = fields[i].get(origin);
                if (null != value) {
                    stringBuffer.append(name).append("=").append(value).append("&");
                }
                fields[i].setAccessible(false);
            } catch (Exception e) {
            }
        }
        return stringBuffer.toString();
    }

    public static boolean isWrapClass(Class clz) {
        try {
            return ((Class) clz.getField("TYPE").get(null)).isPrimitive();
        } catch (Exception e) {
            return false;
        }
    }

    /**
     * 生成md5
     *
     * @param base
     * @return
     */
    public static String getMD5(String base) {
        String md5 = DigestUtils.md5DigestAsHex(base.getBytes());
        return md5;
    }

    private String getParam(Object[] args) {
        if (args == null) {
            return "";
        }
        StringBuffer stringBuffer = new StringBuffer();
        for (Object object : args) {
            if (object != null) {
                if (isWrapClass(object.getClass())) {
                    stringBuffer.append(object.toString());
                } else {
                    stringBuffer.append(getMD5(getNotNullFieldStr(object)));
                }
            }
        }
        return getMD5(stringBuffer.toString());
    }

    private String getPath(HttpServletRequest request) {
        String uri = request.getRequestURI();
        String contextPath = request.getContextPath();
        if (StringUtils.length(contextPath) > 0) {
            uri = StringUtils.substring(uri, contextPath.length());
        }
        return uri;
    }
}

3、本地接口调用内存缓存

package com.idea.aop;

import lombok.extern.slf4j.Slf4j;

import java.util.Date;
import java.util.HashMap;
import java.util.Map;

/**
 * 本地接口调用内存缓存
 */
@Slf4j
public class LocalCache {
    private static LocalCache instance = new LocalCache();

    private LocalCache() {

    }

    public static LocalCache getInstance() {
        return instance;
    }

    /**
     * {key:{EXPIRED:时间戳,INSERT:时间戳},...}
     */
    private Map<String, Map<String, Object>> localMap = new HashMap<>();

    /**
     * 获取锁,当key存在且未过期时可以成功获得锁
     *
     * @param key
     * @param lockMillisecond
     * @return
     */
    public synchronized boolean tryLock(String key, long lockMillisecond) {
        //清理缓存
        clearExpired();
        //key已存在
        if (localMap.containsKey(key)) {
            //已过期
            if (expired(key, lockMillisecond)) {
                //已过期,从Map中移除
                localMap.remove(key);
                addCache(key, lockMillisecond);
                return true;
            } else {
                //未过期
                return false;
            }
        } else {
            //未在执行中
            addCache(key, lockMillisecond);
            return true;
        }
    }

    private void addCache(String key, long lockMillisecond) {
        Map map = new HashMap<>();
        map.put(EXPIRED, lockMillisecond);
        map.put(INSERT, new Date().getTime());
        localMap.put(key, map);
    }

    /**
     * 过期时间_KEY
     */
    /**
     * 创建时间_KEY
     */
    private static final String EXPIRED = "EXPIRED";
    private static final String INSERT = "INSERT";

    /**
     * 是否已过期
     *
     * @param key
     * @param lockMillisecond
     * @return
     */
    private synchronized boolean expired(Object key, long lockMillisecond) {
        Map<String, Object> map = localMap.get(key);
        //不存在 默认返回未过期
        if (map == null) {
            return true;
        }
        //数据接口调用时间戳
        long oldTime = (Long) map.get(INSERT);

        long now = new Date().getTime();

        //已过期:当前时间戳-(开始时间+过期时间)大于 0
        return (now - (oldTime + lockMillisecond)) > 0L;
    }

    /**
     * 清理Map中所有已过期的接口请求
     */
    private synchronized void clearExpired() {
        for (Map.Entry entry : localMap.entrySet()) {
            Object key = entry.getKey();
            Map<String, Object> map = localMap.get(key);
            long expired = (Long) map.get(EXPIRED);
            if (expired(key, expired)) {
                releaseLock(key);
            }
        }
    }

    /**
     * 释放锁
     *
     * @param key
     */
    public synchronized void releaseLock(Object key) {
        localMap.remove(key);
        log.info("本地缓存释放,Key:{}", key);
    }

}

 

标签:return,请求,处理,private,key,防重,lockMillisecond,public,String
来源: https://www.cnblogs.com/zouhong/p/16398806.html