You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

75 lines
1.3 KiB

package baseCode
import (
"strings"
)
const (
BASE = "E8S2DZX9WYLTN6BQF7CP5IK3MJUAR4HV"
DECIMAL = 32
PAD = "A"
LEN = 8
)
// id转code
func Encode(uid uint64) string {
id := uid
mod := uint64(0)
res := ""
for id != 0 {
mod = id % DECIMAL
id = id / DECIMAL
res += string(BASE[mod])
}
resLen := len(res)
if resLen < LEN {
res += PAD
for i := 0; i < LEN-resLen-1; i++ {
res += string(BASE[(int(uid)+i)%DECIMAL])
}
}
return res
}
func ReverseString(str string) string {
strArr := []rune(str)
for i := 0; i < len(strArr)/2; i++ {
strArr[len(strArr)-1-i], strArr[i] = strArr[i], strArr[len(strArr)-1-i]
}
return string(strArr)
}
func Decode(code string) uint64 {
res := uint64(0)
lenCode := len(code)
baseArr := []byte(BASE) // 字符串进制转换为byte数组
baseRev := make(map[byte]int) // 进制数据键值转换为map
for k, v := range baseArr {
baseRev[v] = k
}
// 查找补位字符的位置
isPad := strings.Index(ReverseString(code), PAD)
if isPad != -1 {
lenCode = len(code) - 1 - isPad
}
r := 0
for i := 0; i < lenCode; i++ {
// 补充字符直接跳过
// if string(code[i]) == PAD {
// continue
// }
index, ok := baseRev[code[i]]
if !ok {
return 0
}
b := uint64(1)
for j := 0; j < r; j++ {
b *= DECIMAL
}
res += uint64(index) * b
r++
}
return res
}