reporter to accept observer

pgomulka · Jul 3, 2024 · 825b63e · 825b63e
1 parent 05ceb83
commit 825b63e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/...c/internalClusterTest/java/org/elasticsearch/plugins/internal/DocumentSizeObserverIT.java b/...c/internalClusterTest/java/org/elasticsearch/plugins/internal/DocumentSizeObserverIT.java
@@ -108,7 +108,9 @@ public IndexResult index(Index index) throws IOException {
                         config().getMapperService(),
                         DocumentSizeAccumulator.EMPTY_INSTANCE
                     );
-                    documentParsingReporter.onIndexingCompleted(index.parsedDoc());
+                    ParsedDocument parsedDocument = index.parsedDoc();
+                    DocumentSizeObserver documentSizeObserver = parsedDocument.getDocumentSizeObserver();
+                    documentParsingReporter.onIndexingCompleted(documentSizeObserver);
 
                     return result;
                 }
@@ -149,8 +151,8 @@ public TestDocumentSizeReporter(String indexName) {
         }
 
         @Override
-        public void onIndexingCompleted(ParsedDocument parsedDocument) {
-            DocumentSizeObserver documentSizeObserver = parsedDocument.getDocumentSizeObserver();
+        public void onIndexingCompleted(DocumentSizeObserver documentSizeObserver) {
+            DocumentSizeObserver documentSizeObserver = documentSizeObserver.getDocumentSizeObserver();
             COUNTER.addAndGet(documentSizeObserver.normalisedBytesParsed());
             assertThat(indexName, equalTo(TEST_INDEX_NAME));
         }

diff --git a/server/src/main/java/org/elasticsearch/plugins/internal/DocumentSizeReporter.java b/server/src/main/java/org/elasticsearch/plugins/internal/DocumentSizeReporter.java
@@ -8,8 +8,6 @@
 
 package org.elasticsearch.plugins.internal;
 
-import org.elasticsearch.index.mapper.ParsedDocument;
-
 /**
  * An interface to allow performing an action when parsing and indexing has been completed
  */
@@ -23,10 +21,10 @@ public interface DocumentSizeReporter {
     /**
      * An action to be performed upon finished indexing.
      */
-    default void onParsingCompleted(ParsedDocument parsedDocument) {}
+    default void onParsingCompleted(DocumentSizeObserver documentSizeObserver) {}
 
     /**
      * An action to be performed upon finished indexing.
      */
-    default void onIndexingCompleted(ParsedDocument parsedDocument) {}
+    default void onIndexingCompleted(DocumentSizeObserver documentSizeObserver) {}
 }