Change Token KV func to common func.

Add File based KV func
pull/3924/head
Season Shi 2023-03-23 00:22:15 +08:00
parent 83c86f0e1c
commit 0169188d27
No known key found for this signature in database
GPG Key ID: 024D6B1E861D7FB0
1 changed files with 37 additions and 3 deletions

View File

@ -15,6 +15,40 @@ func fileInit() *badger.DB {
return kv
}
func setFileKV(key string, value string) bool {
kv := fileInit()
err := kv.Update(func(txn *badger.Txn) error {
err := txn.Set([]byte(key), []byte(value))
return err
})
if err != nil {
log.Fatal(err)
return false
}
return true
}
func getFileKV(key string) string {
kv := fileInit()
var valCopy []byte
err := kv.View(func(txn *badger.Txn) error {
item, err := txn.Get([]byte(key))
if err != nil {
return err
}
err = item.Value(func(val []byte) error {
valCopy = append([]byte{}, val...)
return nil
})
return err
})
if err != nil {
log.Fatal(err)
return ""
}
return string(valCopy)
}
func memInit() *badger.DB {
opt := badger.DefaultOptions("").WithInMemory(true)
kv, err := badger.Open(opt)
@ -25,10 +59,10 @@ func memInit() *badger.DB {
return kv
}
func setTokenKV(key string, token string) bool {
func setMemKV(key string, value string) bool {
kv := memInit()
err := kv.Update(func(txn *badger.Txn) error {
err := txn.Set([]byte(key), []byte(token))
err := txn.Set([]byte(key), []byte(value))
return err
})
if err != nil {
@ -38,7 +72,7 @@ func setTokenKV(key string, token string) bool {
return true
}
func getTokenKV(key string) string {
func getMemKV(key string) string {
kv := memInit()
var valCopy []byte
err := kv.View(func(txn *badger.Txn) error {