我写了一个简单的单号生成器

科技 08-16 来源： Lvshen的技术小屋

使用MySQL生成

相信大家在开发上都会遇到业务单号生成的需求，一般的生成格式为：

 前缀 + YYMMDD + 序列号（例如3位序列号）

这里前缀 + 日期没有太大技术含量，主要是在尾部的序列号，这里要求位3位序列号，我们从001开始。当等于10起，为010。当等于100起，为100。第二天重置从001起。这里我们可以使用MySQL数据库存储，这里我们需要记录日期和序号两个字段。MySQL 表如下：

 CREATE TABLE `code_generator` (   `id` bigint(10) NOT NULL AUTO_INCREMENT COMMENT 'id',   `number` bigint(10) DEFAULT NULL COMMENT '序号',   `gnerator_time` datetime DEFAULT NULL COMMENT '生成日期',   `deleted` tinyint(1) DEFAULT '0' COMMENT '是否删除',   `created_by` varchar(32) CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci DEFAULT NULL COMMENT '创建人',   `created_time` datetime DEFAULT NULL COMMENT '创建时间',   `updated_by` varchar(32) CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci DEFAULT NULL COMMENT '更新人',   `updated_time` datetime DEFAULT NULL COMMENT '更新时间',   PRIMARY KEY (`id`) USING BTREE ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='单号生成器'

调用生成方法时，判断当前日期与数据库生成日期gnerator_time是否为同一天。如果为同一天，在当天的数据下面 number + 1。如果不为同一天，删掉当前数据，插入一条number为1的数据。这里在计算number时注意添加锁，防止多线程下计算不准确。

使用Redis生成

当然，除了使用MySQL生成，还可以使用Redis来生成。相较于MySQL，Redis的性能更为优良。

这里我们使用Redisson API。

首先我们使用Spring注入一个Redisson客户端：

 @Bean("redissonService") public RedissonClient getRedissonClient() {      String activeProfile = getActiveProfile();     Config config;     if ("prod".equals(activeProfile)) {         config = useClusterConfig();     } else {         config = useSingleConfig();     }     return Redisson.create(config); }

在配置RedissonClient时，可以设置为单机模式，或者集群模式。

这样就可以从容器中获取bean了。

 @Autowired private RedissonClientConfig redissonService;

生成编号的核心代码如下：

 private synchronized long getSuffixCode(String key) {     RedissonClient redissonClient = redissonService.getRedissonClient();     RAtomicLong atomicVar = redissonClient.getAtomicLong(key);     String todayStr = getTodayStr();     String codeRecord = getCodeRecord(key);     if (!atomicVar.isExists()) {         atomicVar.set(1);     }     if (StringUtils.isNotBlank(codeRecord)) {         if (!isSameDay(todayStr, codeRecord)) {             atomicVar.set(1);         }     }     saveCodeRecord(key, todayStr);     long value = atomicVar.incrementAndGet();     return value; }

上面代码功能为生成序号。我们先在redis中查询，如果通过key能查到对应的值，则在这个值的基础上增加1。如果值不存在，设置为1。当然，如果当天日期与redis记录日期不是同一天，也需要将值置为1。

saveCodeRecord方法代码如下：

 private void saveCodeRecord(String key, String value) {     String recordKey = key + CODE_DATE_RECORD_KEY;     redisUtils.set(recordKey, value);

序号如果小于10，则前面补两个个0。在10到100之间，补两个0。序号补位方法如下：

 private String getSuffixCodeStr(String key, int digit) {     long code = getSuffixCode(key);     return getDigitNumber((int) code, digit); }  public String getDigitNumber(int number, int digit) {     NumberFormat formatter = NumberFormat.getNumberInstance();     formatter.setMinimumIntegerDigits(digit);     formatter.setGroupingUsed(false);     return formatter.format(number); }

之后我们再将前缀，日期，序号拼接起来即可：

 private String generateCode(String prefix, String redisKey, int digit) {     String dateStr = getTodayStr();      String suffixCodeStr = getSuffixCodeStr(redisKey, digit);      String[] codes = {prefix, dateStr, suffixCodeStr};     return String.join("", codes); }

我们再做一层简单的封装：

 public String generatorCode(String prefix, int digit) {     String redisKey = getRedisKey(prefix, "GENERAL_CODE");     return generateCode(prefix, redisKey, digit); }

对generatorCode()说明如下：