添加随机数

randomx.go

@@ -48,6 +48,24 @@ func RandomStr(length int) (string, error) {
 	return string(result), nil
 }
 
+func RandomNum(length int) (string, error) {
+	if length <= 0 {
+		return "", errors.New("位数必须大于0")
+	}
+	str := "0123456789"
+
+	bytes := []rune(str)
+
+	var result = make([]rune, 0, length)
+	r := rand.New(rand.NewSource(time.Now().UnixNano()))
+
+	for i := 0; i < length; i++ {
+		result = append(result, bytes[r.Intn(len(bytes))])
+	}
+
+	return string(result), nil
+}
+
 // 生成指定空间的随机数
 func RandomInt(min, max int) (int, error) {
 	c := max - min