fix: verify logic cluster when import cluster

housepower · Mar 19, 2024 · 7517874 · 7517874
1 parent 052c902
commit 7517874
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/service/clickhouse/clickhouse_service.go b/service/clickhouse/clickhouse_service.go
@@ -199,6 +199,18 @@ func GetCkClusterConfig(conf *model.CKManClickHouseConfig) (string, error) {
 		}
 	}
 
+	if conf.LogicCluster != nil {
+		query := fmt.Sprintf("SELECT count() FROM system.clusters WHERE cluster = '%s'", *conf.LogicCluster)
+		value, err = service.QueryInfo(query)
+		if err != nil {
+			return model.E_DATA_SELECT_FAILED, err
+		}
+		c := value[1][0].(uint64)
+		if c == 0 {
+			return model.E_RECORD_NOT_FOUND, fmt.Errorf("logic cluster %s not exist", *conf.LogicCluster)
+		}
+	}
+
 	value, err = service.QueryInfo("SELECT version()")
 	if err != nil {
 		return model.E_DATA_SELECT_FAILED, err