added url query for creating schema_migrations table on cluster

Signed-off-by: Ildar Valiullin <[email protected]>
golang-migrate · May 22, 2021 · 547a2c9 · 547a2c9
1 parent 1695f40
commit 547a2c9
Showing 1 changed file with 17 additions and 6 deletions.
diff --git a/database/clickhouse/clickhouse.go b/database/clickhouse/clickhouse.go
@@ -30,6 +30,7 @@ var (
 
 type Config struct {
 	DatabaseName          string
+	ClusterName           string
 	MigrationsTable       string
 	MigrationsTableEngine string
 	MultiStatementEnabled bool
@@ -98,6 +99,7 @@ func (ch *ClickHouse) Open(dsn string) (database.Driver, error) {
 			MigrationsTable:       purl.Query().Get("x-migrations-table"),
 			MigrationsTableEngine: migrationsTableEngine,
 			DatabaseName:          purl.Query().Get("database"),
+			ClusterName:           purl.Query().Get("x-cluster-name"),
 			MultiStatementEnabled: purl.Query().Get("x-multi-statement") == "true",
 			MultiStatementMaxSize: multiStatementMaxSize,
 		},
@@ -227,12 +229,21 @@ func (ch *ClickHouse) ensureVersionTable() (err error) {
 	}
 
 	// if not, create the empty migration table
-	query = fmt.Sprintf(`
-		CREATE TABLE %s (
-			version    Int64,
-			dirty      UInt8,
-			sequence   UInt64
-		) Engine=%s`, ch.config.MigrationsTable, ch.config.MigrationsTableEngine)
+	if len(ch.config.ClusterName) > 0 {
+		query = fmt.Sprintf(`
+			CREATE TABLE %s ON CLUSTER %s (
+				version    Int64,
+				dirty      UInt8,
+				sequence   UInt64
+			) Engine=%s`, ch.config.MigrationsTable, ch.config.ClusterName, ch.config.MigrationsTableEngine)
+	} else {
+		query = fmt.Sprintf(`
+			CREATE TABLE %s (
+				version    Int64,
+				dirty      UInt8,
+				sequence   UInt64
+			) Engine=%s`, ch.config.MigrationsTable, ch.config.MigrationsTableEngine)
+	}
 
 	if strings.HasSuffix(ch.config.MigrationsTableEngine, "Tree") {
 		query = fmt.Sprintf(`%s ORDER BY sequence`, query)