Skip to content

Commit c0e473a

Browse files
committed
Add handling for new class
Signed-off-by: nithinkdb <nithin.krishnamurthi@databricks.com>
1 parent 2c2cb73 commit c0e473a

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

connection.go

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -342,6 +342,13 @@ func (c *conn) executeStatement(ctx context.Context, query string, args []driver
342342
return resp, err
343343
}
344344

345+
type dbSqlParam struct {
346+
Ordinal *int32
347+
Name *string
348+
Type *string
349+
Value *string
350+
}
351+
345352
func namedValuesToTSparkParams(args []driver.NamedValue) []*cli_service.TSparkParameter {
346353
var ts []string = []string{"STRING", "DOUBLE", "BOOLEAN", "TIMESTAMP", "FLOAT", "INTEGER", "TINYINT", "SMALLINT", "BIGINT"}
347354
var params []*cli_service.TSparkParameter
@@ -412,6 +419,9 @@ func namedValuesToTSparkParams(args []driver.NamedValue) []*cli_service.TSparkPa
412419
s := t.String()
413420
param.Value.StringValue = &s
414421
param.Type = &ts[3]
422+
case dbSqlParam:
423+
param.Value.StringValue = t.Value
424+
param.Type = t.Type
415425
default:
416426
s := fmt.Sprintf("%s", arg.Value)
417427
param.Value.StringValue = &s

0 commit comments

Comments
 (0)