mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-12-05 11:39:02 +08:00
57 lines
1.2 KiB
Go
57 lines
1.2 KiB
Go
|
package leveldb
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"fmt"
|
||
|
|
||
|
"github.com/chrislusf/seaweedfs/weed/filer"
|
||
|
"github.com/syndtr/goleveldb/leveldb"
|
||
|
)
|
||
|
|
||
|
func (store *LevelDB2Store) KvPut(ctx context.Context, key []byte, value []byte) (err error) {
|
||
|
|
||
|
partitionId := bucketKvKey(key, store.dbCount)
|
||
|
|
||
|
err = store.dbs[partitionId].Put(key, value, nil)
|
||
|
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("kv bucket %d put: %v", partitionId, err)
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (store *LevelDB2Store) KvGet(ctx context.Context, key []byte) (value []byte, err error) {
|
||
|
|
||
|
partitionId := bucketKvKey(key, store.dbCount)
|
||
|
|
||
|
value, err = store.dbs[partitionId].Get(key, nil)
|
||
|
|
||
|
if err == leveldb.ErrNotFound {
|
||
|
return nil, filer.ErrKvNotFound
|
||
|
}
|
||
|
|
||
|
if err != nil {
|
||
|
return nil, fmt.Errorf("kv bucket %d get: %v", partitionId, err)
|
||
|
}
|
||
|
|
||
|
return
|
||
|
}
|
||
|
|
||
|
func (store *LevelDB2Store) KvDelete(ctx context.Context, key []byte) (err error) {
|
||
|
|
||
|
partitionId := bucketKvKey(key, store.dbCount)
|
||
|
|
||
|
err = store.dbs[partitionId].Delete(key, nil)
|
||
|
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("kv bucket %d delete: %v", partitionId, err)
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func bucketKvKey(key []byte, dbCount int) (partitionId int) {
|
||
|
return int(key[len(key)-1]) % dbCount
|
||
|
}
|