Merge pull request #289 from actiontech/fix-issue1659

cb add tbase
actiontech · Jul 31, 2024 · 9b5ffc8 · 9b5ffc8
2 parents 68e39d0 + 8de6532
commit 9b5ffc8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/internal/dms/biz/cloudbeaver.go b/internal/dms/biz/cloudbeaver.go
@@ -1188,7 +1188,7 @@ func (cu *CloudbeaverUsecase) GenerateCloudbeaverConnectionParams(dbService *DBS
 		err = cu.fillMySQLParams(config)
 	case constant.DBTypeTiDB:
 		err = cu.fillTiDBParams(config)
-	case constant.DBTypePostgreSQL:
+	case constant.DBTypePostgreSQL, constant.DBTypeTBase:
 		err = cu.fillPGSQLParams(config)
 	case constant.DBTypeSQLServer:
 		err = cu.fillMSSQLParams(config)

diff --git a/internal/dms/pkg/constant/const.go b/internal/dms/pkg/constant/const.go
@@ -127,6 +127,8 @@ func ParseDBType(s string) (DBType, error) {
 		return DBTypeOceanBaseMySQL, nil
 	case "GoldenDB":
 		return DBTypeGoldenDB, nil
+	case "TBase":
+		return DBTypeTBase, nil
 	default:
 		return "", fmt.Errorf("invalid db type: %s", s)
 	}
@@ -142,6 +144,7 @@ const (
 	DBTypeOceanBaseMySQL DBType = "OceanBase For MySQL"
 	DBTypeTDSQLForInnoDB DBType = "TDSQL For InnoDB"
 	DBTypeGoldenDB       DBType = "GoldenDB"
+	DBTypeTBase          DBType = "TBase"
 )
 
 type FilterCondition struct {