Skip to content
Draft
Changes from 4 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions go/adbc/driver/flightsql/flightsql_statement.go
Original file line number Diff line number Diff line change
Expand Up @@ -514,6 +514,14 @@ func (s *statement) ExecuteQuery(ctx context.Context) (rdr array.RecordReader, n
var header, trailer metadata.MD
opts := append([]grpc.CallOption{}, grpc.Header(&header), grpc.Trailer(&trailer), s.timeouts)
if s.prepared != nil {
if isUpdate := s.prepared.IsUpdate(); isUpdate != nil && *isUpdate {
nrec, err = s.prepared.ExecuteUpdate(ctx, opts...)
if err != nil {
return nil, -1, adbcFromFlightStatusWithDetails(err, header, trailer, "ExecuteUpdate")
}
rdr, err = array.NewRecordReader(arrow.NewSchema(nil, nil), nil)
return
}
info, err = s.prepared.Execute(ctx, opts...)
} else {
info, err = s.query.execute(ctx, s.cnxn, opts...)
Expand Down