Note value source when visiting

1 files changed, 9 insertions, 3 deletions

diff --git a/container-search/src/main/java/com/yahoo/search/query/profile/PrefixQueryProfileVisitor.java b/container-search/src/main/java/com/yahoo/search/query/profile/PrefixQueryProfileVisitor.java
index e47b4bc2bb4..690a48f8124 100644
--- a/container-search/src/main/java/com/yahoo/search/query/profile/PrefixQueryProfileVisitor.java
+++ b/container-search/src/main/java/com/yahoo/search/query/profile/PrefixQueryProfileVisitor.java
@@ -25,12 +25,18 @@ abstract class PrefixQueryProfileVisitor extends QueryProfileVisitor {
     }
 
     @Override
-    public final void onQueryProfile(QueryProfile profile, DimensionBinding binding, QueryProfile owner) {
+    public final void onQueryProfile(QueryProfile profile,
+                                     DimensionBinding binding,
+                                     QueryProfile owner,
+                                     DimensionValues variant) {
         if (prefixComponentIndex < prefix.size()) return; // Not in the prefix yet
-        onQueryProfileInsidePrefix(profile, binding, owner);
+        onQueryProfileInsidePrefix(profile, binding, owner, variant);
     }
 
-    protected abstract void onQueryProfileInsidePrefix(QueryProfile profile, DimensionBinding binding, QueryProfile owner);
+    protected abstract void onQueryProfileInsidePrefix(QueryProfile profile,
+                                                       DimensionBinding binding,
+                                                       QueryProfile owner,
+                                                       DimensionValues variant);
 
     @Override
     public final boolean enter(String name) {