使用redis生成id的工具类

package com.susu.demo.service;

import com.nuctech.common.core.config.jedis.RedisKey;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.support.atomic.RedisAtomicLong;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.io.Serializable;
import java.time.LocalDateTime;
import java.time.ZoneId;
import java.time.ZonedDateTime;
import java.time.format.DateTimeFormatter;
import java.util.Calendar;
import java.util.Date;
import java.util.concurrent.TimeUnit;

/**
 * @Author: susu
 * @Date: 2022/4/1 16:20
 * @since: 1.0.0
 * @Description
 */
@Service
public class GenerateIdService {

	private static Logger logger = LoggerFactory.getLogger(GenerateIdService.class);

	@Resource
	private RedisTemplate<String, Serializable> redisTemplate;


	private String prefixKey = RedisKey.ROOT_PREFIX + "REDIS_ID_KEY:";

	/***
	 * @param
	 * @return: java.lang.String
	 * @Author: susu
	 * @date: 2022/4/2
	 * @since 1.0.0
	 * @Description: 获取id 生成id为当前日期（yyMMddHHmmss）+6位（从000000开始不足位数补0）  20220402094508000001
	 * 分布式情况下 当前日期后边加上当前服务器id
	 */
	public String generateId() {
		// 生成id为当前日期（yyMMddHHmmss）+6位（从000000开始不足位数补0）
		LocalDateTime now = LocalDateTime.now();
		String idPrefix = getIdPrefix(now);// 生成yyyyMMddHHmmss
		String id = idPrefix + String.format("%1$06d", generate(idPrefix, getExpireAtTime(now)));
		return id;
	}

	/**
	 * 获取有过期时间的自增长ID
	 */
	public long generate(String key, Date expireTime) {
		RedisAtomicLong counter = new RedisAtomicLong(prefixKey + key, redisTemplate.getConnectionFactory());
		Long expire = counter.getExpire();
		if (expire == -1) {
			counter.expireAt(expireTime);
		}
		return counter.incrementAndGet();
	}


	public static String getIdPrefix(LocalDateTime now) {
		return now.format(DateTimeFormatter.ofPattern("yyyyMMddHHmmss"));
	}

	public Date getExpireAtTime(LocalDateTime now) {
		ZoneId zoneId = ZoneId.systemDefault();
		LocalDateTime localDateTime = now.plusSeconds(20);
		ZonedDateTime zdt = localDateTime.atZone(zoneId);
		Date date = Date.from(zdt.toInstant());
		return date;
	}

	//------------------按照年月日生成 日增------------------------
	//现在到今天结束的毫秒数
	public Long getCurrent2TodayEndMillisTime() {
		Calendar todayEnd = Calendar.getInstance();
		// Calendar.HOUR 12小时制
		// HOUR_OF_DAY 24小时制
		todayEnd.set(Calendar.HOUR_OF_DAY, 23);
		todayEnd.set(Calendar.MINUTE, 59);
		todayEnd.set(Calendar.SECOND, 59);
		todayEnd.set(Calendar.MILLISECOND, 999);
		return todayEnd.getTimeInMillis() - new Date().getTime();
	}

	/***
	 * @param
	 * @return: java.lang.String
	 * @Author: susu
	 * @date: 2022/4/2
	 * @since 1.0.0
	 * @Description: 获取id 生成id为当前日期（yyMMdd）+6位（从000000开始不足位数补0） 20220402000024
	 */
	public String generateIdByDay() {
		// 生成id为当前日期（yyMMdd）+6位（从000000开始不足位数补0）
		LocalDateTime now = LocalDateTime.now();
		String idPrefix = getIdPrefixYMD(now);// 生成yyyyMM
		String id = idPrefix + String.format("%1$06d", generateByDay(idPrefix, getCurrent2TodayEndMillisTime()));
		return id;
	}

	public long generateByDay(String key, Long expireTime) {
		RedisAtomicLong counter = new RedisAtomicLong(prefixKey + key, redisTemplate.getConnectionFactory());
		Long expire = counter.getExpire();
		if (expire == -1) {
			counter.expire(expireTime, TimeUnit.MILLISECONDS);
		}
		return counter.incrementAndGet();
	}

	public static String getIdPrefixYMD(LocalDateTime now) {
		return now.format(DateTimeFormatter.ofPattern("yyyyMMdd"));
	}
}