实际应用中,纯数字六位邀请码的实现(java实现)
首先,邀请码的硬性特征:唯一性和不可重复。这是最重要的,所以无论是什么算法我们都要确保唯一性的校验,就算是有10亿分之一的可能重复,也是不行的。
其次,我们的业务上有几点软性要求:
1、是需要纯数字的,不要跟字母结合。
2、不要顺序的,尽可能随机。
基于上述硬性和两点软性要求,我们给出的方案是,利用数据库或者redis作为唯一性的校验。众所周知,数据库是基于硬盘存储的,除了业务强相关的数据,我们尽量减少对数据库的连接和查询,避免资源性能和接口性能问题。而redis的储存是在内存中的,性能方便肯定是比数据库好的。所以,在我们生产代码中唯一性的检验是利用redis的集合来做校验的。
下面是java邀请码工具类的代码实现
package com.example.demo.utils;
import org.springframework.beans.factory.annotation.Autowired;
import java.util.concurrent.TimeUnit;
public class ShareCodeUtils {
@Autowired
private RedisTemplate redisTemplate;
// 邀请码的长度
private static final int CODE_LENGTH = 6;
// redis集合的key值
private static final String REDIS_KEY = "INVITE_CODE";
public String generate() {
String code;
do {
code = generateCode();
} while (redisTemplate.opsForSet().isMember(REDIS_KEY,code));
redisTemplate.opsForSet().add(REDIS_KEY, code);
return code;
}
private String generateCode() {
StringBuilder sb = new StringBuilder();
for (int i = 0; i < CODE_LENGTH; i++) {
sb.append((int) (Math.random() * 10));
}
return sb.toString();
}
}