实际应用中，纯数字六位邀请码的实现（java实现）

​ 实际应用中，纯数字六位邀请码的实现（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();
    }

}