Read bucket name from config yaml

Signed-off-by: cai.zhang <cai.zhang@zilliz.com>

conf/conf.go

@@ -71,6 +71,7 @@ type Writer struct {
 	Parallelism        int
 	TopicStart         int
 	TopicEnd           int
+	Bucket             string
 }
 
 type ServerConfig struct {

conf/config.yaml

@@ -57,6 +57,7 @@ writer:
   parallelism: 100
   topicstart: 0
   topicend: 128
+  bucket: "zilliz-hz"
 
 proxy:
   timezone: UTC+8

storage/internal/S3/s3_engine.go

@@ -6,12 +6,13 @@ import (
 	"github.com/aws/aws-sdk-go/aws"
 	"github.com/aws/aws-sdk-go/aws/session"
 	"github.com/aws/aws-sdk-go/service/s3"
-//	"github.com/aws/aws-sdk-go/service/s3/s3manager"
+	"github.com/czs007/suvlim/conf"
+	//	"github.com/aws/aws-sdk-go/service/s3/s3manager"
 	. "github.com/czs007/suvlim/storage/pkg/types"
 	"io"
 )
 
-var bucketName = "zilliz-hz"
+var bucketName = conf.Config.Writer.Bucket
 
 type S3Store struct {
 	client *s3.S3

storage/internal/minio/minio_storeEngine.go

@@ -3,12 +3,13 @@ package minio_driver
 import (
 	"bytes"
 	"context"
+	"github.com/czs007/suvlim/conf"
+	. "github.com/czs007/suvlim/storage/pkg/types"
 	"github.com/minio/minio-go/v7"
 	"io"
-	. "github.com/czs007/suvlim/storage/pkg/types"
 )
 
-var bucketName = "zilliz"
+var bucketName = conf.Config.Writer.Bucket
 
 type minioStore struct {
 	client *minio.Client