Update server/src/main/java/org/elasticsearch/index/query/InnerHitCon…

…textBuilder.java Co-authored-by: Mike Pellegrini <mike.pellegrini@elastic.co>
kderusso · Dec 12, 2024 · 1fc6125 · 1fc6125
1 parent 278e860
commit 1fc6125
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/server/src/main/java/org/elasticsearch/index/query/InnerHitContextBuilder.java b/server/src/main/java/org/elasticsearch/index/query/InnerHitContextBuilder.java
@@ -66,9 +66,9 @@ public InnerHitBuilder innerHitBuilder() {
     public static void extractInnerHits(QueryBuilder query, Map<String, InnerHitContextBuilder> innerHitBuilders) {
         if (query instanceof AbstractQueryBuilder) {
             ((AbstractQueryBuilder<?>) query).extractInnerHitBuilders(innerHitBuilders);
-        } else if (query instanceof InterceptedQueryBuilderWrapper) {
+        } else if (query instanceof InterceptedQueryBuilderWrapper interceptedQuery) {
             // Unwrap an intercepted query here
-            extractInnerHits(((InterceptedQueryBuilderWrapper) query).queryBuilder, innerHitBuilders);
+            extractInnerHits(interceptedQuery.queryBuilder, innerHitBuilders);
         } else {
             throw new IllegalStateException(
                 "provided query builder [" + query.getClass() + "] class should inherit from AbstractQueryBuilder, but it doesn't"