Redis实现分布式自增序列号

需求背景

根据不同的业务场景生成唯一的自增序列号，达成最大值后重置

技术实现

数据库方式

通过设置对应表相关字段的自增序列步长和初始值
这种方案的好处是不用动任何业务代码，缺点是数据库如果多节点集群部署的话就无法保证唯一性

Redis方式

通过给定key的increment方法进行自增的设置

这种方案的优点是可以实现多节点部署的序列号唯一性，缺点是需要修改业务代码

对应的java版本实现代码如下：

package com.test.service.impl;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.ValueOperations;
import org.springframework.stereotype.Service;


@Service
@Slf4j
public class PrimaryKeyService {

    @Autowired
    RedisTemplate<String,String> redisTemplate;

    public String getAutoIncreSerialNumber(String redisKey, int length) {
        ValueOperations<String, String> operations = redisTemplate.opsForValue();
        operations.increment(redisKey, 1L);
        String value = operations.get(redisKey).toString();

        String numStr =  addZero(value,length);
        if(getMaxNumber(length).equals(numStr)){
            log.info("the serialNumber is maxNum, reset the key {}",redisKey);
            redisTemplate.delete(redisKey);
        }
        log.info("the serialNumber is {}",numStr);
        return numStr;
    }

    /**
     * 根据位数生成最大值
     * @param len
     * @return
     */
    private String getMaxNumber(int len){
        StringBuffer rStr = new StringBuffer();
        for (int i = 0; i < len; i++) {
            rStr.append("9");
        }
        return rStr.toString();
    }
    // 根据位数自动补零
    private String addZero(String numStr, Integer maxNum) {
        int addNum = maxNum - numStr.length();
        StringBuffer rStr = new StringBuffer();
        for (int i = 0; i < addNum; i++) {
            rStr.append("0");
        }
        rStr.append(numStr);
        return rStr.toString();
    }
}