Skip to content

Commit

Permalink
Merge branch 'main' into kderusso/sparse-vector-yaml-errors
Browse files Browse the repository at this point in the history
  • Loading branch information
kderusso authored Dec 17, 2024
2 parents e3d37c4 + 6d943e9 commit b80424e
Show file tree
Hide file tree
Showing 5 changed files with 44 additions and 4 deletions.
5 changes: 5 additions & 0 deletions docs/changelog/118674.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
pr: 118674
summary: Ignore failures from renormalizing buckets in read-only index
area: Machine Learning
type: enhancement
issues: []
2 changes: 0 additions & 2 deletions muted-tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -302,8 +302,6 @@ tests:
- class: org.elasticsearch.xpack.security.QueryableReservedRolesIT
method: testDeletingAndCreatingSecurityIndexTriggersSynchronization
issue: https://github.com/elastic/elasticsearch/issues/118806
- class: org.elasticsearch.xpack.esql.session.IndexResolverFieldNamesTests
issue: https://github.com/elastic/elasticsearch/issues/118814
- class: org.elasticsearch.index.engine.RecoverySourcePruneMergePolicyTests
method: testPruneSome
issue: https://github.com/elastic/elasticsearch/issues/118728
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@

import org.elasticsearch.Build;
import org.elasticsearch.test.ESTestCase;
import org.elasticsearch.xpack.esql.action.EsqlCapabilities;
import org.elasticsearch.xpack.esql.parser.EsqlParser;
import org.elasticsearch.xpack.esql.parser.ParsingException;

Expand Down Expand Up @@ -1364,6 +1365,7 @@ public void testMetrics() {
}

public void testLookupJoin() {
assumeTrue("LOOKUP JOIN available as snapshot only", EsqlCapabilities.Cap.JOIN_LOOKUP_V7.isEnabled());
assertFieldNames(
"FROM employees | KEEP languages | RENAME languages AS language_code | LOOKUP JOIN languages_lookup ON language_code",
Set.of("languages", "languages.*", "language_code", "language_code.*"),
Expand All @@ -1372,6 +1374,7 @@ public void testLookupJoin() {
}

public void testLookupJoinKeep() {
assumeTrue("LOOKUP JOIN available as snapshot only", EsqlCapabilities.Cap.JOIN_LOOKUP_V7.isEnabled());
assertFieldNames(
"""
FROM employees
Expand All @@ -1385,6 +1388,7 @@ public void testLookupJoinKeep() {
}

public void testLookupJoinKeepWildcard() {
assumeTrue("LOOKUP JOIN available as snapshot only", EsqlCapabilities.Cap.JOIN_LOOKUP_V7.isEnabled());
assertFieldNames(
"""
FROM employees
Expand All @@ -1398,6 +1402,7 @@ public void testLookupJoinKeepWildcard() {
}

public void testMultiLookupJoin() {
assumeTrue("LOOKUP JOIN available as snapshot only", EsqlCapabilities.Cap.JOIN_LOOKUP_V7.isEnabled());
assertFieldNames(
"""
FROM sample_data
Expand All @@ -1410,6 +1415,7 @@ public void testMultiLookupJoin() {
}

public void testMultiLookupJoinKeepBefore() {
assumeTrue("LOOKUP JOIN available as snapshot only", EsqlCapabilities.Cap.JOIN_LOOKUP_V7.isEnabled());
assertFieldNames(
"""
FROM sample_data
Expand All @@ -1423,6 +1429,7 @@ public void testMultiLookupJoinKeepBefore() {
}

public void testMultiLookupJoinKeepBetween() {
assumeTrue("LOOKUP JOIN available as snapshot only", EsqlCapabilities.Cap.JOIN_LOOKUP_V7.isEnabled());
assertFieldNames(
"""
FROM sample_data
Expand All @@ -1447,6 +1454,7 @@ public void testMultiLookupJoinKeepBetween() {
}

public void testMultiLookupJoinKeepAfter() {
assumeTrue("LOOKUP JOIN available as snapshot only", EsqlCapabilities.Cap.JOIN_LOOKUP_V7.isEnabled());
assertFieldNames(
"""
FROM sample_data
Expand All @@ -1473,6 +1481,7 @@ public void testMultiLookupJoinKeepAfter() {
}

public void testMultiLookupJoinKeepAfterWildcard() {
assumeTrue("LOOKUP JOIN available as snapshot only", EsqlCapabilities.Cap.JOIN_LOOKUP_V7.isEnabled());
assertFieldNames(
"""
FROM sample_data
Expand All @@ -1486,6 +1495,7 @@ public void testMultiLookupJoinKeepAfterWildcard() {
}

public void testMultiLookupJoinSameIndex() {
assumeTrue("LOOKUP JOIN available as snapshot only", EsqlCapabilities.Cap.JOIN_LOOKUP_V7.isEnabled());
assertFieldNames(
"""
FROM sample_data
Expand All @@ -1499,6 +1509,7 @@ public void testMultiLookupJoinSameIndex() {
}

public void testMultiLookupJoinSameIndexKeepBefore() {
assumeTrue("LOOKUP JOIN available as snapshot only", EsqlCapabilities.Cap.JOIN_LOOKUP_V7.isEnabled());
assertFieldNames(
"""
FROM sample_data
Expand All @@ -1513,6 +1524,7 @@ public void testMultiLookupJoinSameIndexKeepBefore() {
}

public void testMultiLookupJoinSameIndexKeepBetween() {
assumeTrue("LOOKUP JOIN available as snapshot only", EsqlCapabilities.Cap.JOIN_LOOKUP_V7.isEnabled());
assertFieldNames(
"""
FROM sample_data
Expand All @@ -1538,6 +1550,7 @@ public void testMultiLookupJoinSameIndexKeepBetween() {
}

public void testMultiLookupJoinSameIndexKeepAfter() {
assumeTrue("LOOKUP JOIN available as snapshot only", EsqlCapabilities.Cap.JOIN_LOOKUP_V7.isEnabled());
assertFieldNames(
"""
FROM sample_data
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,10 +8,12 @@

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.elasticsearch.action.bulk.BulkItemResponse;
import org.elasticsearch.action.bulk.BulkRequest;
import org.elasticsearch.action.bulk.BulkResponse;
import org.elasticsearch.action.index.IndexRequest;
import org.elasticsearch.client.internal.Client;
import org.elasticsearch.cluster.metadata.IndexMetadata;
import org.elasticsearch.common.util.concurrent.ThreadContext;
import org.elasticsearch.xcontent.ToXContent;
import org.elasticsearch.xcontent.XContentBuilder;
Expand Down Expand Up @@ -102,7 +104,29 @@ public void executeRequest() {
try (ThreadContext.StoredContext ignore = client.threadPool().getThreadContext().stashWithOrigin(ML_ORIGIN)) {
BulkResponse addRecordsResponse = client.bulk(bulkRequest).actionGet();
if (addRecordsResponse.hasFailures()) {
logger.error("[{}] Bulk index of results has errors: {}", jobId, addRecordsResponse.buildFailureMessage());
// Implementation note: Ignore the failures from writing to the read-only index, as it comes
// from changing the index format version.
boolean hasNonReadOnlyFailures = false;
for (BulkItemResponse response : addRecordsResponse.getItems()) {
if (response.isFailed() == false) {
continue;
}
if (response.getFailureMessage().contains(IndexMetadata.INDEX_READ_ONLY_BLOCK.description())) {
// We expect this to happen when the old index is made read-only and being reindexed
logger.debug(
"[{}] Ignoring failure to write renormalized results to a read-only index [{}]: {}",
jobId,
response.getFailure().getIndex(),
response.getFailureMessage()
);
} else {
hasNonReadOnlyFailures = true;
break;
}
}
if (hasNonReadOnlyFailures) {
logger.error("[{}] Bulk index of results has errors: {}", jobId, addRecordsResponse.buildFailureMessage());
}
}
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3353,7 +3353,7 @@ Alejandro
Amabile
Anoosh
Basil
Bojan
Brendon
// end::filterToday
;

Expand Down

0 comments on commit b80424e

Please sign in to comment.