Add Close to StoreClient
parent
836d5af9e4
commit
389bda7d40
|
|
@ -18,6 +18,10 @@ func NewBolt(path string) (*BoltClient, error) {
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (bc *BoltClient) Close() error {
|
||||||
|
return bc.db.Close()
|
||||||
|
}
|
||||||
|
|
||||||
func (bc *BoltClient) Set(namespace, key string, value []byte) error {
|
func (bc *BoltClient) Set(namespace, key string, value []byte) error {
|
||||||
return bc.db.Update(func(tx *bolt.Tx) error {
|
return bc.db.Update(func(tx *bolt.Tx) error {
|
||||||
bucket, err := tx.CreateBucketIfNotExists([]byte(namespace))
|
bucket, err := tx.CreateBucketIfNotExists([]byte(namespace))
|
||||||
|
|
|
||||||
|
|
@ -45,6 +45,7 @@ func Test_BoltSetListGet(t *testing.T) {
|
||||||
defer os.Remove(tmp.Name())
|
defer os.Remove(tmp.Name())
|
||||||
var sc Client
|
var sc Client
|
||||||
sc, err = NewBolt(tmp.Name())
|
sc, err = NewBolt(tmp.Name())
|
||||||
|
defer sc.Close()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Errorf("failed to create bolt %v: %v", sc, err)
|
t.Errorf("failed to create bolt %v: %v", sc, err)
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -4,4 +4,5 @@ type Client interface {
|
||||||
Get(string, string) ([]byte, error)
|
Get(string, string) ([]byte, error)
|
||||||
Set(string, string, []byte) error
|
Set(string, string, []byte) error
|
||||||
List(string, string) ([]string, error)
|
List(string, string) ([]string, error)
|
||||||
|
Close() error
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue