mirror of https://github.com/Xhofe/alist
parent
83c86f0e1c
commit
0169188d27
|
@ -15,6 +15,40 @@ func fileInit() *badger.DB {
|
||||||
return kv
|
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 {
|
func memInit() *badger.DB {
|
||||||
opt := badger.DefaultOptions("").WithInMemory(true)
|
opt := badger.DefaultOptions("").WithInMemory(true)
|
||||||
kv, err := badger.Open(opt)
|
kv, err := badger.Open(opt)
|
||||||
|
@ -25,10 +59,10 @@ func memInit() *badger.DB {
|
||||||
return kv
|
return kv
|
||||||
}
|
}
|
||||||
|
|
||||||
func setTokenKV(key string, token string) bool {
|
func setMemKV(key string, value string) bool {
|
||||||
kv := memInit()
|
kv := memInit()
|
||||||
err := kv.Update(func(txn *badger.Txn) error {
|
err := kv.Update(func(txn *badger.Txn) error {
|
||||||
err := txn.Set([]byte(key), []byte(token))
|
err := txn.Set([]byte(key), []byte(value))
|
||||||
return err
|
return err
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -38,7 +72,7 @@ func setTokenKV(key string, token string) bool {
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
func getTokenKV(key string) string {
|
func getMemKV(key string) string {
|
||||||
kv := memInit()
|
kv := memInit()
|
||||||
var valCopy []byte
|
var valCopy []byte
|
||||||
err := kv.View(func(txn *badger.Txn) error {
|
err := kv.View(func(txn *badger.Txn) error {
|
||||||
|
|
Loading…
Reference in New Issue