121 lines
2.1 KiB
Go
121 lines
2.1 KiB
Go
|
//
|
|||
|
|
// str_helper.go
|
|||
|
|
// Copyright (C) 2022 tiglog <me@tiglog.com>
|
|||
|
|
//
|
|||
|
|
// Distributed under terms of the MIT license.
|
|||
|
|
//
|
|||
|
|
|
|||
|
|
package helper
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"math/rand"
|
|||
|
|
"time"
|
|||
|
|
"unicode"
|
|||
|
|
|
|||
|
|
"github.com/rs/xid"
|
|||
|
|
"hexq.cn/tiglog/golib/crypto/gmd5"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 是否是字符串
|
|||
|
|
func IsString(v interface{}) bool {
|
|||
|
|
_, ok := v.(string)
|
|||
|
|
return ok
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 随机字符串
|
|||
|
|
func RandString(n int) string {
|
|||
|
|
letterRunes := []rune("1234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")
|
|||
|
|
rand.Seed(time.Now().UnixNano())
|
|||
|
|
b := make([]rune, n)
|
|||
|
|
for i := range b {
|
|||
|
|
b[i] = letterRunes[rand.Intn(len(letterRunes))]
|
|||
|
|
}
|
|||
|
|
return string(b)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 首字母大写
|
|||
|
|
func UcFirst(str string) string {
|
|||
|
|
for _, v := range str {
|
|||
|
|
u := string(unicode.ToUpper(v))
|
|||
|
|
return u + str[len(u):]
|
|||
|
|
}
|
|||
|
|
return ""
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 首字母小写
|
|||
|
|
func LcFirst(str string) string {
|
|||
|
|
for _, v := range str {
|
|||
|
|
u := string(unicode.ToLower(v))
|
|||
|
|
return u + str[len(u):]
|
|||
|
|
}
|
|||
|
|
return ""
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 乱序字符串
|
|||
|
|
func Shuffle(str string) string {
|
|||
|
|
if str == "" {
|
|||
|
|
return str
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
runes := []rune(str)
|
|||
|
|
index := 0
|
|||
|
|
|
|||
|
|
for i := len(runes) - 1; i > 0; i-- {
|
|||
|
|
index = rand.Intn(i + 1)
|
|||
|
|
|
|||
|
|
if i != index {
|
|||
|
|
runes[i], runes[index] = runes[index], runes[i]
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return string(runes)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 反序字符串
|
|||
|
|
func Reverse(str string) string {
|
|||
|
|
n := len(str)
|
|||
|
|
runes := make([]rune, n)
|
|||
|
|
for _, r := range str {
|
|||
|
|
n--
|
|||
|
|
runes[n] = r
|
|||
|
|
}
|
|||
|
|
return string(runes[n:])
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 唯一字符串
|
|||
|
|
// 返回字符串长度为 20
|
|||
|
|
func GenId() string {
|
|||
|
|
guid := xid.New()
|
|||
|
|
return guid.String()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 可变长度的唯一字符串
|
|||
|
|
// 长度太短，可能就不唯一了
|
|||
|
|
// 长度大于等于 16 为最佳
|
|||
|
|
// 长度小于20时，为 GenId 的值 md5 后的前缀，因此，理论上前6位也能在大多数情况
|
|||
|
|
// 下唯一
|
|||
|
|
func Uniq(l int) string {
|
|||
|
|
if l <= 0 {
|
|||
|
|
panic("wrong length param")
|
|||
|
|
}
|
|||
|
|
ret := GenId()
|
|||
|
|
hl := len(ret)
|
|||
|
|
if l < hl {
|
|||
|
|
t, err := gmd5.EncryptString(ret)
|
|||
|
|
if err != nil {
|
|||
|
|
return ret[hl-l:]
|
|||
|
|
}
|
|||
|
|
return t[:l]
|
|||
|
|
}
|
|||
|
|
mhac_len := 6
|
|||
|
|
pl := len(ret)
|
|||
|
|
var hash string
|
|||
|
|
for l > pl {
|
|||
|
|
hash = GenId()
|
|||
|
|
hash = hash[mhac_len:]
|
|||
|
|
ret += hash
|
|||
|
|
pl += len(hash)
|
|||
|
|
}
|
|||
|
|
// log.Println("ret=", ret, ", pl=", pl, ", l=", l)
|
|||
|
|
return ret[0:l]
|
|||
|
|
}
|