我正在参与「启航计划」

背景

最近分配到一个需求，要生成6位数验证码，运用验证码做一系列操作。由于需求运用验证码去确定用户信息，所以有唯一性要求。

计划一：随机函数Random()

运用随机函数生成1000000以内的数，然后再补齐位数，然后校验唯一性，假如已存在就在生成一个随机数。

/**
 * 计划一：随机函数Random()
 */
@Test
public void testRandom() {
    Random rd = new Random();
    for (int i = 0; i < 10; i++) {
        val numberCode = rd.nextInt(1000000);
        System.out.println(String.format("%06d", numberCode));
    }
}

输出结果：

731311
508327
383928
551535
623440
758321
284932
039829
365014
500294
Process finished with exit code 0

优点：

• 完成简单快捷

缺点：

• 重复概率相对较大

计划二：Redis令牌桶方式

数字码的生成首要考虑效率问题，假如运用随机函数random()随机生成1000000以内的随机数，可能比较简单呈现数据重复问题。因而我们先初始化将000000-999999总的100万个数据先放到redis中，然后随机获取一个，等到redis中没有数据再从头初始化100万个数据，这样理论上短时间内不会呈现已存在的数据。

private final String KEY_NUMBER_CODE = "visitor-number-code";
@Autowired
private RedisTemplate<String, String> redisTemplate;
public String getNumberCode(String key) {
    String numberCode = redisTemplate.opsForSet().pop(key);
    if (StringUtils.isBlank(numberCode)) {
        // 运用Redis分布式锁操控并发
        numberCode = initNumberCode(key);
    }
    return numberCode;
}
/**
 * Redis初始化1000000个数据
 */
public String initNumberCode(String key) {
    // 双重查看，防止多个进行一起在等候分布式锁初始化
    String numberCode = redisTemplate.opsForSet().pop(key);
    if (StringUtils.isNotBlank(numberCode)) {
        return numberCode;
    }
    // 将数据初始化到数组再一次性放到Redis中，提高功能
    List<String> list = new ArrayList<>();
    for (int i = 1; i < 1000000; i++) {
        list.add(String.format("%06d", i));
    }
    String[] arr = list.toArray(new String[0]);
    redisTemplate.opsForSet().add(key , arr);
    return redisTemplate.opsForSet().pop(key);
}

代码完成关键：

• 运用Redis分布式锁操控并发
• 双重查看，防止多个进行一起在等候分布式锁初始化
• 将数据初始化到数组再一次性放到Redis中，提高功能

补偿措施

我们要赶快减少数字码呈现重复的概率，假如呈现重复数据要有重试机制，从头获取数字码。

由于6位数字码是从000000-999999只有100万个数据，假如不及时铲除数字码的话，日积月累总会用完的，因而要拟定失效铲除战略。或者运用字符组合，构成更加丰富的验证码。