【Java】Java中用redis生成每天从0开始的业务ID

场景

业务场景是生成一个唯一主键，主键格式为 字母前缀 + 年月日 + 自增数字（位数固定每天从0开始）
示例：TEST2020042900000015

思路

Redis Incr 命令将 key 中储存的数字值增一。
如果 key 不存在，那么 key 的值会先被初始化为 0 ，然后再执行 INCR 操作。

利用redis的Incr方法实现自增的序列，由于每天都要从0开始，因此每天生成一个key（固定的英文前缀+年月日），然后设置超时时间为一天。

实现

package com.lh.springbootredis.controller;

import com.lh.springbootredis.utils.RedisUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.annotation.Resource;
import java.text.DecimalFormat;

@Slf4j
@RestController
public class RedisController {

    @Resource
    private RedisUtil redisUtil;

    /**
     * redis key 前缀
     */
    private final static String KEY_FREFIX = "TEST_ID";

    /**
     * ID 前缀
     */
    private final static String ID_FREFIX = "TEST";

    /**
     * 获取id
     * <p>
     * id结构 字母前缀 + 年月日 + 自增数字（位数固定每天从0开始）
     *
     * @param busiDate 取系统当前时间格式化 yyyyMMdd
     * @return id
     */
    @GetMapping("/getId")
    public String getId(String busiDate) {
        // 获取key增1的值 （当key不存在的时候初始化值是0）
        long incr = redisUtil.incr(KEY_FREFIX + busiDate, 1);
        if (incr == 1) {
            //设置超时时间为一天
            redisUtil.expire(KEY_FREFIX + busiDate, 24 * 60 * 60);
        }
        return createId(busiDate, incr);
    }

    private String createId(String busiDate, long incr) {
        DecimalFormat df = new DecimalFormat("00000000");
        String format = df.format(incr);
        return ID_FREFIX + busiDate + format;
    }
}

@Component
public class RedisUtil {

    @Autowired
    private RedisTemplate<String, Object> redisTemplate;

    /**
     * 递增
     *
     * @param key   键
     * @param delta 要增加几(大于0)
     * @return
     */
    public long incr(String key, long delta) {
        if (delta < 0) {
            throw new RuntimeException("递增因子必须大于0");
        }
        return redisTemplate.opsForValue().increment(key, delta);
    }

    /**
     * 指定缓存失效时间
     *
     * @param key  键
     * @param time 时间(秒)
     * @return
     */
    public boolean expire(String key, long time) {
        try {
            if (time > 0) {
                redisTemplate.expire(key, time, TimeUnit.SECONDS);
            }
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }
}