feat: the format of key is now constant with that of OpenAI

This commit is contained in:
JustSong
2023-06-03 10:53:25 +08:00
parent 98f1a627f0
commit 2847a08852
4 changed files with 29 additions and 4 deletions

View File

@@ -5,6 +5,7 @@ import (
"github.com/google/uuid"
"html/template"
"log"
"math/rand"
"net"
"os/exec"
"runtime"
@@ -133,6 +134,29 @@ func GetUUID() string {
return code
}
const keyChars = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
func init() {
rand.Seed(time.Now().UnixNano())
}
func GenerateKey() string {
rand.Seed(time.Now().UnixNano())
key := make([]byte, 48)
for i := 0; i < 16; i++ {
key[i] = keyChars[rand.Intn(len(keyChars))]
}
uuid_ := GetUUID()
for i := 0; i < 32; i++ {
c := uuid_[i]
if i%2 == 0 && c >= 'a' && c <= 'z' {
c = c - 'a' + 'A'
}
key[i+16] = c
}
return string(key)
}
func GetTimestamp() int64 {
return time.Now().Unix()
}