Skip to content
Snippets Groups Projects
Commit e9dc96f9 authored by Yihao Dai's avatar Yihao Dai Committed by yefu.chen
Browse files

Fix bucket name in index builder and query node


Signed-off-by: default avatarbigsheeper <yihao.dai@zilliz.com>
parent af1941b6
No related branches found
No related tags found
No related merge requests found
......@@ -83,7 +83,7 @@ func CreateBuilder(ctx context.Context) (*Builder, error) {
return nil, err
}
b.kv, err = miniokv.NewMinIOKV(b.loopCtx, minIOClient, "milvus-distributed-indexbuilder")
b.kv, err = miniokv.NewMinIOKV(b.loopCtx, minIOClient, Params.MinioBucketName)
if err != nil {
return nil, err
}
......
......@@ -23,6 +23,7 @@ type ParamTable struct {
MinIOAccessKeyID string
MinIOSecretAccessKey string
MinIOUseSSL bool
MinioBucketName string
}
var Params ParamTable
......@@ -38,6 +39,7 @@ func (pt *ParamTable) Init() {
pt.initMinIOAccessKeyID()
pt.initMinIOSecretAccessKey()
pt.initMinIOUseSSL()
pt.initMinioBucketName()
}
func (pt *ParamTable) initAddress() {
......@@ -142,3 +144,11 @@ func (pt *ParamTable) initMinIOUseSSL() {
panic(err)
}
}
func (pt *ParamTable) initMinioBucketName() {
bucketName, err := pt.Load("minio.bucketName")
if err != nil {
panic(err)
}
pt.MinioBucketName = bucketName
}
......@@ -47,9 +47,7 @@ func newLoadIndexService(ctx context.Context, replica collectionReplica) *loadIn
panic(err)
}
// TODO: load bucketName from config
bucketName := "query-node-load-index-service-minio"
MinioKV, err := minioKV.NewMinIOKV(ctx1, minioClient, bucketName)
MinioKV, err := minioKV.NewMinIOKV(ctx1, minioClient, Params.MinioBucketName)
if err != nil {
panic(err)
}
......
......@@ -74,8 +74,7 @@ func TestLoadIndexService(t *testing.T) {
Secure: Params.MinioUseSSLStr,
})
assert.Equal(t, err, nil)
bucketName := "query-node-load-index-service-minio"
minioKV, err := minioKV.NewMinIOKV(node.queryNodeLoopCtx, minioClient, bucketName)
minioKV, err := minioKV.NewMinIOKV(node.queryNodeLoopCtx, minioClient, Params.MinioBucketName)
assert.Equal(t, err, nil)
indexPaths := make([]string, 0)
for _, index := range binarySet {
......
......@@ -26,6 +26,7 @@ type ParamTable struct {
MinioAccessKeyID string
MinioSecretAccessKey string
MinioUseSSLStr bool
MinioBucketName string
// dm
InsertChannelNames []string
......@@ -98,6 +99,7 @@ func (p *ParamTable) Init() {
p.initMinioAccessKeyID()
p.initMinioSecretAccessKey()
p.initMinioUseSSLStr()
p.initMinioBucketName()
p.initPulsarAddress()
p.initETCDAddress()
......@@ -174,6 +176,14 @@ func (p *ParamTable) initMinioUseSSLStr() {
p.MinioUseSSLStr = sslBoolean
}
func (p *ParamTable) initMinioBucketName() {
bucketName, err := p.Load("minio.bucketName")
if err != nil {
panic(err)
}
p.MinioBucketName = bucketName
}
func (p *ParamTable) initPulsarAddress() {
url, err := p.Load("_PulsarAddress")
if err != nil {
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment