redis RedisAtomicLong生成自增主键

1.自增id 生成类

RedisSequenceFactory 是一个简单封装类，用于使用redisTemplate生成自增ID值。

//
// Source code recreated from a .class file by IntelliJ IDEA
// (powered by Fernflower decompiler)
//

package com.jiu.common.redis;

import java.util.ArrayList;
import java.util.List;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.support.atomic.RedisAtomicLong;

public class RedisSequenceFactory {
    private static final Logger log = LoggerFactory.getLogger(RedisSequenceFactory.class);
    @Autowired
    private ObjectRedisTemplate<Integer> redisTemplate;

    public RedisSequenceFactory() {
    }

    public void set(String key, long value) {
        RedisAtomicLong counter = new RedisAtomicLong(key, this.redisTemplate.getConnectionFactory());
        counter.set(value);
    }

    public long generate(String key, int increment) {
        RedisAtomicLong counter = new RedisAtomicLong(key, this.redisTemplate.getConnectionFactory());
        return counter.addAndGet((long)increment);
    }

    public List<Long> generateBatch(String key, int increment, int size) {
        RedisAtomicLong counter = new RedisAtomicLong(key, this.redisTemplate.getConnectionFactory());
        long max = counter.addAndGet((long)(increment * size));
        long min = max - (long)(increment * (size - 1));
        List<Long> list = new ArrayList();
        list.add(min);

        for(int i = 1; i < size; ++i) {
            list.add(min + (long)(increment * i));
        }

        return list;
    }

    public long queryValue(String key) {
        Integer val = (Integer)this.redisTemplate.get(key);
        return val == null ? 0L : (long)val.intValue();
    }
}

测试代码

  public static String tradeIdIncRedisKey = "order:orderid_inc";
     @Test
    public long generateId(){
        return redisSequenceFactory.generate(tradeIdIncRedisKey,1);
    }