fix(go): correct ingestion and validation fixtures

go/bulk_ingest.go

@@ -17,6 +17,7 @@ package databricks
 import (
 	"context"
 	"database/sql/driver"
+	"encoding/hex"
 	"fmt"
 	"strings"
 
@@ -52,6 +53,8 @@ func (s *statementImpl) executeIngest(ctx context.Context) (int64, error) {
 
 	totalRows := int64(0)
 	params := make([]driver.NamedValue, s.boundStream.Schema().NumFields())
+	errorHelper := s.ErrorHelper
+	errorHelper.ErrorInspector = DatabricksErrorInspector{ingest: true}
 
 	for s.boundStream.Next() {
 		recordBatch := s.boundStream.RecordBatch()
@@ -70,7 +73,7 @@ func (s *statementImpl) executeIngest(ctx context.Context) (int64, error) {
 			// Use ExecContext directly instead of PrepareContext because Databricks doesn't do server-side statement preparation
 			result, err := s.conn.conn.ExecContext(ctx, insertSQL, valuesToInterfaces(params)...)
 			if err != nil {
-				return totalRows, s.ErrorHelper.Errorf(adbc.StatusInternal, "failed to execute the query: %v", err)
+				return totalRows, errorHelper.WrapInternal(err, "failed to execute the query")
 			}
 
 			rows, _ := result.RowsAffected()
@@ -97,7 +100,7 @@ func (s *statementImpl) createTableIfNeeded(ctx context.Context, tableName strin
 	case adbc.OptionValueIngestModeReplace:
 		dropSQL := fmt.Sprintf("DROP TABLE IF EXISTS %s", tableName)
 		if _, err := s.conn.conn.ExecContext(ctx, dropSQL); err != nil {
-			return s.ErrorHelper.Errorf(adbc.StatusInternal, "failed to drop the table: %v", err)
+			return s.ErrorHelper.WrapInternal(err, "failed to drop the table")
 		}
 		return s.createTable(ctx, tableName, schema, false)
 
@@ -134,7 +137,7 @@ func (s *statementImpl) createTable(ctx context.Context, tableName string, schem
 
 	_, err := s.conn.conn.ExecContext(ctx, sql.String())
 	if err != nil {
-		return s.ErrorHelper.Errorf(adbc.StatusInternal, "failed to create table: %v", err)
+		return s.ErrorHelper.WrapInternal(err, "failed to create table")
 	}
 	return nil
 }
@@ -161,10 +164,11 @@ func buildInsertSQL(tableName string, schema *arrow.Schema) (string, error) {
 			sql.WriteString(", ")
 		}
 
-		if field.Type.ID() == arrow.FIXED_SIZE_BINARY {
-			// Use UNHEX() to convert hex string to binary
+		switch field.Type.ID() {
+		case arrow.BINARY, arrow.LARGE_BINARY, arrow.BINARY_VIEW, arrow.FIXED_SIZE_BINARY:
+			// Send hex strings to preserve arbitrary bytes through text parameters.
 			sql.WriteString("UNHEX(?)")
-		} else {
+		default:
 			sql.WriteString("?")
 		}
 	}
@@ -253,14 +257,13 @@ func extractGoValue(arr arrow.Array, idx int) (any, error) {
 		return arr.(*array.StringView).Value(idx), nil
 
 	case arrow.BINARY:
-		return arr.(*array.Binary).Value(idx), nil
+		return hex.EncodeToString(arr.(*array.Binary).Value(idx)), nil
 	case arrow.LARGE_BINARY:
-		return arr.(*array.LargeBinary).Value(idx), nil
+		return hex.EncodeToString(arr.(*array.LargeBinary).Value(idx)), nil
 	case arrow.BINARY_VIEW:
-		return arr.(*array.BinaryView).Value(idx), nil
+		return hex.EncodeToString(arr.(*array.BinaryView).Value(idx)), nil
 	case arrow.FIXED_SIZE_BINARY:
-		// Convert to hex string for use with UNHEX() SQL function
-		return fmt.Sprintf("%x", arr.(*array.FixedSizeBinary).Value(idx)), nil
+		return hex.EncodeToString(arr.(*array.FixedSizeBinary).Value(idx)), nil
 
 	case arrow.DATE32:
 		return arr.(*array.Date32).Value(idx).ToTime(), nil

go/driver.go

@@ -89,8 +89,10 @@ func NewDriver(alloc memory.Allocator) driverbase.DriverWithContext {
 		panic(err)
 	}
 
+	base := driverbase.NewDriverImplBase(info, alloc)
+	base.ErrorHelper.ErrorInspector = DatabricksErrorInspector{}
 	return driverbase.NewDriver(&driverImpl{
-		DriverImplBase: driverbase.NewDriverImplBase(info, alloc),
+		DriverImplBase: base,
 	})
 }
 

go/error_inspector.go

@@ -0,0 +1,52 @@
+// Copyright (c) 2026 ADBC Drivers Contributors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//         http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package databricks
+
+import (
+	"errors"
+
+	"github.com/apache/arrow-adbc/go/adbc"
+	dbsqlerr "github.com/databricks/databricks-sql-go/errors"
+)
+
+type DatabricksErrorInspector struct {
+	ingest bool
+}
+
+// InspectError maps Databricks execution errors to ADBC errors and preserves SQLSTATE.
+func (d DatabricksErrorInspector) InspectError(err error, defaultStatus adbc.Status) adbc.Error {
+	adbcErr := adbc.Error{Code: defaultStatus, Msg: err.Error()}
+	var executionErr dbsqlerr.DBExecutionError
+	for errors.As(err, &executionErr) {
+		if sqlState := executionErr.SqlState(); sqlState != "" {
+			copy(adbcErr.SqlState[:], sqlState)
+			switch sqlState {
+			case "42P07": // TABLE_OR_VIEW_ALREADY_EXISTS
+				adbcErr.Code = adbc.StatusAlreadyExists
+			case "42703": // UNRESOLVED_COLUMN
+				adbcErr.Code = adbc.StatusInvalidArgument
+				if d.ingest {
+					// A missing destination column means the existing table's
+					// schema conflicts with the ingest input.
+					adbcErr.Code = adbc.StatusAlreadyExists
+				}
+			}
+			break
+		}
+		// A wrapper may have no SQLSTATE even when its cause has one.
+		err = errors.Unwrap(executionErr)
+	}
+	return adbcErr
+}

go/statement.go

@@ -90,7 +90,7 @@ func (s *statementImpl) Prepare(ctx context.Context) error {
 
 	stmt, err := s.conn.conn.PrepareContext(ctx, s.query)
 	if err != nil {
-		return s.ErrorHelper.Errorf(adbc.StatusInvalidState, "failed to prepare statement: %v", err)
+		return s.ErrorHelper.WrapInvalidState(err, "failed to prepare statement")
 	}
 
 	s.prepared = stmt
@@ -150,7 +150,7 @@ func executeQuery(ctx context.Context, conn *sql.Conn, query string, args []driv
 	})
 
 	if err != nil {
-		return nil, errorHelper.Errorf(adbc.StatusInternal, "failed to execute query: %v", err)
+		return nil, errorHelper.WrapInternal(err, "failed to execute query")
 	}
 
 	defer func() {
@@ -241,7 +241,7 @@ func (s *statementImpl) executeUpdate(ctx context.Context, query string, args []
 		result, err = s.conn.conn.ExecContext(ctx, query, values...)
 	}
 	if err != nil {
-		return -1, s.ErrorHelper.Errorf(adbc.StatusInternal, "failed to execute update: %v", err)
+		return -1, s.ErrorHelper.WrapInternal(err, "failed to execute update")
 	}
 
 	rowsAffected, err := result.RowsAffected()

go/validation/queries/ingest/decimal_scale_negative.toml

@@ -0,0 +1,17 @@
+# Copyright (c) 2026 ADBC Drivers Contributors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# DECIMAL scale must be between zero and precision.
+# https://docs.databricks.com/aws/en/sql/language-manual/data-types/decimal-type
+hide = true

go/validation/queries/ingest/timestamp_ns.txtcase

@@ -15,6 +15,8 @@
 // part: metadata
 
 [tags]
+partial-support = true
+caveats = ["Nanosecond timestamps are truncated to microseconds."]
 sql-type-name = "TIMESTAMP_NTZ"
 
 // part: expected_schema

go/validation/queries/ingest/timestamptz_ns.txtcase

@@ -15,6 +15,8 @@
 // part: metadata
 
 [tags]
+partial-support = true
+caveats = ["Nanosecond timestamps are truncated to microseconds."]
 sql-type-name = "TIMESTAMP"
 
 // part: expected_schema

go/validation/queries/type/bind/float64.txtcase

@@ -0,0 +1,22 @@
+// Copyright (c) 2026 ADBC Drivers Contributors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//         http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// part: metadata
+
+[tags]
+sql-type-name = "DOUBLE"
+
+// part: setup_query
+
+CREATE TABLE test_float64 (res DOUBLE);

go/validation/queries/type/bind/timestamp_ms.txtcase

@@ -12,6 +12,11 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// part: metadata
+
+[tags]
+sql-type-name = "TIMESTAMP_NTZ"
+
 // part: setup_query
 
 CREATE TABLE test_timestamp (idx INT, res TIMESTAMP_NTZ);

go/validation/queries/type/bind/timestamp_ns.txtcase

@@ -12,6 +12,13 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// part: metadata
+
+[tags]
+sql-type-name = "TIMESTAMP_NTZ"
+partial-support = true
+caveats = ["Nanosecond timestamps are truncated to microseconds."]
+
 // part: setup_query
 
 CREATE TABLE test_timestamp (idx INT, res TIMESTAMP_NTZ);

go/validation/queries/type/bind/timestamp_s.txtcase

@@ -12,6 +12,11 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// part: metadata
+
+[tags]
+sql-type-name = "TIMESTAMP_NTZ"
+
 // part: setup_query
 
 CREATE TABLE test_timestamp (idx INT, res TIMESTAMP_NTZ);

go/validation/queries/type/bind/timestamp_us.txtcase

@@ -12,6 +12,11 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// part: metadata
+
+[tags]
+sql-type-name = "TIMESTAMP_NTZ"
+
 // part: setup_query
 
 CREATE TABLE test_timestamp (idx INT, res TIMESTAMP_NTZ);

go/validation/queries/type/bind/timestamptz_ms.txtcase

@@ -12,6 +12,15 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// part: metadata
+
+[tags]
+sql-type-name = "TIMESTAMP"
+
+// part: setup_query
+
+CREATE TABLE test_timestamptz (idx INT, res TIMESTAMP, res2 TIMESTAMP);
+
 // part: expected_schema
 
 {"format": "+s", "children": [

go/validation/queries/type/bind/timestamptz_ns.txtcase

@@ -12,6 +12,17 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// part: metadata
+
+[tags]
+sql-type-name = "TIMESTAMP"
+partial-support = true
+caveats = ["Nanosecond timestamps are truncated to microseconds."]
+
+// part: setup_query
+
+CREATE TABLE test_timestamptz (idx INT, res TIMESTAMP, res2 TIMESTAMP);
+
 // part: expected_schema
 
 {"format": "+s", "children": [

go/validation/queries/type/bind/timestamptz_s.txtcase

@@ -12,6 +12,15 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// part: metadata
+
+[tags]
+sql-type-name = "TIMESTAMP"
+
+// part: setup_query
+
+CREATE TABLE test_timestamptz (idx INT, res TIMESTAMP, res2 TIMESTAMP);
+
 // part: expected_schema
 
 {"format": "+s", "children": [

go/validation/queries/type/bind/timestamptz_us.txtcase

@@ -12,6 +12,15 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// part: metadata
+
+[tags]
+sql-type-name = "TIMESTAMP"
+
+// part: setup_query
+
+CREATE TABLE test_timestamptz (idx INT, res TIMESTAMP, res2 TIMESTAMP);
+
 // part: expected_schema
 
 {"format": "+s", "children": [

go/validation/queries/type/select/timestamp0.toml

@@ -0,0 +1,17 @@
+# Copyright (c) 2026 ADBC Drivers Contributors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Databricks timestamp precision starts at 6 fractional digits.
+# https://docs.databricks.com/aws/en/sql/language-manual/data-types/timestamp-type
+hide = true

go/validation/queries/type/select/timestamp0tz.toml

@@ -0,0 +1,17 @@
+# Copyright (c) 2026 ADBC Drivers Contributors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Databricks timestamp precision starts at 6 fractional digits.
+# https://docs.databricks.com/aws/en/sql/language-manual/data-types/timestamp-type
+hide = true

go/validation/queries/type/select/timestamp1.toml

@@ -0,0 +1,17 @@
+# Copyright (c) 2026 ADBC Drivers Contributors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Databricks timestamp precision starts at 6 fractional digits.
+# https://docs.databricks.com/aws/en/sql/language-manual/data-types/timestamp-type
+hide = true