Don't remove state until after onCreationResponse is successful

deephaven · Nov 8, 2023 · b674e05 · b674e05
1 parent 68b63b2
commit b674e05
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/java-client/session/src/main/java/io/deephaven/client/impl/ExportStates.java b/java-client/session/src/main/java/io/deephaven/client/impl/ExportStates.java
@@ -398,11 +398,12 @@ public void onNext(ExportedTableCreationResponse value) {
                         "Not expecting export creation responses for empty tickets");
             }
             final int exportId = ExportTicketHelper.ticketToExportId(value.getResultId().getTicket(), "export");
-            final State state = newStates.remove(exportId);
+            final State state = newStates.get(exportId);
             if (state == null) {
                 throw new IllegalStateException("Unable to find state for creation response");
             }
             state.onCreationResponse(value);
+            newStates.remove(exportId, state);
         }
 
         @Override