Fix double ref counting from previous PR

Author: GalLalouche

x-pack/plugin/esql/compute/src/main/java/org/elasticsearch/compute/lucene/LuceneOperator.java

@@ -86,7 +86,7 @@ protected LuceneOperator(
         LuceneSliceQueue sliceQueue
     ) {
         this.shardContextCounters = shardContextCounters;
-        shardContextCounters.forEach(RefCounted::incRef);
+        shardContextCounters.forEach(RefCounted::mustIncRef);
         this.blockFactory = blockFactory;
         this.maxPageSize = maxPageSize;
         this.sliceQueue = sliceQueue;

x-pack/plugin/esql/compute/src/main/java/org/elasticsearch/compute/lucene/LuceneSourceOperator.java

@@ -227,7 +227,6 @@ public LuceneSourceOperator(
         boolean needsScore
     ) {
         super(shardContextCounters, blockFactory, maxPageSize, sliceQueue);
-        shardContextCounters.forEach(RefCounted::mustIncRef);
         this.minPageSize = Math.max(1, maxPageSize / 2);
         this.remainingDocs = limit;
         this.limiter = limiter;

x-pack/plugin/esql/compute/src/main/java/org/elasticsearch/compute/lucene/LuceneTopNSourceOperator.java

@@ -241,10 +241,12 @@ private Page emit(boolean startEmitting) {
                 }
             }
 
-            shard = blockFactory.newConstantIntBlockWith(perShardCollector.shardContext.index(), size);
+            int shardId = perShardCollector.shardContext.index();
+            shard = blockFactory.newConstantIntBlockWith(shardId, size);
             segments = currentSegmentBuilder.build();
             docs = currentDocsBuilder.build();
-            docBlock = new DocVector(ShardRefCounted.fromList(contexts), shard.asVector(), segments, docs, null).asBlock();
+            ShardRefCounted shardRefCounted = ShardRefCounted.single(shardId, contexts.get(shardId));
+            docBlock = new DocVector(shardRefCounted, shard.asVector(), segments, docs, null).asBlock();
             shard = null;
             segments = null;
             docs = null;

x-pack/plugin/esql/compute/src/main/java/org/elasticsearch/compute/lucene/TimeSeriesSourceOperator.java

@@ -30,7 +30,6 @@
 import org.elasticsearch.compute.data.OrdinalBytesRefVector;
 import org.elasticsearch.compute.data.Page;
 import org.elasticsearch.compute.operator.Operator;
-import org.elasticsearch.core.RefCounted;
 import org.elasticsearch.core.Releasable;
 import org.elasticsearch.core.Releasables;
 import org.elasticsearch.xcontent.XContentBuilder;
@@ -64,7 +63,6 @@ public final class TimeSeriesSourceOperator extends LuceneOperator {
         int limit
     ) {
         super(contexts, blockFactory, maxPageSize, sliceQueue);
-        contexts.forEach(RefCounted::mustIncRef);
         this.maxPageSize = maxPageSize;
         this.blockFactory = blockFactory;
         this.remainingDocs = limit;