feat(backend): 支持单据类型和集群类型的映射 TencentBlueKing#6818

dbm-ui/backend/ticket/builders/common/bigdata.py

@@ -16,6 +16,7 @@
 from rest_framework.serializers import ValidationError
 
 from backend.configuration.constants import AffinityEnum, DBType
+from backend.db_meta.enums import ClusterType
 from backend.db_meta.enums.cluster_phase import ClusterPhase
 from backend.db_meta.models.cluster import Cluster
 from backend.db_meta.models.instance import StorageInstance
@@ -269,26 +270,32 @@ def post_callback(self):
 
 class BaseEsTicketFlowBuilder(BigDataTicketFlowBuilderPatchMixin, TicketFlowBuilder):
     group = DBType.Es.value
+    cluster_types = [ClusterType.Es.value]
 
 
 class BaseDorisTicketFlowBuilder(BigDataTicketFlowBuilderPatchMixin, TicketFlowBuilder):
     group = DBType.Doris.value
+    cluster_types = [ClusterType.Doris.value]
 
 
 class BaseHdfsTicketFlowBuilder(BigDataTicketFlowBuilderPatchMixin, TicketFlowBuilder):
     group = DBType.Hdfs.value
+    cluster_types = [ClusterType.Hdfs.value]
 
 
 class BaseKafkaTicketFlowBuilder(BigDataTicketFlowBuilderPatchMixin, TicketFlowBuilder):
     group = DBType.Kafka.value
+    cluster_types = [ClusterType.Kafka.value]
 
 
 class BasePulsarTicketFlowBuilder(BigDataTicketFlowBuilderPatchMixin, TicketFlowBuilder):
     group = DBType.Pulsar.value
+    cluster_types = [ClusterType.Pulsar.value]
 
 
 class BaseInfluxDBTicketFlowBuilder(InfluxdbTicketFlowBuilderPatchMixin, TicketFlowBuilder):
     group = DBType.InfluxDB.value
+    cluster_types = [ClusterType.Influxdb.value]
 
 
 class BaseCloudTicketFlowBuilder(TicketFlowBuilder):

dbm-ui/backend/ticket/builders/mongodb/base.py

@@ -16,6 +16,7 @@
 from rest_framework import serializers
 
 from backend.configuration.constants import AffinityEnum, DBType
+from backend.db_meta.enums import ClusterType
 from backend.db_meta.models import Cluster, Machine
 from backend.db_services.mongodb.resources.query import MongoDBListRetrieveResource
 from backend.flow.utils.mongodb.db_table_filter import MongoDbTableFilter
@@ -44,6 +45,17 @@ def validate(self, attrs):
 
 class BaseMongoDBTicketFlowBuilder(MongoDBTicketFlowBuilderPatchMixin, TicketFlowBuilder):
     group = DBType.MongoDB.value
+    cluster_types = [ClusterType.MongoReplicaSet.value, ClusterType.MongoShardedCluster.value]
+
+
+class BaseMongoReplicaSetTicketFlowBuilder(MongoDBTicketFlowBuilderPatchMixin, TicketFlowBuilder):
+    group = DBType.MongoDB.value
+    cluster_types = [ClusterType.MongoReplicaSet.value]
+
+
+class BaseMongoShardedTicketFlowBuilder(MongoDBTicketFlowBuilderPatchMixin, TicketFlowBuilder):
+    group = DBType.MongoDB.value
+    cluster_types = [ClusterType.MongoShardedCluster.value]
 
 
 class BaseMongoDBOperateDetailSerializer(SkipToRepresentationMixin, serializers.Serializer):

dbm-ui/backend/ticket/builders/mongodb/mongo_add_mongos.py

@@ -20,8 +20,8 @@
 from backend.ticket.builders.mongodb.base import (
     BaseMongoDBOperateDetailSerializer,
     BaseMongoDBOperateResourceParamBuilder,
-    BaseMongoDBTicketFlowBuilder,
     BaseMongoOperateFlowParamBuilder,
+    BaseMongoShardedTicketFlowBuilder,
 )
 from backend.ticket.constants import TicketType
 
@@ -68,7 +68,7 @@ def post_callback(self):
 
 
 @builders.BuilderFactory.register(TicketType.MONGODB_ADD_MONGOS, is_apply=True)
-class MongoDBAddMongosApplyFlowBuilder(BaseMongoDBTicketFlowBuilder):
+class MongoDBAddMongosApplyFlowBuilder(BaseMongoShardedTicketFlowBuilder):
     serializer = MongoDBAddMongosDetailSerializer
     inner_flow_builder = MongoDBAddMongosFlowParamBuilder
     inner_flow_name = _("MongoDB 扩容接入层执行")

dbm-ui/backend/ticket/builders/mongodb/mongo_reduce_mongos.py

@@ -18,8 +18,8 @@
 from backend.ticket.builders.common.base import CommonValidate, HostInfoSerializer
 from backend.ticket.builders.mongodb.base import (
     BaseMongoDBOperateDetailSerializer,
-    BaseMongoDBTicketFlowBuilder,
     BaseMongoOperateFlowParamBuilder,
+    BaseMongoShardedTicketFlowBuilder,
 )
 from backend.ticket.constants import TicketType
 
@@ -74,7 +74,7 @@ def format_ticket_data(self):
 
 
 @builders.BuilderFactory.register(TicketType.MONGODB_REDUCE_MONGOS)
-class MongoDBAddMongosApplyFlowBuilder(BaseMongoDBTicketFlowBuilder):
+class MongoDBAddMongosApplyFlowBuilder(BaseMongoShardedTicketFlowBuilder):
     serializer = MongoDBReduceMongosDetailSerializer
     inner_flow_builder = MongoDBReduceMongosFlowParamBuilder
     inner_flow_name = _("MongoDB 缩容接入层执行")

dbm-ui/backend/ticket/builders/mongodb/mongo_replicaset_apply.py

@@ -18,7 +18,10 @@
 from backend.flow.engine.controller.mongodb import MongoDBController
 from backend.iam_app.dataclass.actions import ActionEnum
 from backend.ticket import builders
-from backend.ticket.builders.mongodb.base import BaseMongoDBOperateResourceParamBuilder, BaseMongoDBTicketFlowBuilder
+from backend.ticket.builders.mongodb.base import (
+    BaseMongoDBOperateResourceParamBuilder,
+    BaseMongoReplicaSetTicketFlowBuilder,
+)
 from backend.ticket.constants import TicketType
 
 
@@ -95,7 +98,7 @@ def post_callback(self):
     cluster_type=ClusterType.MongoReplicaSet,
     iam=ActionEnum.MONGODB_APPLY,
 )
-class MongoReplicaSetApplyFlowBuilder(BaseMongoDBTicketFlowBuilder):
+class MongoReplicaSetApplyFlowBuilder(BaseMongoReplicaSetTicketFlowBuilder):
     serializer = MongoReplicaSetApplyDetailSerializer
     inner_flow_builder = MongoReplicaSetApplyFlowParamBuilder
     inner_flow_name = _("MongoDB 副本集集群部署执行")

dbm-ui/backend/ticket/builders/mongodb/mongo_scale_updown.py

@@ -22,7 +22,7 @@
 from backend.ticket.builders.mongodb.base import (
     BaseMongoDBOperateDetailSerializer,
     BaseMongoDBOperateResourceParamBuilder,
-    BaseMongoDBTicketFlowBuilder,
+    BaseMongoShardedTicketFlowBuilder,
 )
 from backend.ticket.builders.mongodb.mongo_backup import MongoDBBackupFlowParamBuilder
 from backend.ticket.constants import TicketType
@@ -109,7 +109,7 @@ def post_callback(self):
 
 
 @builders.BuilderFactory.register(TicketType.MONGODB_SCALE_UPDOWN, is_apply=True)
-class MongoDBScaleUpDownFlowBuilder(BaseMongoDBTicketFlowBuilder):
+class MongoDBScaleUpDownFlowBuilder(BaseMongoShardedTicketFlowBuilder):
     serializer = MongoDBScaleUpDownDetailSerializer
     inner_flow_builder = MongoDBScaleUpDownFlowParamBuilder
     inner_flow_name = _("MongoDB 集群容量变更执行")

dbm-ui/backend/ticket/builders/mongodb/mongo_shard_apply.py

@@ -18,7 +18,10 @@
 from backend.flow.engine.controller.mongodb import MongoDBController
 from backend.iam_app.dataclass.actions import ActionEnum
 from backend.ticket import builders
-from backend.ticket.builders.mongodb.base import BaseMongoDBOperateResourceParamBuilder, BaseMongoDBTicketFlowBuilder
+from backend.ticket.builders.mongodb.base import (
+    BaseMongoDBOperateResourceParamBuilder,
+    BaseMongoShardedTicketFlowBuilder,
+)
 from backend.ticket.constants import TicketType
 
 
@@ -95,7 +98,7 @@ def post_callback(self):
     cluster_type=ClusterType.MongoShardedCluster,
     iam=ActionEnum.MONGODB_APPLY,
 )
-class MongoShardedClusterApplyFlowBuilder(BaseMongoDBTicketFlowBuilder):
+class MongoShardedClusterApplyFlowBuilder(BaseMongoShardedTicketFlowBuilder):
     serializer = MongoShardedClusterApplyDetailSerializer
     inner_flow_builder = MongoShardedClusterApplyFlowParamBuilder
     inner_flow_name = _("MongoDB 分片集群部署执行")

dbm-ui/backend/ticket/builders/mysql/base.py

@@ -35,6 +35,17 @@
 
 class BaseMySQLTicketFlowBuilder(MySQLTicketFlowBuilderPatchMixin, TicketFlowBuilder):
     group = DBType.MySQL.value
+    cluster_types = [ClusterType.TenDBSingle.value, ClusterType.TenDBHA.value]
+
+
+class BaseMySQLSingleTicketFlowBuilder(MySQLTicketFlowBuilderPatchMixin, TicketFlowBuilder):
+    group = DBType.MySQL.value
+    cluster_types = [ClusterType.TenDBSingle.value]
+
+
+class BaseMySQLHATicketFlowBuilder(MySQLTicketFlowBuilderPatchMixin, TicketFlowBuilder):
+    group = DBType.MySQL.value
+    cluster_types = [ClusterType.TenDBHA.value]
 
 
 class MySQLBasePauseParamBuilder(builders.PauseParamBuilder):

dbm-ui/backend/ticket/builders/mysql/mysql_add_slave.py

@@ -18,7 +18,7 @@
 from backend.ticket import builders
 from backend.ticket.builders.common.base import BaseOperateResourceParamBuilder, HostInfoSerializer
 from backend.ticket.builders.common.constants import MySQLBackupSource
-from backend.ticket.builders.mysql.base import BaseMySQLTicketFlowBuilder, MySQLBaseOperateDetailSerializer
+from backend.ticket.builders.mysql.base import BaseMySQLHATicketFlowBuilder, MySQLBaseOperateDetailSerializer
 from backend.ticket.constants import TicketType
 
 
@@ -82,7 +82,7 @@ def post_callback(self):
 
 
 @builders.BuilderFactory.register(TicketType.MYSQL_ADD_SLAVE, is_apply=True)
-class MysqlAddSlaveFlowBuilder(BaseMySQLTicketFlowBuilder):
+class MysqlAddSlaveFlowBuilder(BaseMySQLHATicketFlowBuilder):
     serializer = MysqlAddSlaveDetailSerializer
     inner_flow_builder = MysqlAddSlaveParamBuilder
     inner_flow_name = _("添加从库执行")

dbm-ui/backend/ticket/builders/mysql/mysql_checksum.py

@@ -25,7 +25,7 @@
 from backend.ticket.builders.common.constants import MySQLChecksumTicketMode, MySQLDataRepairTriggerMode
 from backend.ticket.builders.common.field import DBTimezoneField
 from backend.ticket.builders.mysql.base import (
-    BaseMySQLTicketFlowBuilder,
+    BaseMySQLHATicketFlowBuilder,
     DBTableField,
     MySQLBaseOperateDetailSerializer,
 )
@@ -137,7 +137,7 @@ def format_ticket_data(self):
 
 
 @builders.BuilderFactory.register(TicketType.MYSQL_CHECKSUM)
-class MySQLChecksumFlowBuilder(BaseMySQLTicketFlowBuilder):
+class MySQLChecksumFlowBuilder(BaseMySQLHATicketFlowBuilder):
     serializer = MySQLChecksumDetailSerializer
     # 流程构造类
     checksum_flow_builder = MySQLChecksumFlowParamBuilder

dbm-ui/backend/ticket/builders/mysql/mysql_flashback.py

@@ -20,7 +20,7 @@
 from backend.ticket.builders.common.constants import MYSQL_BINLOG_ROLLBACK
 from backend.ticket.builders.common.field import DBTimezoneField
 from backend.ticket.builders.mysql.base import (
-    BaseMySQLTicketFlowBuilder,
+    BaseMySQLHATicketFlowBuilder,
     DBTableField,
     MySQLBaseOperateDetailSerializer,
 )
@@ -72,7 +72,7 @@ class MySQLFlashbackFlowParamBuilder(builders.FlowParamBuilder):
 
 
 @builders.BuilderFactory.register(TicketType.MYSQL_FLASHBACK)
-class MySQLFlashbackFlowBuilder(BaseMySQLTicketFlowBuilder):
+class MySQLFlashbackFlowBuilder(BaseMySQLHATicketFlowBuilder):
     serializer = MySQLFlashbackDetailSerializer
     inner_flow_builder = MySQLFlashbackFlowParamBuilder
     inner_flow_name = _("闪回执行")

dbm-ui/backend/ticket/builders/mysql/mysql_ha_apply.py

@@ -23,7 +23,7 @@
 from backend.flow.engine.controller.mysql import MySQLController
 from backend.iam_app.dataclass.actions import ActionEnum
 from backend.ticket.builders import BuilderFactory
-from backend.ticket.builders.mysql.base import BaseMySQLTicketFlowBuilder
+from backend.ticket.builders.mysql.base import BaseMySQLHATicketFlowBuilder
 from backend.ticket.builders.mysql.mysql_single_apply import (
     MysqlSingleApplyDetailSerializer,
     MysqlSingleApplyFlowParamBuilder,
@@ -124,7 +124,7 @@ def post_callback(self):
 @BuilderFactory.register(
     TicketType.MYSQL_HA_APPLY, is_apply=True, cluster_type=ClusterType.TenDBHA, iam=ActionEnum.MYSQL_APPLY
 )
-class MysqlHAApplyFlowBuilder(BaseMySQLTicketFlowBuilder):
+class MysqlHAApplyFlowBuilder(BaseMySQLHATicketFlowBuilder):
     serializer = MysqlHAApplyDetailSerializer
     inner_flow_builder = MysqlHAApplyFlowParamBuilder
     inner_flow_name = _("MySQL高可用部署执行")

dbm-ui/backend/ticket/builders/mysql/mysql_ha_backup.py

@@ -15,7 +15,7 @@
 from backend.flow.engine.controller.mysql import MySQLController
 from backend.ticket import builders
 from backend.ticket.builders.mysql.base import (
-    BaseMySQLTicketFlowBuilder,
+    BaseMySQLHATicketFlowBuilder,
     DBTableField,
     MySQLBaseOperateDetailSerializer,
 )
@@ -51,7 +51,7 @@ class MySQLHaBackupFlowParamBuilder(builders.FlowParamBuilder):
 
 
 @builders.BuilderFactory.register(TicketType.MYSQL_HA_DB_TABLE_BACKUP)
-class MySQLHaBackupFlowBuilder(BaseMySQLTicketFlowBuilder):
+class MySQLHaBackupFlowBuilder(BaseMySQLHATicketFlowBuilder):
     serializer = MySQLHaBackupDetailSerializer
     inner_flow_builder = MySQLHaBackupFlowParamBuilder
     inner_flow_name = _("库表备份执行")

dbm-ui/backend/ticket/builders/mysql/mysql_ha_clear.py

@@ -17,7 +17,7 @@
 from backend.flow.engine.controller.mysql import MySQLController
 from backend.ticket import builders
 from backend.ticket.builders.mysql.base import (
-    BaseMySQLTicketFlowBuilder,
+    BaseMySQLHATicketFlowBuilder,
     DBTableField,
     MySQLBaseOperateDetailSerializer,
 )
@@ -56,7 +56,7 @@ def format_ticket_data(self):
 
 
 @builders.BuilderFactory.register(TicketType.MYSQL_HA_TRUNCATE_DATA)
-class MySQLHaClearFlowBuilder(BaseMySQLTicketFlowBuilder):
+class MySQLHaClearFlowBuilder(BaseMySQLHATicketFlowBuilder):
     serializer = MySQLHaClearDetailSerializer
     inner_flow_builder = MySQLHaClearFlowParamBuilder
     inner_flow_name = _("清档执行")

dbm-ui/backend/ticket/builders/mysql/mysql_ha_destroy.py

@@ -18,7 +18,7 @@
 from backend.flow.engine.controller.tbinlogdumper import TBinlogDumperController
 from backend.iam_app.dataclass.actions import ActionEnum
 from backend.ticket import builders
-from backend.ticket.builders.mysql.base import BaseMySQLTicketFlowBuilder, MySQLClustersTakeDownDetailsSerializer
+from backend.ticket.builders.mysql.base import BaseMySQLHATicketFlowBuilder, MySQLClustersTakeDownDetailsSerializer
 from backend.ticket.builders.tbinlogdumper.dumper_reduce_nodes import TbinlogdumperReduceNodesFlowParamBuilder
 from backend.ticket.constants import FlowRetryType, FlowType, TicketType
 from backend.ticket.models import Flow
@@ -43,7 +43,7 @@ def format_ticket_data(self):
 @builders.BuilderFactory.register(
     TicketType.MYSQL_HA_DESTROY, phase=ClusterPhase.DESTROY, iam=ActionEnum.MYSQL_DESTROY
 )
-class MysqlHaDestroyFlowBuilder(BaseMySQLTicketFlowBuilder):
+class MysqlHaDestroyFlowBuilder(BaseMySQLHATicketFlowBuilder):
     """Mysql下架流程的构建基类"""
 
     serializer = MysqlHADestroyDetailSerializer

dbm-ui/backend/ticket/builders/mysql/mysql_ha_disable.py

@@ -15,7 +15,7 @@
 from backend.flow.engine.controller.mysql import MySQLController
 from backend.iam_app.dataclass.actions import ActionEnum
 from backend.ticket import builders
-from backend.ticket.builders.mysql.base import BaseMySQLTicketFlowBuilder, MySQLClustersTakeDownDetailsSerializer
+from backend.ticket.builders.mysql.base import BaseMySQLHATicketFlowBuilder, MySQLClustersTakeDownDetailsSerializer
 from backend.ticket.constants import FlowRetryType, TicketType
 
 
@@ -30,7 +30,7 @@ class MysqlHADisableFlowParamBuilder(builders.FlowParamBuilder):
 @builders.BuilderFactory.register(
     TicketType.MYSQL_HA_DISABLE, phase=ClusterPhase.OFFLINE, iam=ActionEnum.MYSQL_ENABLE_DISABLE
 )
-class MysqlHaDisableFlowBuilder(BaseMySQLTicketFlowBuilder):
+class MysqlHaDisableFlowBuilder(BaseMySQLHATicketFlowBuilder):
     """Mysql下架流程的构建基类"""
 
     serializer = MysqlHADisableDetailSerializer

dbm-ui/backend/ticket/builders/mysql/mysql_ha_enable.py

@@ -15,7 +15,7 @@
 from backend.flow.engine.controller.mysql import MySQLController
 from backend.iam_app.dataclass.actions import ActionEnum
 from backend.ticket import builders
-from backend.ticket.builders.mysql.base import BaseMySQLTicketFlowBuilder, MySQLClustersTakeDownDetailsSerializer
+from backend.ticket.builders.mysql.base import BaseMySQLHATicketFlowBuilder, MySQLClustersTakeDownDetailsSerializer
 from backend.ticket.constants import FlowRetryType, TicketType
 
 
@@ -30,7 +30,7 @@ class MysqlHAEnableFlowParamBuilder(builders.FlowParamBuilder):
 @builders.BuilderFactory.register(
     TicketType.MYSQL_HA_ENABLE, phase=ClusterPhase.ONLINE, iam=ActionEnum.MYSQL_ENABLE_DISABLE
 )
-class MysqlHaEnableFlowBuilder(BaseMySQLTicketFlowBuilder):
+class MysqlHaEnableFlowBuilder(BaseMySQLHATicketFlowBuilder):
     """Mysql下架流程的构建基类"""
 
     serializer = MysqlHAEnableDetailSerializer

dbm-ui/backend/ticket/builders/mysql/mysql_ha_full_backup.py

@@ -18,7 +18,7 @@
 from backend.flow.engine.controller.mysql import MySQLController
 from backend.ticket import builders
 from backend.ticket.builders.common.base import fetch_cluster_ids
-from backend.ticket.builders.mysql.base import BaseMySQLTicketFlowBuilder, MySQLBaseOperateDetailSerializer
+from backend.ticket.builders.mysql.base import BaseMySQLHATicketFlowBuilder, MySQLBaseOperateDetailSerializer
 from backend.ticket.constants import FlowRetryType, TicketType
 
 
@@ -61,7 +61,7 @@ class MySQLHaFullBackupFlowParamBuilder(builders.FlowParamBuilder):
 
 
 @builders.BuilderFactory.register(TicketType.MYSQL_HA_FULL_BACKUP)
-class MySQLHaFullBackupFlowBuilder(BaseMySQLTicketFlowBuilder):
+class MySQLHaFullBackupFlowBuilder(BaseMySQLHATicketFlowBuilder):
     serializer = MySQLHaFullBackupDetailSerializer
     inner_flow_builder = MySQLHaFullBackupFlowParamBuilder
     inner_flow_name = _("全库备份执行")

dbm-ui/backend/ticket/builders/mysql/mysql_ha_metadata_import.py

@@ -21,7 +21,7 @@
 from backend.db_meta.models import AppCache, DBModule, Spec
 from backend.flow.engine.controller.mysql import MySQLController
 from backend.ticket import builders
-from backend.ticket.builders.mysql.base import BaseMySQLTicketFlowBuilder, MySQLBaseOperateDetailSerializer
+from backend.ticket.builders.mysql.base import BaseMySQLHATicketFlowBuilder, MySQLBaseOperateDetailSerializer
 from backend.ticket.constants import FlowRetryType, TicketType
 
 logger = logging.getLogger("root")
@@ -237,7 +237,7 @@ class TenDBHAMetadataImportFlowParamBuilder(builders.FlowParamBuilder):
 
 
 @builders.BuilderFactory.register(TicketType.MYSQL_HA_METADATA_IMPORT)
-class TenDBHAMetadataImportFlowBuilder(BaseMySQLTicketFlowBuilder):
+class TenDBHAMetadataImportFlowBuilder(BaseMySQLHATicketFlowBuilder):
     serializer = TenDBHAMetadataImportDetailSerializer
     inner_flow_builder = TenDBHAMetadataImportFlowParamBuilder
     inner_flow_name = _("MySQL高可用元数据导入")

dbm-ui/backend/ticket/builders/mysql/mysql_ha_rename.py

@@ -18,7 +18,7 @@
 from backend.ticket import builders
 from backend.ticket.builders.common.base import CommonValidate
 from backend.ticket.builders.mysql.base import (
-    BaseMySQLTicketFlowBuilder,
+    BaseMySQLHATicketFlowBuilder,
     DBTableField,
     MySQLBaseOperateDetailSerializer,
 )
@@ -61,7 +61,7 @@ def format_ticket_data(self):
 
 
 @builders.BuilderFactory.register(TicketType.MYSQL_HA_RENAME_DATABASE)
-class MySQLHaRenameFlowBuilder(BaseMySQLTicketFlowBuilder):
+class MySQLHaRenameFlowBuilder(BaseMySQLHATicketFlowBuilder):
     serializer = MySQLHaRenameSerializer
     inner_flow_builder = MySQLHaRenameFlowParamBuilder
     inner_flow_name = _("DB重命名执行")