Redis生成分布式环境下自增ID

相信大家在分布式系统一定会遇到如何生成唯一Id; 

uuid 可以但是,作为数据库主键很浪费性能(索引相关),

雪花算法可以, 但是很依赖于环境.

这里推荐大家使用redis生成;

 


1. 需求明确

我们生成的Id组成为, 时间戳+ 自增Id, 补0 

例如: 今天是2020.2.28 ,  补6个0 , 那么生成的id范围是20200228000001-202002289999999

2.技术选型

Redis中的RedisAtomicLong 具备原子性, 可以通过其incr方法进行递增.

 

2. Id生成类

package com.dyhospital.cloudhis.common.redis.util;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.support.atomic.RedisAtomicLong;
import org.springframework.stereotype.Component;

import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;
import java.util.concurrent.TimeUnit;

/**
 * Description: 分布式Id生成工厂
 * User: zhouzhou
 * Date: 2020-02-28
 * Time: 10:14
 */
@Component
public class IdGeneratorFactory {

    @Autowired
    RedisTemplate<String, String> redisTemplate;

    /**
     * @param key
     * @param value
     * @param expireTime
     * @Title: set
     * @Description: set cache.
     */
    public void set(String key, int value, Date expireTime) {
        RedisAtomicLong counter = new RedisAtomicLong(key, redisTemplate.getConnectionFactory());
        counter.set(value);
        counter.expireAt(expireTime);
    }

    /**
     * @param key
     * @param value
     * @param timeout
     * @param unit
     * @Title: set
     * @Description: set cache.
     */
    public void set(String key, int value, long timeout, TimeUnit unit) {
        RedisAtomicLong counter = new RedisAtomicLong(key, redisTemplate.getConnectionFactory());
        counter.set(value);
        counter.expire(timeout, unit);
    }

    /**
     * @param key
     * @return
     * @Title: generate
     * @Description: Atomically increments by one the current value.
     */
    public long generate(String key) {
        RedisAtomicLong counter = new RedisAtomicLong(key, redisTemplate.getConnectionFactory());
        return counter.incrementAndGet();
    }

    /**
     * @param key
     * @return
     * @Title: generate
     * @Description: Atomically increments by one the current value.
     */
    public long generate(String key, Date expireTime) {
        RedisAtomicLong counter = new RedisAtomicLong(key, redisTemplate.getConnectionFactory());
        counter.expireAt(expireTime);
        return counter.incrementAndGet();
    }

    /**
     * @param key
     * @param increment
     * @return
     * @Title: generate
     * @Description: Atomically adds the given value to the current value.
     */
    public long generate(String key, int increment) {
        RedisAtomicLong counter = new RedisAtomicLong(key, redisTemplate.getConnectionFactory());
        return counter.addAndGet(increment);
    }

    /**
     * @param key
     * @param increment
     * @param expireTime
     * @return
     * @Title: generate
     * @Description: Atomically adds the given value to the current value.
     */
    public long generate(String key, int increment, Date expireTime) {
        RedisAtomicLong counter = new RedisAtomicLong(key, redisTemplate.getConnectionFactory());
        counter.expireAt(expireTime);
        return counter.addAndGet(increment);
    }

    /**
     * 根据业务key, 长度, 获取key, 以日期作为前缀
     * <p>
     *     key = order, length = 5 ,当天日期2050年1月1日
     *     结果: 2050010100001
     * </p>
     * @param key
     * @param length
     * @return
     */
    public String generateIdByToday(String key, Integer length) {
        RedisAtomicLong counter = new RedisAtomicLong(key, redisTemplate.getConnectionFactory());

        long num = counter.incrementAndGet();
        counter.expireAt(getTodayEndTime());
        String id = getToday() +  String.format("%0" + length + "d", num);
        return id;

    }


    /**
     * @Title: getTodayEndTime 获取今日最后的时间
     * @Description: Get the cache expire time.
     * @return
     */
    public static Date getTodayEndTime() {
        Calendar todayEnd = Calendar.getInstance();
        todayEnd.set(Calendar.HOUR_OF_DAY, 23);
        todayEnd.set(Calendar.MINUTE, 59);
        todayEnd.set(Calendar.SECOND, 59);
        todayEnd.set(Calendar.MILLISECOND, 999);
        return todayEnd.getTime();
    }

    /**
     * @Title: getTodayEndTime 今天的日期格式: 20190101
     * @Description: Get the cache expire time.
     * @return
     */
    public static String getToday() {
        SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd");
        return sdf.format(new Date());
    }

}

3. 我们来测试下

测试类如下:

   // ---------------------  分布式自增Id 测试  ---------------------------
    @ApiOperation("根据key生成自增Id")
    @RequestMapping(value = "/getId/{key}", method = RequestMethod.GET)
    public GenericResponse<Long> getId(@PathVariable("key") String key) {
        logger.info("生成Id中");
        long generate = idGeneratorFactory.generate(key, IdGeneratorFactory.getTodayEndTime());
        return new GenericResponse<>(generate);
    }

    @ApiOperation("根据key生成自增Id,日期版")
    @RequestMapping(value = "/getIdByToday/{key}", method = RequestMethod.GET)
    public GenericResponse<String> getIdByToday(@PathVariable("key") String key) {
        logger.info("生成Id中");
        String id = idGeneratorFactory.generateIdByToday(key, 6);
        return new GenericResponse<>(id);
    }

测试结果如下: 

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 黑客帝国 设计师:上身试试 返回首页