lodsve.core.utils.RandomUtils Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2018  Sun.Hao
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see .
 */

package lodsve.core.utils;

import org.apache.commons.lang3.ArrayUtils;

import java.util.concurrent.ThreadLocalRandom;

/**
 * 随机生成数字或者字符串的工具类.
 *
 * @author sunhao([email protected])
 * @date 2012-4-24 下午08:25:57
 */
public class RandomUtils {
    private static final String LETTER_LOWER = "abcdefghijklmnopqrstuvwxyz";
    private static final String LETTER_UPPER = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
    private static final String DIGIT = "0123456789";

    /**
     * 私有化构造器
     */
    private RandomUtils() {
    }

    /**
     * 获取长度为length的随机数
     *
     * @param length 长度
     * @return 获取的随机数(String)
     */
    public static String getRandomNum(int length) {
        if (length < 1) {
            return StringUtils.EMPTY;
        }

        long returnRandom = 0;
        long min = 1;
        long max = 1;

        for (int i = 0; i < length - 1; i++) {
            min *= 10;
        }
        for (int j = 0; j < length; j++) {
            max *= 10;
        }

        ThreadLocalRandom random = ThreadLocalRandom.current();
        while (returnRandom <= min) {
            returnRandom = random.nextInt() * max;
        }

        return Long.toString(returnRandom);
    }

    /**
     * 生成随机数字
     *
     * @param length 生成的密码的总长度
     * @return 数字的字符串
     */
    public static String randomNum(int length) {
        if (length < 1) {
            return StringUtils.EMPTY;
        }
        // 10个数字
        final int maxNum = 10;
        char[] str = DIGIT.toCharArray();

        StringBuffer tempString = new StringBuffer(100);
        generateRandomNumberString(tempString, length, maxNum, str);

        return tempString.toString();
    }

    /**
     * 生成随机字符串
     *
     * @param length 生成的密码的总长度
     * @return 字符串(大小写字母 + 数字)
     */
    public static String randomString(int length, boolean lower, boolean upper, boolean digit) {
        if (length < 1) {
            return StringUtils.EMPTY;
        }
        //26个字母*2 + 10个数字
        final int maxNum = 62;
        char[] str = {};

        if (lower) {
            str = ArrayUtils.addAll(str, LETTER_LOWER.toCharArray());
        }
        if (upper) {
            str = ArrayUtils.addAll(str, LETTER_UPPER.toCharArray());
        }
        if (digit) {
            str = ArrayUtils.addAll(str, DIGIT.toCharArray());
        }

        StringBuffer tempString = new StringBuffer(100);
        generateRandomNumberString(tempString, length, maxNum, str);

        return tempString.toString();
    }

    /**
     * 生成随机字符串
     *
     * @param length 生成的密码的总长度
     * @return 字符串(大小写字母 + 数字)
     */
    public static synchronized String randomString(int length) {
        if (length < 1) {
            return StringUtils.EMPTY;
        }

        ThreadLocalRandom random = ThreadLocalRandom.current();
        char[] numbersAndLetters = ArrayUtils.addAll(null, DIGIT.toCharArray());
        numbersAndLetters = ArrayUtils.addAll(numbersAndLetters, LETTER_LOWER.toCharArray());
        numbersAndLetters = ArrayUtils.addAll(numbersAndLetters, DIGIT.toCharArray());
        numbersAndLetters = ArrayUtils.addAll(numbersAndLetters, LETTER_UPPER.toCharArray());

        char[] randBuffer = new char[length];
        for (int i = 0; i < randBuffer.length; i++) {
            randBuffer[i] = numbersAndLetters[random.nextInt(72)];
        }
        return new String(randBuffer);
    }

    private static void generateRandomNumberString(StringBuffer tempString, int length, int maxNum, char[] str) {
        ThreadLocalRandom random = ThreadLocalRandom.current();
        int i, count = 0;
        while (count < length) {
            /*
              生成随机数，取绝对值，防止生成负数
              Random.nextInt(int num)	返回一个在[0, n)
              所以不会造成数组下标越界的情况
             */
            i = Math.abs(random.nextInt(maxNum));
            if (i >= 0 && i < str.length) {
                tempString.append(str[i]);
                count++;
            }
        }
    }
}