mirror of
https://github.com/tgdrive/teldrive.git
synced 2024-11-14 19:56:33 +08:00
38 lines
677 B
Go
38 lines
677 B
Go
package kv
|
|
|
|
import (
|
|
"go.etcd.io/bbolt"
|
|
)
|
|
|
|
type Bolt struct {
|
|
bucket []byte
|
|
db *bbolt.DB
|
|
}
|
|
|
|
func (b *Bolt) Get(key string) ([]byte, error) {
|
|
var val []byte
|
|
|
|
if err := b.db.View(func(tx *bbolt.Tx) error {
|
|
val = tx.Bucket(b.bucket).Get([]byte(key))
|
|
return nil
|
|
}); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if val == nil {
|
|
return nil, ErrNotFound
|
|
}
|
|
return val, nil
|
|
}
|
|
|
|
func (b *Bolt) Set(key string, val []byte) error {
|
|
return b.db.Update(func(tx *bbolt.Tx) error {
|
|
return tx.Bucket(b.bucket).Put([]byte(key), val)
|
|
})
|
|
}
|
|
|
|
func (b *Bolt) Delete(key string) error {
|
|
return b.db.Update(func(tx *bbolt.Tx) error {
|
|
return tx.Bucket(b.bucket).Delete([]byte(key))
|
|
})
|
|
}
|