feat: bcs-common mongodb支持配置replicaset

TencentBlueKing · Dec 11, 2024 · 10f70e4 · 10f70e4
1 parent daf7299
commit 10f70e4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/bcs-common/pkg/odm/drivers/mongo/mongo.go b/bcs-common/pkg/odm/drivers/mongo/mongo.go
@@ -40,6 +40,7 @@ type Options struct {
 	MaxPoolSize           uint64
 	MinPoolSize           uint64
 	Hosts                 []string
+	Replicaset            string
 	Monitor               *event.CommandMonitor
 }
 
@@ -67,6 +68,9 @@ func NewDB(opt *Options) (*DB, error) {
 		Hosts:   opt.Hosts,
 		Monitor: opt.Monitor,
 	}
+	if opt.Replicaset != "" {
+		mCliOpt.ReplicaSet = &opt.Replicaset
+	}
 	if opt.MaxPoolSize != 0 {
 		mCliOpt.MaxPoolSize = &opt.MaxPoolSize
 	}