mirror of
https://github.com/chrislusf/seaweedfs
synced 2024-08-24 18:50:55 +02:00
55 lines
1.4 KiB
Go
55 lines
1.4 KiB
Go
package arangodb
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/arangodb/go-driver"
|
|
"github.com/seaweedfs/seaweedfs/weed/filer"
|
|
"github.com/seaweedfs/seaweedfs/weed/glog"
|
|
)
|
|
|
|
func (store *ArangodbStore) KvPut(ctx context.Context, key []byte, value []byte) (err error) {
|
|
model := &Model{
|
|
Key: hashString(".kvstore." + string(key)),
|
|
Directory: ".kvstore." + string(key),
|
|
Meta: bytesToArray(value),
|
|
}
|
|
|
|
exists, err := store.kvCollection.DocumentExists(ctx, model.Key)
|
|
if err != nil {
|
|
return fmt.Errorf("kv put: %v", err)
|
|
}
|
|
if exists {
|
|
_, err = store.kvCollection.UpdateDocument(ctx, model.Key, model)
|
|
} else {
|
|
_, err = store.kvCollection.CreateDocument(ctx, model)
|
|
}
|
|
if err != nil {
|
|
return fmt.Errorf("kv put: %v", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
func (store *ArangodbStore) KvGet(ctx context.Context, key []byte) (value []byte, err error) {
|
|
var model Model
|
|
_, err = store.kvCollection.ReadDocument(ctx, hashString(".kvstore."+string(key)), &model)
|
|
if driver.IsNotFound(err) {
|
|
return nil, filer.ErrKvNotFound
|
|
}
|
|
if err != nil {
|
|
glog.Errorf("kv get: %s %v", string(key), err)
|
|
return nil, filer.ErrKvNotFound
|
|
}
|
|
return arrayToBytes(model.Meta), nil
|
|
}
|
|
|
|
func (store *ArangodbStore) KvDelete(ctx context.Context, key []byte) (err error) {
|
|
_, err = store.kvCollection.RemoveDocument(ctx, hashString(".kvstore."+string(key)))
|
|
if err != nil {
|
|
glog.Errorf("kv del: %v", err)
|
|
return filer.ErrKvNotFound
|
|
}
|
|
return nil
|
|
}
|