短信验证码生成工具类

在手机号验证码登录的场景，需要实现验证码的随机生成。

本文提供一个Java工具类，可以生成指定位数的验证码，还可以指定生成的验证码不出现重复数字。

GenerateCodeUtil

import lombok.extern.slf4j.Slf4j;

import java.security.SecureRandom;
import java.util.*;

/**
 * @author 曳戈泰尔
 * @version 1.0
 * @description 生成验证码工具类
 * @date 2023/6/1 8:02 PM
 */
@Slf4j
public class GenerateCodeUtil {

    /**
     * 生成？位的数字类型的短信验证码
     *
     * @param count 位数
     * @return 验证码
     */
    public static String generatedCode(int count) {

        // 获取随机数，不可重复
        List<Integer> set = getRandomNumber(count);
        // 获取随机数，可重复
        // List<Integer> set = getRandomNumberByRepetition(count);

        // 使用迭代器
        Iterator<Integer> iterator = set.iterator();

        // 临时记录数据
        StringBuilder temp = new StringBuilder();
        while (iterator.hasNext()) {
            temp.append(iterator.next());
        }

        // 返回数据
        return temp.toString();
    }

    /**
     * 返回？位的随机数字；不可重复
     *
     * @param count 位数
     * @return 随机数字
     */
    public static List<Integer> getRandomNumber(int count) {

        // 使用SET保证写入的数据不重复
        Set<Integer> set = new HashSet<>(count);

        // 随机数
        Random random = new SecureRandom();

        // 循环产生随机数
        while (set.size() < count) {
            // nextInt返回一个伪随机数
            set.add(random.nextInt(10));
        }

        // 返回数据 集合
        return new ArrayList<>(set);
    }

    /**
     * 返回？位的随机数字；可重复
     *
     * @param count 位数
     * @return 随机数字
     */
    public static List<Integer> getRandomNumberByRepetition(int count) {

        // 使用SET保证写入的数据不重复
        List<Integer> list = new ArrayList<>(count);

        // 随机数
        Random random = new SecureRandom();

        // 循环产生随机数
        while (list.size() < count) {
            // nextInt返回一个伪随机数
            list.add(random.nextInt(10));
        }

        // 返回数据 集合
        return list;
    }
}