mirror of
https://github.com/chrislusf/seaweedfs
synced 2024-08-25 03:01:15 +02:00
69 lines
1.6 KiB
Go
69 lines
1.6 KiB
Go
//go:build elastic
|
|
// +build elastic
|
|
|
|
package elastic
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/seaweedfs/seaweedfs/weed/filer"
|
|
|
|
jsoniter "github.com/json-iterator/go"
|
|
elastic "github.com/olivere/elastic/v7"
|
|
"github.com/seaweedfs/seaweedfs/weed/glog"
|
|
)
|
|
|
|
func (store *ElasticStore) KvDelete(ctx context.Context, key []byte) (err error) {
|
|
deleteResult, err := store.client.Delete().
|
|
Index(indexKV).
|
|
Type(indexType).
|
|
Id(string(key)).
|
|
Do(ctx)
|
|
if err == nil {
|
|
if deleteResult.Result == "deleted" || deleteResult.Result == "not_found" {
|
|
return nil
|
|
}
|
|
}
|
|
glog.Errorf("delete key(id:%s) %v.", string(key), err)
|
|
return fmt.Errorf("delete key %v.", err)
|
|
}
|
|
|
|
func (store *ElasticStore) KvGet(ctx context.Context, key []byte) (value []byte, err error) {
|
|
searchResult, err := store.client.Get().
|
|
Index(indexKV).
|
|
Type(indexType).
|
|
Id(string(key)).
|
|
Do(ctx)
|
|
if elastic.IsNotFound(err) {
|
|
return value, filer.ErrKvNotFound
|
|
}
|
|
if searchResult != nil && searchResult.Found {
|
|
esEntry := &ESKVEntry{}
|
|
if err := jsoniter.Unmarshal(searchResult.Source, esEntry); err == nil {
|
|
return esEntry.Value, nil
|
|
}
|
|
}
|
|
glog.Errorf("find key(%s),%v.", string(key), err)
|
|
return value, filer.ErrKvNotFound
|
|
}
|
|
|
|
func (store *ElasticStore) KvPut(ctx context.Context, key []byte, value []byte) (err error) {
|
|
esEntry := &ESKVEntry{value}
|
|
val, err := jsoniter.Marshal(esEntry)
|
|
if err != nil {
|
|
glog.Errorf("insert key(%s) %v.", string(key), err)
|
|
return fmt.Errorf("insert key %v.", err)
|
|
}
|
|
_, err = store.client.Index().
|
|
Index(indexKV).
|
|
Type(indexType).
|
|
Id(string(key)).
|
|
BodyJson(string(val)).
|
|
Do(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("kv put: %v", err)
|
|
}
|
|
return nil
|
|
}
|