[9.0] Avoid reading unnecessary dimension values when downsampling (#124451)

docs/changelog/124451.yaml

@@ -0,0 +1,5 @@
+pr: 124451
+summary: Improve downsample performance by avoiding to read unnecessary dimension values when downsampling.
+area: Downsampling
+type: bug
+issues: []

x-pack/plugin/downsample/src/main/java/org/elasticsearch/xpack/downsample/DimensionFieldProducer.java

@@ -44,16 +44,29 @@ void reset() {
             isEmpty = true;
         }
 
-        void collect(final Object value) {
+        void collectOnce(final Object value) {
+            assert isEmpty;
             Objects.requireNonNull(value);
-            if (isEmpty) {
-                this.value = value;
-                this.isEmpty = false;
-                return;
-            }
-            if (value.equals(this.value) == false) {
-                throw new IllegalArgumentException("Dimension value changed without tsid change [" + value + "] != [" + this.value + "]");
+            this.value = value;
+            this.isEmpty = false;
+        }
+
+        /**
+         * This is an expensive check, that slows down downsampling significantly.
+         * Given that index is sorted by tsid as primary key, this shouldn't really happen.
+         */
+        boolean validate(FormattedDocValues docValues, int docId) throws IOException {
+            if (docValues.advanceExact(docId)) {
+                int docValueCount = docValues.docValueCount();
+                for (int i = 0; i < docValueCount; i++) {
+                    var value = docValues.nextValue();
+                    if (value.equals(this.value) == false) {
+                        assert false : "Dimension value changed without tsid change [" + value + "] != [" + this.value + "]";
+                    }
+                }
             }
+
+            return true;
         }
     }
 
@@ -69,12 +82,17 @@ public boolean isEmpty() {
 
     @Override
     public void collect(FormattedDocValues docValues, int docId) throws IOException {
+        if (dimension.isEmpty == false) {
+            assert dimension.validate(docValues, docId);
+            return;
+        }
+
         if (docValues.advanceExact(docId) == false) {
             return;
         }
         int docValueCount = docValues.docValueCount();
         for (int i = 0; i < docValueCount; i++) {
-            this.dimension.collect(docValues.nextValue());
+            this.dimension.collectOnce(docValues.nextValue());
         }
     }