订单场景-基于Redisson实现订单号生成

  <properties>
      <java.version>8</java.version>
      <redisson.version>3.8.2</redisson.version>
  </properties>
  
  <dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter-data-redis</artifactId>
  </dependency>
  <dependency>
      <groupId>org.redisson</groupId>
      <artifactId>redisson</artifactId>
      <version>${redisson.version}</version>
      <optional>true</optional>
  </dependency>
  <dependency>
      <groupId>org.redisson</groupId>
      <artifactId>redisson-spring-boot-starter</artifactId>
      <version>${redisson.version}</version>
  </dependency>

#Redisson配置
singleServerConfig:
  address: "redis://127.0.0.1:6379"
  password: 123456
  clientName: Geo
  #选择使用哪个数据库0~15
  database: 7
  idleConnectionTimeout: 10000
  pingTimeout: 1000
  connectTimeout: 10000
  timeout: 3000
  retryAttempts: 3
  retryInterval: 1500
  reconnectionTimeout: 3000
  failedAttempts: 3
  subscriptionsPerConnection: 5
  subscriptionConnectionMinimumIdleSize: 1
  subscriptionConnectionPoolSize: 50
  connectionMinimumIdleSize: 32
  connectionPoolSize: 64
  dnsMonitoringInterval: 5000
  #dnsMonitoring: false

threads: 0
nettyThreads: 0
codec:
  class: "org.redisson.codec.JsonJacksonCodec"
transportMode: "NIO"

/**
 * @author Liutx
 * @since 2023-12-01 10:29
 */
@Slf4j
@Configuration
public class RedissonConfig {
    @Bean
    public RedissonClient redisson() throws IOException {
        // 本例子使用的是yaml格式的配置文件，读取使用Config.fromYAML，如果是Json文件，则使用Config.fromJSON
        Config config = Config.fromYAML(RedissonConfig.class.getClassLoader().getResource("redisson-config.yml"));
        return Redisson.create(config);
    }
}

/**
 * @author Liutx
 * @since 2023-12-01 09:23
 */
@Slf4j
@Component
public class GenOrderCode {

	// 创建Redisson客户端实例
	private final RedissonClient redissonClient;

	public GenOrderCode(RedissonClient redissonClient) {
		this.redissonClient = redissonClient;
	}

	/**
	 * 生成指定长度的订单号
	 *
	 * @param length   订单总长度
	 * @param prefix   前缀 当天日期-yyMMdd
	 * @param lockKey 分布式锁id
	 * @return 订单号
	 */
	public String genOrderCode(int length, String prefix, String lockKey) {
		// 检查参数合法性
		if (length <= 0) {
			log.warn("获取订单号：订单总长度不能小于0");
			throw new RuntimeException("订单总长度或随机码长度不能小于0");
		}
		if (length <= prefix.length()) {
			log.warn("获取订单号：订单总长度长度小于前缀长度");
			throw new RuntimeException("订单总长度长度小于前缀长度");
		}

		// 获取分布式锁
		RLock lock = redissonClient.getLock(lockKey);
		lock.lock();
		try {
			// 从Redis中获取递增的序列号
			RAtomicLong counter = redissonClient.getAtomicLong("counter");
			// 递增计数器
			long incrementedValue = counter.incrementAndGet();
			String counterValue = String.valueOf(incrementedValue);
			int incrLength = counterValue.length();

			// 如果前缀长度加数字自增长度大于指定位数，则直接使用自增数据
			if (incrLength + prefix.length() > length) {
				return prefix + counterValue;
			}

			// 生成随机码
			int randomLength = length - incrLength;
			String randomAlphabetic = RandomStringUtils.randomAlphabetic(randomLength);
			// 格式化订单号
			String orderCode = prefix + randomAlphabetic + counterValue;
			log.info("根据规则生成的订单号:{}", orderCode);
			return orderCode;
		} finally {
			// 释放锁
			lock.unlock();
		}
	}
}

/**
 * @author Liutx
 * @since 2023-12-01 14:17
 */
@Service
public class OrderServiceImpl implements OrderService {

	private final GenOrderCode genOrderCode;

	public OrderServiceImpl(GenOrderCode genOrderCode) {
		this.genOrderCode = genOrderCode;
	}

	@Override
	public ResultResponse<String> orderNo() {
		ResultResponse<String> resultResponse = ResultResponse.newSuccessInstance();
		LocalDate currentDate = LocalDate.now();
		DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyMMdd");
		String orderPrefix = currentDate.format(formatter);
		//redisKey = Enum + ":" + orderPrefix;=> order:231201，每天的key都不一样，生成的订单号以天区分
		String key = "com" + orderPrefix;
		String orderCode = genOrderCode.genOrderCode(25, orderPrefix, key);
		return resultResponse.succeed().data(orderCode);
	}
}

{
  "success": true,
  "trace": "704acdb4-b694-4cb7-a6a0-aaf034e96eb4",
  "code": "OK",
  "message": "操作成功",
  "data": "231201cwmRZKTPPySiJvJvgjzeb1121"
}