使用jedis连接redis-cluster实现纯redis实现的ID生成器

上一篇演示了使用jedis连接redis-cluster实现纯redis实现的bloom过滤器,本篇演示通过redis实现ID生成器。

package com.coderman.jedis.clusterdemo.idgenerator;

import com.coderman.jedis.clusterdemo.ClusterTest;
import org.junit.Test;

import java.time.Instant;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.LocalTime;
import java.time.ZoneId;
import java.time.format.DateTimeFormatter;
import java.time.temporal.ChronoField;
import java.util.Date;

/**
 * @Author fanchunshuai
 * @Date 2020/1/16 11
 * @Description:
 * 参考：https://www.cnblogs.com/juihai/p/8904246.html
 *
 */
public class IdGeneratorTest extends ClusterTest {
    /**
     * 带业务单前缀的
     */
    private static final String ORDER_KEY = "YWSQD";

    /**
     * 带业务单前缀的
     */
    private static final String ORDER_DATE_KEY = "YWSQD_DATE";

    /**
     * 默认自增的
     */
    private static final String DEFAULT_KEY = "DEFAULT";

    /**
     * 不带日期的定长单号生产
     */
    @Test
    public void testORDER(){
        cluster.set(ORDER_KEY,"00001");
        String value = null;
        for (int i = 0;i< 10000;i++){
            cluster.incr(ORDER_KEY);
            value = cluster.get(ORDER_KEY);
            switch (value.length()){
                case 1:
                    value ="000" + value;
                    break;
                case 2:
                    value ="00" + value;
                    break;
                case 3:
                    value ="0" + value;
                    break;
            }
            String orderNumber = ORDER_KEY+value;
            System.out.println("orderNumber = "+orderNumber);
        }
    }

    /**
     * 使用带有日期的业务单号生成
     */
    @Test
    public void testOrderWithDate(){
        /**
         * 这里的过期时间可以设置为1000，代表一个周期，比如每天的单号生成是从YWSQD_DATE0001开始的
         * 每天凌晨重置过期时间和值，这样的话可以不需要自己维护一套单号自动生成库
         *
         */
        cluster.setex(ORDER_DATE_KEY,1000,"0000");
        String value = null;
        for (int i = 0;i< 10000;i++){
            cluster.incr(ORDER_KEY);
            value = cluster.get(ORDER_KEY);
            switch (value.length()){
                case 1:
                    value ="000" + value;
                    break;
                case 2:
                    value ="00" + value;
                    break;
                case 3:
                    value ="0" + value;
                    break;
            }
            String orderNumber = ORDER_KEY+formatAnyTime(new Date()) +value;
            System.out.println("orderNumber = "+orderNumber);
        }
    }

    static DateTimeFormatter formatter2 = DateTimeFormatter.ofPattern("yyyyMMdd");

    /**
     * 格式化指定时间
     */
    public static String formatAnyTime(Date date) {
        return formatter2.format(toLocalDateTime(date));
    }


    private static LocalDateTime toLocalDateTime(Date date) {
        Instant instant = date.toInstant();
        ZoneId zone = ZoneId.systemDefault();
        return LocalDateTime.ofInstant(instant, zone);
    }

    /**
     * 获取年
     *
     * @return 年
     */
    private static int getYear() {
        LocalTime localTime = LocalTime.now();
        return localTime.get(ChronoField.YEAR);
    }

    /**
     * 获取月份
     *
     * @return 月份
     */
    private static int getMonth() {
        LocalTime localTime = LocalTime.now();
        return localTime.get(ChronoField.MONTH_OF_YEAR);
    }

    /**
     * 获取某月的第几天
     *
     * @return 几号
     */
    private static int getMonthOfDay() {
        LocalTime localTime = LocalTime.now();
        return localTime.get(ChronoField.DAY_OF_MONTH);
    }

}