redis 增量爬虫 redis自增长

官网文档 https://docs.spring.io/spring-data/redis/docs/current/api/org/springframework/data/redis/support/atomic/package-summary.html

基本用法

private final RedisTemplate redisTemplate;
...
RedisAtomicInteger redisAtomicInteger = new RedisAtomicInteger(key, redisTemplate.getConnectionFactory());
//先自增，再获取自增之后的值
Integer code = redisAtomicInteger.incrementAndGet();
//初始化一个值，不初始化默认0，第一次获取永远是1
redisAtomicInteger.set(value);

类型也可以是 RedisAtomicDouble，RedisAtomicLong，默认0

典型应用

1，生成订单号

RedisAtomicInteger redisAtomicInteger = new RedisAtomicInteger(key, redisTemplate.getConnectionFactory())
//每日0点重置单号 
redisAtomicInteger.set(0);
//获取自增单号单号,业务标识 + 日期 + 自增数字
String orderNo = PRE + yyyyMMdd + redisAtomicInteger.incrementAndGet();

2，抢购库存

RedisAtomicInteger redisAtomicInteger = new RedisAtomicInteger(key, redisTemplate.getConnectionFactory())
//抢购开始前初始化总库存
redisAtomicInteger.set(500);
//活动开始每下一单原子性的-1
int nowNum = redisAtomicInteger.decrementAndGet();
if(nowNum<0){return "已经卖完了";}

3，幂等接口控制

比如用户抢答，或者是提交某些插入操作，只允许一次，需要避免重复操作

private final IdempotentComponent idempotentComponent;
...
//操作唯一标识
String lockKey = 操作人+订单类型+店铺编码+日期;
//11秒内只有第一次提交生效
boolean check = idempotentComponent.createIdempotentTally(lockKey,11L, TimeUnit.SECONDS);
if (!check) {
    return Result.failure(ResultCodeEnum.OPERATION_FAILED, "请勿重复执行！！");
}

import com.joybo.common.entity.IdempotentProperties;
import org.apache.commons.lang3.StringUtils;
import org.redisson.api.RAtomicLong;
import org.redisson.api.RedissonClient;
import org.springframework.stereotype.Component;

import java.util.concurrent.TimeUnit;

/**
 * @Description 幂等接口/方法
 * 对于⼀个⽅法/接⼝/函数采⽤同⼀个数据请求执⾏任意次数(>=2)⽽对数据只产⽣0次或1次变更。
 * 则认为该接⼝/⽅法/函数具有幂等性
 * @Date 2020/9/21 18:15
 * @Author yule
 */
@Component
public class IdempotentComponent {

    //只是为了获取应用名，避免多个服务key撞车
    private final IdempotentProperties idempotentProperties;
    private final RedissonClient redissonClient;

    private final String IDEMPOTENT_KEY = "idempotent_key:";

    public IdempotentComponent(RedissonClient redissonClient, IdempotentProperties idempotentProperties) {
        this.redissonClient = redissonClient;
        this.idempotentProperties = idempotentProperties;
    }

    /**
     * redis实现幂等控制
     * @param idempotentKey 幂等键
     * @param idempotentTime 时间
     * @param timeUnit 单位
     * @return
     */
    public boolean createIdempotentTally(String idempotentKey, Long idempotentTime, TimeUnit timeUnit) {
        idempotentKey = StringUtils.join(idempotentProperties.getApplicationName(), IDEMPOTENT_KEY, idempotentKey);
        RAtomicLong atomicLong = redissonClient.getAtomicLong(idempotentKey);
        if (atomicLong.isExists()) {
            return false;
        }
        long idempotentCount = atomicLong.incrementAndGet();
        if (idempotentCount == 1) {
            // 保证过期时间只设置一次,不进行多次续期
            atomicLong.expire(idempotentTime, timeUnit);
            return true;
        }
        return false;
    }
}