add parsing (only) of container-phase in schema

1 files changed, 15 insertions, 1 deletions

diff --git a/config-model/src/main/java/com/yahoo/schema/parser/ParsedRankProfile.java b/config-model/src/main/java/com/yahoo/schema/parser/ParsedRankProfile.java
index 64dd8dd0ad4..2de7fab09f4 100644
--- a/config-model/src/main/java/com/yahoo/schema/parser/ParsedRankProfile.java
+++ b/config-model/src/main/java/com/yahoo/schema/parser/ParsedRankProfile.java
@@ -53,6 +53,8 @@ class ParsedRankProfile extends ParsedBlock {
     private final Map<Reference, RankProfile.Constant> constants = new LinkedHashMap<>();
     private final Map<Reference, RankProfile.Input> inputs = new LinkedHashMap<>();
     private final List<OnnxModel> onnxModels = new ArrayList<>();
+    private Integer containerPhaseRerankCount = null;
+    private String containerPhaseExpression = null;
 
     ParsedRankProfile(String name) {
         super(name, "rank-profile");
@@ -77,6 +79,8 @@ class ParsedRankProfile extends ParsedBlock {
     List<ParsedRankFunction> getFunctions() { return List.copyOf(functions.values()); }
     List<MutateOperation> getMutateOperations() { return List.copyOf(mutateOperations); }
     List<String> getInherited() { return List.copyOf(inherited); }
+    Optional<Integer> getContainerPhaseRerankCount() { return Optional.ofNullable(this.containerPhaseRerankCount); }
+    Optional<String> getContainerPhaseExpression() { return Optional.ofNullable(this.containerPhaseExpression); }
 
     Map<String, Boolean> getFieldsWithRankFilter() { return Collections.unmodifiableMap(fieldsRankFilter); }
     Map<String, Integer> getFieldsWithRankWeight() { return Collections.unmodifiableMap(fieldsRankWeight); }
@@ -197,11 +201,21 @@ class ParsedRankProfile extends ParsedBlock {
         this.secondPhaseExpression = expression;
     }
 
+    void setContainerPhaseExpression(String expression) {
+        verifyThat(containerPhaseExpression == null, "already has container-phase expression");
+        this.containerPhaseExpression = expression;
+    }
+
+    void setContainerPhaseRerankCount(int count) {
+        verifyThat(containerPhaseRerankCount == null, "already has container-phase rerank-count");
+        this.containerPhaseRerankCount = count;
+    }
+
     void setStrict(boolean strict) {
         verifyThat(this.strict == null, "already has strict");
         this.strict = strict;
     }
-  
+
     void setTermwiseLimit(double limit) {
         verifyThat(termwiseLimit == null, "already has termwise-limit");
         this.termwiseLimit = limit;