上一篇演示了使用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);
}
}