CH replicated dialect from config

chapsuk · Jan 29, 2023 · d1ca637 · d1ca637
1 parent 77b4b87
commit d1ca637
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/config/config.go b/config/config.go
@@ -134,6 +134,12 @@ func fillDBConfig(cfg *driver.Config) {
 		return
 	}
 
+	if viper.IsSet("clickhouse_replicated.dsn") {
+		cfg.Dialect = "clickhouse-replicated"
+		cfg.Dsn = viper.GetString("clickhouse_replicated.dsn")
+		return
+	}
+
 	if viper.IsSet("vertica.dsn") {
 		cfg.Dialect = "vertica"
 		cfg.Dsn = viper.GetString("vertica.dsn")

diff --git a/driver/goose/driver.go b/driver/goose/driver.go
@@ -10,7 +10,6 @@ import (
 type Goose struct {
 	db               *sql.DB
 	dir              string
-	dialect          string
 	versionTableName string
 }
 
@@ -23,19 +22,19 @@ func New(dialect, dsn, tableName, dir string) (*Goose, error) {
 	orig.SetTableName(tableName)
 	orig.SetLogger(&utils.StdLogger{})
 
+	driverName := dialect
 	if dialect == "clickhouse-replicated" {
-		dialect = "clickhouse"
+		driverName = "clickhouse"
 	}
 
-	db, err := sql.Open(dialect, dsn)
+	db, err := sql.Open(driverName, dsn)
 	if err != nil {
 		return nil, err
 	}
 
 	return &Goose{
 		db:               db,
 		dir:              dir,
-		dialect:          dialect,
 		versionTableName: tableName,
 	}, nil
 }