config-model/src/main/java/com/yahoo/schema/expressiontransforms/InputRecorderContext.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.schema.expressiontransforms;

import com.yahoo.schema.RankProfile;
import com.yahoo.searchlib.rankingexpression.transform.TransformContext;

import java.util.HashSet;
import java.util.Set;

class InputRecorderContext extends TransformContext {

    private final RankProfileTransformContext parent;
    private final Set<String> localVariables = new HashSet<>();

    public RankProfile rankProfile() { return parent.rankProfile(); }
    public Set<String> localVariables() { return localVariables; }

    public InputRecorderContext(RankProfileTransformContext parent) {
        super(parent.constants(), parent.types());
        this.parent = parent;
    }

    public InputRecorderContext(InputRecorderContext parent) {
        super(parent.constants(), parent.types());
        this.parent = parent.parent;
        this.localVariables.addAll(parent.localVariables);
    }
}