101 lines
2.5 KiB
Go
101 lines
2.5 KiB
Go
|
package utility
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"strconv"
|
|||
|
|
"strings"
|
|||
|
|
|
|||
|
|
"go.uber.org/zap"
|
|||
|
|
|
|||
|
|
cryptoRand "crypto/rand"
|
|||
|
|
"fmt"
|
|||
|
|
"math"
|
|||
|
|
"math/big"
|
|||
|
|
"math/rand"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
log "github.com/go-admin-team/go-admin-core/logger"
|
|||
|
|
"github.com/rs/xid"
|
|||
|
|
"github.com/sony/sonyflake"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
const base62Chars = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
|
|||
|
|
|
|||
|
|
// GetXid Package xid is a globally unique id generator library
|
|||
|
|
// 包xid是一个全局唯一的id生成器库
|
|||
|
|
func GetXid() string {
|
|||
|
|
return xid.New().String()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetRandIntStr 生成len位的随机数字
|
|||
|
|
func GetRandIntStr(len int, prefix string) string {
|
|||
|
|
rand.Seed(time.Now().UnixNano())
|
|||
|
|
|
|||
|
|
num := rand.Int31n(int32(math.Pow(10, float64(len))))
|
|||
|
|
x := fmt.Sprintf("%s%0*d", prefix, len, num)
|
|||
|
|
return x
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GenerateRandString 生成指定位数的字符串
|
|||
|
|
// 虽然繁琐 但理解之后就觉得很精妙
|
|||
|
|
func GenerateRandString(length int) string {
|
|||
|
|
var chars = []byte(`ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789`) // 长度:(1,256)
|
|||
|
|
rand.Seed(time.Now().UnixNano())
|
|||
|
|
clen := len(chars)
|
|||
|
|
maxRb := 255 - (256 % clen) // [-1,255] 255 - (256%36) = 251 避免模偏倚 为了每个字符被取到的几率相等
|
|||
|
|
b := make([]byte, length)
|
|||
|
|
r := make([]byte, length+(length/4)) // storage for random bytes. 存储随机字节
|
|||
|
|
|
|||
|
|
for i := 0; ; {
|
|||
|
|
// 将随机的byte值填充到byte数组中 以供使用
|
|||
|
|
if _, err := rand.Read(r); err != nil {
|
|||
|
|
log.Error(`GenerateRandString`, zap.Error(err))
|
|||
|
|
return ``
|
|||
|
|
}
|
|||
|
|
for _, rb := range r {
|
|||
|
|
c := int(rb)
|
|||
|
|
if c > maxRb {
|
|||
|
|
// Skip this number to avoid modulo bias.跳过这个数字以避免模偏倚
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
b[i] = chars[c%clen]
|
|||
|
|
i++
|
|||
|
|
if i == length { // 直到取到合适的长度
|
|||
|
|
return string(b)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GenerateBase62Key 生成指定长度的随机 Base62（数字 + 大小写字母）字符串
|
|||
|
|
func GenerateBase62Key(length int) (string, error) {
|
|||
|
|
var b strings.Builder
|
|||
|
|
b.Grow(length)
|
|||
|
|
|
|||
|
|
for i := 0; i < length; i++ {
|
|||
|
|
n, err := cryptoRand.Int(cryptoRand.Reader, big.NewInt(int64(len(base62Chars))))
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
b.WriteByte(base62Chars[n.Int64()])
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return b.String(), nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var sf *sonyflake.Sonyflake
|
|||
|
|
|
|||
|
|
func InitSnowflake() {
|
|||
|
|
sf = sonyflake.NewSonyflake(sonyflake.Settings{})
|
|||
|
|
if sf == nil {
|
|||
|
|
log.Fatalf("Failed to initialize sonyflake")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func GenerateTraceID() string {
|
|||
|
|
id, err := sf.NextID()
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalf("Failed to generate ID: %v", err)
|
|||
|
|
}
|
|||
|
|
return strconv.FormatUint(id, 10)
|
|||
|
|
}
|