Optimize time-series source operator

Author: dnhatn

x-pack/plugin/esql/compute/src/main/java/org/elasticsearch/compute/data/DocVector.java

@@ -67,6 +67,12 @@ public DocVector(IntVector shards, IntVector segments, IntVector docs, Boolean s
         blockFactory().adjustBreaker(BASE_RAM_BYTES_USED);
     }
 
+    public DocVector(IntVector shards, IntVector segments, IntVector docs, int[] docMapForwards, int[] docMapBackwards) {
+        this(shards, segments, docs, null);
+        this.shardSegmentDocMapForwards = docMapForwards;
+        this.shardSegmentDocMapBackwards = docMapBackwards;
+    }
+
     public IntVector shards() {
         return shards;
     }
@@ -209,8 +215,12 @@ protected void swap(int i, int j) {
         }
     }
 
+    public static long sizeOfSegmentDocMap(int positionCount) {
+        return 2 * (((long) RamUsageEstimator.NUM_BYTES_ARRAY_HEADER) + ((long) Integer.BYTES) * positionCount);
+    }
+
     private long sizeOfSegmentDocMap() {
-        return 2 * (((long) RamUsageEstimator.NUM_BYTES_ARRAY_HEADER) + ((long) Integer.BYTES) * shards.getPositionCount());
+        return sizeOfSegmentDocMap(getPositionCount());
     }
 
     @Override