oops

dnhatn · dnhatn · commit 556543495bee · 2025-02-05T13:36:04.000-08:00
diff --git a/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/DataNodeRequestSender.java b/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/DataNodeRequestSender.java
@@ -149,7 +149,7 @@ void onAfter(List<DriverProfile> profiles) {
             public void onResponse(DataNodeComputeResponse response) {
                 // remove failures of successful shards
                 for (ShardId shardId : targetShards.shardIds()) {
-                    if (shardFailures.containsKey(shardId) == false) {
+                    if (response.shardLevelFailures().containsKey(shardId) == false) {
                         shardFailures.remove(shardId);
                     }
                 }

Original file line number	Diff line number	Diff line change
`@@ -149,7 +149,7 @@ void onAfter(List<DriverProfile> profiles) {`
`149`	`149`	`public void onResponse(DataNodeComputeResponse response) {`
`150`	`150`	`// remove failures of successful shards`
`151`	`151`	`for (ShardId shardId : targetShards.shardIds()) {`
`152`		`- if (shardFailures.containsKey(shardId) == false) {`
	`152`	`+ if (response.shardLevelFailures().containsKey(shardId) == false) {`
`153`	`153`	`shardFailures.remove(shardId);`
`154`	`154`	`}`
`155`	`155`	`}`