Skip to content

Commit

Permalink
Add argument ticket to the getStreamStatement methods
Browse files Browse the repository at this point in the history
  • Loading branch information
jcralmeida authored and vfraga committed Mar 29, 2022
1 parent 6a16ac3 commit 93d08e1
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ default void getStream(CallContext context, Ticket ticket, ServerStreamListener

if (command.is(TicketStatementQuery.class)) {
getStreamStatement(
FlightSqlUtils.unpackOrThrow(command, TicketStatementQuery.class), context, listener);
FlightSqlUtils.unpackOrThrow(command, TicketStatementQuery.class), context, listener, ticket);
} else if (command.is(CommandPreparedStatementQuery.class)) {
getStreamPreparedStatement(
FlightSqlUtils.unpackOrThrow(command, CommandPreparedStatementQuery.class), context, listener);
Expand Down Expand Up @@ -354,8 +354,8 @@ SchemaResult getSchemaStatement(CommandStatementQuery command, CallContext conte
* @param context Per-call context.
* @param listener An interface for sending data back to the client.
*/
void getStreamStatement(TicketStatementQuery ticket, CallContext context,
ServerStreamListener listener);
void getStreamStatement(TicketStatementQuery ticketStatementQuery, CallContext context,
ServerStreamListener listener, Ticket ticket);

/**
* Returns data for a particular prepared statement query instance.
Expand Down

0 comments on commit 93d08e1

Please sign in to comment.