mirror of https://github.com/Xhofe/alist
parent
83c86f0e1c
commit
0169188d27
|
@ -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 {
|
||||
|
|
Loading…
Reference in New Issue