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

…ents PR update
stevelorddremio · Jun 3, 2024 · 23c79b2 · 23c79b2
1 parent 417230c
commit 23c79b2
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) {