update TransportSearchAction to mitigate test failure

Author: drempapis

muted-tests.yml

@@ -212,9 +212,6 @@ tests:
 - class: org.elasticsearch.action.RejectionActionIT
   method: testSimulatedSearchRejectionLoad
   issue: https://github.com/elastic/elasticsearch/issues/125901
-- class: org.elasticsearch.search.basic.SearchWithRandomDisconnectsIT
-  method: testSearchWithRandomDisconnects
-  issue: https://github.com/elastic/elasticsearch/issues/122707
 - class: org.elasticsearch.xpack.test.rest.XPackRestIT
   method: test {p0=transform/transforms_reset/Test force reseting a running transform}
   issue: https://github.com/elastic/elasticsearch/issues/126240

server/src/main/java/org/elasticsearch/action/search/TransportSearchAction.java

@@ -1618,6 +1618,7 @@ public void runNewSearchPhase(
                 shardIterators.size(),
                 exc -> searchTransportService.cancelSearchTask(task, "failed to merge result [" + exc.getMessage() + "]")
             );
+            final ActionListener<SearchResponse> releasingListener = ActionListener.runAfter(listener, queryResultConsumer::close);
             boolean success = false;
             try {
                 final AbstractSearchAsyncAction<?> searchPhase;
@@ -1632,7 +1633,7 @@ public void runNewSearchPhase(
                         executor,
                         queryResultConsumer,
                         searchRequest,
-                        listener,
+                        releasingListener,
                         shardIterators,
                         timeProvider,
                         clusterState,
@@ -1652,7 +1653,7 @@ public void runNewSearchPhase(
                         executor,
                         queryResultConsumer,
                         searchRequest,
-                        listener,
+                        releasingListener,
                         shardIterators,
                         timeProvider,
                         clusterState,