fix(mongodb): 副本集部署方案 #7704

# Reviewed, transaction id: 22531
TencentBlueKing · Nov 1, 2024 · 7d5e6bf · 7d5e6bf
1 parent 417a0ba
commit 7d5e6bf
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/dbm-ui/backend/db_services/dbresource/constants.py b/dbm-ui/backend/db_services/dbresource/constants.py
@@ -38,6 +38,7 @@
     ClusterType.TendisTwemproxyRedisInstance: TendisCacheSpecFilter,
     ClusterType.TwemproxyTendisSSDInstance: TendisSSDSpecFilter,
     ClusterType.MongoShardedCluster: MongoDBShardSpecFilter,
+    ClusterType.MongoReplicaSet: MongoDBShardSpecFilter,
 }
 
 

diff --git a/dbm-ui/backend/db_services/dbresource/handlers.py b/dbm-ui/backend/db_services/dbresource/handlers.py
@@ -349,11 +349,14 @@ def custom_filter(self):
 
 
 class MongoDBShardSpecFilter(object):
-    """mongodb分片集群的部署方案"""
+    """mongodb集群的部署方案"""
 
     def __init__(self, capacity, spec_cluster_type, spec_machine_type, **kwargs):
-        if spec_cluster_type != ClusterType.MongoShardedCluster or spec_machine_type != MachineType.MONGODB:
-            raise SpecOperateException(_("请保证输入的集群类型是MongoShardedCluster，且机器规格为mongodb"))
+        if (
+            spec_cluster_type not in [ClusterType.MongoShardedCluster, ClusterType.MongoReplicaSet]
+            or spec_machine_type != MachineType.MONGODB
+        ):
+            raise SpecOperateException(_("请保证输入的集群类型是{}，且机器规格为mongodb").format(spec_cluster_type))
 
         self.specs: List[Dict[str, Any]] = []
         mongodb_specs = Spec.objects.filter(