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

超、凢脫俗 2024-03-16 17:26 49阅读 0赞

> 首先，邀请码的硬性特征：唯一性和不可重复。这是最重要的，所以无论是什么算法我们都要确保唯一性的校验，就算是有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();
        }
    
    }