Merge pull request #572 from actiontech/feat-data-export-ce

Feat data export ce

Author: LordofAvernus

internal/dms/biz/db_service.go

@@ -484,8 +484,7 @@ func filterExportSupportedDb(dbServices []*DBService, opPermissionType dmsCommon
 	}
 	ret := make([]*DBService, 0)
 	for _, item := range dbServices {
-		_, ok := pkgConst.SupportedDataExportDBTypes[pkgConst.DBType(item.DBType)]
-		if ok {
+		if pkgConst.CheckDBTypeIfDataExportSupported(item.DBType) {
 			ret = append(ret, item)
 		}
 	}

internal/dms/pkg/constant/const.go

@@ -267,12 +267,23 @@ const (
 	DBTypeGaussDB        DBType = "GaussDB / openGauss"
 )
 
-var SupportedDataExportDBTypes = map[DBType]struct{}{
+var supportedDataExportDBTypes = map[DBType]struct{}{
 	DBTypeMySQL:          {},
 	DBTypePostgreSQL:     {},
 	DBTypeOracle:         {},
 	DBTypeSQLServer:      {},
 	DBTypeOceanBaseMySQL: {},
+	DBTypeHive:           {},
+	DBTypeDM:             {},
+}
+
+func CheckDBTypeIfDataExportSupported(dbtype string) bool {
+	dbType, err := ParseDBType(dbtype)
+	if err != nil {
+		return false
+	}
+	_, ok := supportedDataExportDBTypes[dbType]
+	return ok
 }
 
 type FilterCondition struct {