From fd9a49256436ae1d2a678f56cdbe9f5d9e6854bb Mon Sep 17 00:00:00 2001
From: Rafael Telles <rafael@telles.dev>
Date: Tue, 10 Aug 2021 15:15:55 -0300
Subject: [PATCH] Fix wrong PreparedStatement cache invalidation

---
 .../java/org/apache/arrow/flight/sql/FlightSqlExample.java  | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/java/flight/flight-sql/src/test/java/org/apache/arrow/flight/sql/FlightSqlExample.java b/java/flight/flight-sql/src/test/java/org/apache/arrow/flight/sql/FlightSqlExample.java
index fdf0ea00946ff..fbb60771a3182 100644
--- a/java/flight/flight-sql/src/test/java/org/apache/arrow/flight/sql/FlightSqlExample.java
+++ b/java/flight/flight-sql/src/test/java/org/apache/arrow/flight/sql/FlightSqlExample.java
@@ -609,8 +609,8 @@ private static VectorSchemaRoot getSqlInfoRoot(final DatabaseMetaData metaData,
   @Override
   public void getStreamPreparedStatement(final CommandPreparedStatementQuery command, final CallContext context,
                                          final Ticket ticket, final ServerStreamListener listener) {
-    StatementContext<PreparedStatement> statementContext =
-        preparedStatementLoadingCache.getIfPresent(command.getPreparedStatementHandle());
+    ByteString handle = command.getPreparedStatementHandle();
+    StatementContext<PreparedStatement> statementContext = preparedStatementLoadingCache.getIfPresent(handle);
     assert statementContext != null;
     try (PreparedStatement statement = statementContext.getStatement();
          ResultSet resultSet = statement.executeQuery()) {
@@ -635,7 +635,7 @@ public void getStreamPreparedStatement(final CommandPreparedStatementQuery comma
       listener.error(e);
     } finally {
       listener.completed();
-      commandExecutePreparedStatementLoadingCache.invalidate(command);
+      commandExecutePreparedStatementLoadingCache.invalidate(handle);
     }
   }