diff --git a/server/src/internalClusterTest/java/org/elasticsearch/plugins/internal/XContentMeteringParserDecoratorIT.java b/server/src/internalClusterTest/java/org/elasticsearch/plugins/internal/XContentMeteringParserDecoratorIT.java index f70667b91aec8..b59afbdc461f8 100644 --- a/server/src/internalClusterTest/java/org/elasticsearch/plugins/internal/XContentMeteringParserDecoratorIT.java +++ b/server/src/internalClusterTest/java/org/elasticsearch/plugins/internal/XContentMeteringParserDecoratorIT.java @@ -10,6 +10,7 @@ package org.elasticsearch.plugins.internal; import org.elasticsearch.action.index.IndexRequest; +import org.elasticsearch.index.Index; import org.elasticsearch.index.IndexSettings; import org.elasticsearch.index.engine.EngineFactory; import org.elasticsearch.index.engine.InternalEngine; @@ -104,7 +105,7 @@ public IndexResult index(Index index) throws IOException { IndexResult result = super.index(index); DocumentSizeReporter documentParsingReporter = documentParsingProvider.newDocumentSizeReporter( - shardId.getIndexName(), + shardId.getIndex(), config().getMapperService(), DocumentSizeAccumulator.EMPTY_INSTANCE ); @@ -131,11 +132,11 @@ public XContentMeteringParserDecorator newMeteringParserDecorator(IndexReque @Override public DocumentSizeReporter newDocumentSizeReporter( - String indexName, + Index index, MapperService mapperService, DocumentSizeAccumulator documentSizeAccumulator ) { - return new TestDocumentSizeReporter(indexName); + return new TestDocumentSizeReporter(index); } }; } @@ -143,10 +144,10 @@ public DocumentSizeReporter newDocumentSizeReporter( public static class TestDocumentSizeReporter implements DocumentSizeReporter { - private final String indexName; + private final Index index; - public TestDocumentSizeReporter(String indexName) { - this.indexName = indexName; + public TestDocumentSizeReporter(Index index) { + this.index = index; } @Override @@ -155,7 +156,7 @@ public void onIndexingCompleted(ParsedDocument parsedDocument) { if (delta > XContentMeteringParserDecorator.UNKNOWN_SIZE) { COUNTER.addAndGet(delta); } - assertThat(indexName, equalTo(TEST_INDEX_NAME)); + assertThat(index.getName(), equalTo(TEST_INDEX_NAME)); } } diff --git a/server/src/main/java/org/elasticsearch/plugins/internal/DocumentParsingProvider.java b/server/src/main/java/org/elasticsearch/plugins/internal/DocumentParsingProvider.java index 9df7fd4c3bd43..1334f13d94b37 100644 --- a/server/src/main/java/org/elasticsearch/plugins/internal/DocumentParsingProvider.java +++ b/server/src/main/java/org/elasticsearch/plugins/internal/DocumentParsingProvider.java @@ -10,6 +10,7 @@ package org.elasticsearch.plugins.internal; import org.elasticsearch.action.index.IndexRequest; +import org.elasticsearch.index.Index; import org.elasticsearch.index.mapper.MapperService; /** @@ -23,7 +24,7 @@ public interface DocumentParsingProvider { * @return an instance of a reporter to use when parsing has been completed and indexing successful */ default DocumentSizeReporter newDocumentSizeReporter( - String indexName, + Index index, MapperService mapperService, DocumentSizeAccumulator documentSizeAccumulator ) {