Clean up.

artie-labs · Oct 2, 2024 · 34e8099 · 34e8099
1 parent 042d8ef
commit 34e8099
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/lib/config/constants/constants.go b/lib/config/constants/constants.go
@@ -78,6 +78,7 @@ const (
 // TODO: Add Databricks to this list once it's ready
 var ValidDestinations = []DestinationKind{
 	BigQuery,
+	Databricks,
 	MSSQL,
 	Redshift,
 	S3,

diff --git a/lib/config/destination_types.go b/lib/config/destination_types.go
@@ -13,7 +13,7 @@ type BigQuery struct {
 
 type Databricks struct {
 	Host                string `yaml:"host"`
-	HttpPath            string `json:"httpPath"`
+	HttpPath            string `yaml:"httpPath"`
 	Port                int    `yaml:"port"`
 	Catalog             string `yaml:"catalog"`
 	PersonalAccessToken string `yaml:"personalAccessToken"`

diff --git a/lib/destination/utils/load.go b/lib/destination/utils/load.go
@@ -3,6 +3,8 @@ package utils
 import (
 	"fmt"
 
+	"github.com/artie-labs/transfer/clients/databricks"
+
 	"github.com/artie-labs/transfer/clients/bigquery"
 	"github.com/artie-labs/transfer/clients/mssql"
 	"github.com/artie-labs/transfer/clients/redshift"
@@ -45,6 +47,8 @@ func LoadDataWarehouse(cfg config.Config, store *db.Store) (destination.DataWare
 		return s, nil
 	case constants.BigQuery:
 		return bigquery.LoadBigQuery(cfg, store)
+	case constants.Databricks:
+		return databricks.LoadStore(cfg)
 	case constants.MSSQL:
 		s, err := mssql.LoadStore(cfg)
 		if err != nil {