import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
 
import java.text.SimpleDateFormat;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.temporal.ChronoUnit;
import java.util.Date;
import java.util.concurrent.TimeUnit;
 
/**
 * redis的increment 遞增方法 | 處理防重復和并發(fā)問題
 */
@Component
public class OrderNumberCodeUtils {
    private static final String PREFIX = "DD";
    private static final String DATE_FORMAT = "yyyyMMdd";
    private static final String ORDER_SERIAL_NUMBER = "order_serial_number";
 
    private static RedisTemplate redisTemplate;
 
    @Autowired
    public void redisTemplate(RedisTemplate redisTemplate){
        OrderNumberCodeUtils.redisTemplate = redisTemplate;
    }
 
    public static String generateOrderNumber() {
 
        StringBuffer stringBuffer = new StringBuffer();
        stringBuffer.append(PREFIX);
        // 獲取當前日期
        SimpleDateFormat dateFormat = new SimpleDateFormat(DATE_FORMAT);
        String currentDate = dateFormat.format(new Date());
        stringBuffer.append(currentDate);
        // 獲取流水號
        Long increment = redisTemplate.opsForValue().increment(ORDER_SERIAL_NUMBER, 1);
		
		/**
		 * 返回值過期時間，單位為秒。
		 * 如果返回-2，則表示該鍵不存在；
		 * 如果返回-1，則表示該鍵沒有設置過期時間；
		 */
		Long expire = redisTemplate.getExpire(ORDER_SERIAL_NUMBER, TimeUnit.SECONDS);
		if(expire == -1){
			// 獲取距離當天結束的秒數(shù)
			LocalDateTime endOfDay = LocalDate.now().atTime(23, 59, 59);
			long secondsToMidnight = LocalDateTime.now().until(endOfDay, ChronoUnit.SECONDS);
            //初始設置過期時間
            redisTemplate.expire(ORDER_SERIAL_NUMBER, secondsToMidnight, TimeUnit.SECONDS);
        }
        String format = String.format("%05d", increment);
        stringBuffer.append(format);
        return stringBuffer.toString();
    }
}