Merge pull request #2 from AntonOyung/CLOUDP-268006_2

AntonOyung · web-flow · commit b02bcbda8c12 · 2024-10-08T00:27:25.000-04:00
CLOUDP-268006: Add support for drop in cluster mode
diff --git a/database/clickhouse/clickhouse.go b/database/clickhouse/clickhouse.go
@@ -314,6 +314,10 @@ func (ch *ClickHouse) Drop() (err error) {
 
 		query = "DROP TABLE IF EXISTS " + quoteIdentifier(ch.config.DatabaseName) + "." + quoteIdentifier(table)
 
+		if len(ch.config.ClusterName) > 0 {
+			query = query + " ON CLUSTER " + ch.config.ClusterName
+		}
+
 		if _, err := ch.conn.Exec(query); err != nil {
 			return &database.Error{OrigErr: err, Query: []byte(query)}
 		}

Original file line number	Diff line number	Diff line change
`@@ -314,6 +314,10 @@ func (ch *ClickHouse) Drop() (err error) {`
`314`	`314`
`315`	`315`	`query = "DROP TABLE IF EXISTS " + quoteIdentifier(ch.config.DatabaseName) + "." + quoteIdentifier(table)`
`316`	`316`
	`317`	`+ if len(ch.config.ClusterName) > 0 {`
	`318`	`+ query = query + " ON CLUSTER " + ch.config.ClusterName`
	`319`	`+ }`
	`320`	`+`
`317`	`321`	`if _, err := ch.conn.Exec(query); err != nil {`
`318`	`322`	`return &database.Error{OrigErr: err, Query: []byte(query)}`
`319`	`323`	`}`