Fix for query closing on iterative algorithms.

madgik · Jan 18, 2021 · 27b72fc · 27b72fc
1 parent 9fd60d5
commit 27b72fc
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/...java/madgik/exareme/master/engine/iterations/handler/NIterativeAlgorithmResultEntity.java b/...java/madgik/exareme/master/engine/iterations/handler/NIterativeAlgorithmResultEntity.java
@@ -99,11 +99,15 @@ public void produceContent(ContentEncoder encoder, IOControl ioctrl) throws IOEx
                     this.buffer.compact();
                     if (i < 1 && !buffering) {
                         encoder.complete();
+                        closeQuery();
+                        close();
                     }
                 } else {
                     encoder.write(ByteBuffer.wrap(
                             finalizeQueryStatus.getError().getBytes()));
                     encoder.complete();
+                    closeQuery();
+                    close();
                 }
             } else {
                 // Algorithm execution failed, notify the client.
@@ -164,13 +168,12 @@ public void produceContent(ContentEncoder encoder, IOControl ioctrl) throws IOEx
                 encoder.write(buffer);
                 this.buffer.compact();
                 encoder.complete();
+                closeQuery();
+                close();
             }
         } finally {
             if (iterativeAlgorithmState != null)
                 iterativeAlgorithmState.releaseLock();
-
-            closeQuery();
-            close();
         }
     }