Skip to content

Commit

Permalink
Clean up.
Browse files Browse the repository at this point in the history
  • Loading branch information
Tang8330 committed Oct 3, 2024
1 parent 759db4d commit 599e168
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 6 deletions.
4 changes: 2 additions & 2 deletions clients/databricks/dialect/dialect.go
Original file line number Diff line number Diff line change
Expand Up @@ -37,8 +37,8 @@ func (DatabricksDialect) BuildAlterColumnQuery(tableID sql.TableIdentifier, colu
return fmt.Sprintf("ALTER TABLE %s %s COLUMN %s", tableID.FullyQualifiedName(), columnOp, colSQLPart)
}

func (DatabricksDialect) BuildIsNotToastValueExpression(tableAlias constants.TableAlias, column columns.Column) string {
colName := sql.QuoteTableAliasColumn(tableAlias, column, sd)
func (d DatabricksDialect) BuildIsNotToastValueExpression(tableAlias constants.TableAlias, column columns.Column) string {
colName := sql.QuoteTableAliasColumn(tableAlias, column, d)
if column.KindDetails == typing.Struct {
return fmt.Sprintf("COALESCE(%s != {'key': '%s'}, true)", colName, constants.ToastUnavailableValuePlaceholder)
}
Expand Down
6 changes: 2 additions & 4 deletions clients/databricks/dialect/typing.go
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ func (DatabricksDialect) DataTypeForKind(kindDetails typing.KindDetails, _ bool)
case typing.Integer.Kind:
return "BIGINT"
case typing.Struct.Kind:
return "VARIANT"
return "STRING"
case typing.Array.Kind:
// Databricks requires arrays to be typed. As such, we're going to use an array of strings.
return "ARRAY<string>"
Expand Down Expand Up @@ -56,7 +56,7 @@ func (DatabricksDialect) KindForDataType(rawType string, _ string) (typing.KindD
}

switch rawType {
case "string", "binary":
case "string", "binary", "variant", "object":
return typing.String, nil
case "bigint":
return typing.KindDetails{Kind: typing.Integer.Kind, OptionalIntegerKind: typing.ToPtr(typing.BigIntegerKind)}, nil
Expand All @@ -74,8 +74,6 @@ func (DatabricksDialect) KindForDataType(rawType string, _ string) (typing.KindD
return typing.NewKindDetailsFromTemplate(typing.ETime, ext.TimestampTzKindType), nil
case "timestamp_ntz":
return typing.NewKindDetailsFromTemplate(typing.ETime, ext.TimestampTzKindType), nil
case "variant", "object":
return typing.Struct, nil
}

return typing.Invalid, fmt.Errorf("unsupported data type: %q", rawType)
Expand Down

0 comments on commit 599e168

Please sign in to comment.