admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-28 22:45:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

[skip ci]Add minio comment (#8165)

Browse Source 
Signed-off-by: godchen <qingxiang.chen@zilliz.com>
This commit is contained in:
godchen 2021-09-17 23:59:47 +08:00 committed by GitHub
parent 266db4c883
commit 74bb712527
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
internal/kv/minio/minio_kv.go
View File

@ -41,6 +41,7 @@ type Option struct {
CreateBucket bool // when bucket not existed, create it
}
// NewMinIOKV creates MinIOKV to save and load object to MinIOKV.
func NewMinIOKV(ctx context.Context, option *Option) (*MinIOKV, error) {
var minIOClient *minio.Client
var err error
@ -86,11 +87,13 @@ func NewMinIOKV(ctx context.Context, option *Option) (*MinIOKV, error) {
return kv, nil
}
// Exist check whether a key exists in MinIO.
func (kv *MinIOKV) Exist(key string) bool {
_, err := kv.minioClient.StatObject(kv.ctx, kv.bucketName, key, minio.StatObjectOptions{})
return err == nil
}
// LoadWithPrefix load objects with the same prefix @key from minio .
func (kv *MinIOKV) LoadWithPrefix(key string) ([]string, []string, error) {
objects := kv.minioClient.ListObjects(kv.ctx, kv.bucketName, minio.ListObjectsOptions{Prefix: key})
@ -108,6 +111,7 @@ func (kv *MinIOKV) LoadWithPrefix(key string) ([]string, []string, error) {
return objectsKeys, objectsValues, nil
}
// LoadWithPrefix load an object with @key.
func (kv *MinIOKV) Load(key string) (string, error) {
object, err := kv.minioClient.GetObject(kv.ctx, kv.bucketName, key, minio.GetObjectOptions{})
if object != nil {
@ -158,6 +162,7 @@ func (kv *MinIOKV) FGetObjects(keys []string, localPath string) error {
return nil
}
// MultiLoad loads objects with multi @keys.
func (kv *MinIOKV) MultiLoad(keys []string) ([]string, error) {
var resultErr error
var objectsValues []string
@ -174,6 +179,7 @@ func (kv *MinIOKV) MultiLoad(keys []string) ([]string, error) {
return objectsValues, resultErr
}
// Save object with @key to Minio. Object value is @value.
func (kv *MinIOKV) Save(key, value string) error {
reader := strings.NewReader(value)
_, err := kv.minioClient.PutObject(kv.ctx, kv.bucketName, key, reader, int64(len(value)), minio.PutObjectOptions{})
@ -185,6 +191,8 @@ func (kv *MinIOKV) Save(key, value string) error {
return err
}
// Save MultiObject, the path is the key of @kvs. The object value is the value
// of @kvs.
func (kv *MinIOKV) MultiSave(kvs map[string]string) error {
var resultErr error
for key, value := range kvs {
@ -198,6 +206,7 @@ func (kv *MinIOKV) MultiSave(kvs map[string]string) error {
return resultErr
}
// LoadWithPrefix remove all objects with the same prefix @prefix from minio .
func (kv *MinIOKV) RemoveWithPrefix(prefix string) error {
objectsCh := make(chan minio.ObjectInfo)
@ -217,11 +226,13 @@ func (kv *MinIOKV) RemoveWithPrefix(prefix string) error {
return nil
}
// Remove delete an object with @key.
func (kv *MinIOKV) Remove(key string) error {
err := kv.minioClient.RemoveObject(kv.ctx, kv.bucketName, string(key), minio.RemoveObjectOptions{})
return err
}
// MultiRemove delete a objects with @keys.
func (kv *MinIOKV) MultiRemove(keys []string) error {
var resultErr error
for _, key := range keys {