apacheGH-37720: [Java][FlightSQL] Implement stateless prepared statem…

…ents PR update
stevelorddremio · May 16, 2024 · cf3ad66 · cf3ad66
1 parent 367a159
commit cf3ad66
Showing 1 changed file with 15 additions and 11 deletions.
diff --git a/java/flight/flight-sql/src/main/java/org/apache/arrow/flight/sql/FlightSqlClient.java b/java/flight/flight-sql/src/main/java/org/apache/arrow/flight/sql/FlightSqlClient.java
@@ -1057,17 +1057,21 @@ public FlightInfo execute(final CallOption... options) {
 
  if (parameterBindingRoot != null && parameterBindingRoot.getRowCount() > 0) {
  try (final SyncPutListener putListener = putParameters(descriptor, options)) {
- final PutResult read = putListener.read();
- if (read != null) {
- try (final ArrowBuf metadata = read.getApplicationMetadata()) {
- final FlightSql.DoPutPreparedStatementResult doPutPreparedStatementResult =
- FlightSql.DoPutPreparedStatementResult.parseFrom(metadata.nioBuffer());
- descriptor = FlightDescriptor
- .command(Any.pack(CommandPreparedStatementQuery.newBuilder()
- .setPreparedStatementHandle(
- doPutPreparedStatementResult.getPreparedStatementHandle())
- .build())
- .toByteArray());
+ if (getParameterSchema().getFields().size() > 0 &&
+ parameterBindingRoot != null &&
+ parameterBindingRoot.getRowCount() > 0) {
+ final PutResult read = putListener.read();
+ if (read != null) {
+ try (final ArrowBuf metadata = read.getApplicationMetadata()) {
+ final FlightSql.DoPutPreparedStatementResult doPutPreparedStatementResult =
+ FlightSql.DoPutPreparedStatementResult.parseFrom(metadata.nioBuffer());
+ descriptor = FlightDescriptor
+ .command(Any.pack(CommandPreparedStatementQuery.newBuilder()
+ .setPreparedStatementHandle(
+ doPutPreparedStatementResult.getPreparedStatementHandle())
+ .build())
+ .toByteArray());
+ }
  }
  }
  } catch (final InterruptedException | ExecutionException e) {