[Snowflake] Fixing DSN for custom host (#62)

artie-labs · Mar 7, 2023 · 1c601b4 · 1c601b4
1 parent ec33415
commit 1c601b4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/clients/snowflake/snowflake.go b/clients/snowflake/snowflake.go
@@ -105,14 +105,21 @@ func (s *Store) Merge(ctx context.Context, tableData *optimization.TableData) er
 }
 
 func (s *Store) ReestablishConnection(ctx context.Context) {
-	dsn, err := gosnowflake.DSN(&gosnowflake.Config{
+	cfg := &gosnowflake.Config{
 		Account:   config.GetSettings().Config.Snowflake.AccountID,
 		User:      config.GetSettings().Config.Snowflake.Username,
 		Password:  config.GetSettings().Config.Snowflake.Password,
 		Warehouse: config.GetSettings().Config.Snowflake.Warehouse,
 		Region:    config.GetSettings().Config.Snowflake.Region,
-	})
+	}
+
+	if config.GetSettings().Config.Snowflake.Host != "" {
+		// If the host is specified
+		cfg.Host = config.GetSettings().Config.Snowflake.Host
+		cfg.Region = ""
+	}
 
+	dsn, err := gosnowflake.DSN(cfg)
 	if err != nil {
 		logger.FromContext(ctx).Fatalf("failed to get snowflake dsn, err: %v", err)
 	}

diff --git a/lib/config/config.go b/lib/config/config.go
@@ -48,6 +48,7 @@ type Config struct {
 		Password  string `yaml:"password"`
 		Warehouse string `yaml:"warehouse"`
 		Region    string `yaml:"region"`
+		Host      string `yaml:"host"`
 	}
 
 	Reporting struct {