client/src/main/java/ai/vespa/client/dsl/GroupOperation.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
29
30
31
32
33
34

// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package ai.vespa.client.dsl;

import java.util.Objects;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class GroupOperation implements IGroupOperation {

    private final String type;
    private Object value;
    private Aggregator[] aggregators;

    public GroupOperation(String type, Object value) {
        this.type = type;
        this.value = value;
    }

    public GroupOperation(String type, Aggregator[] aggregators) {
        this.type = type;
        this.aggregators = aggregators;
    }

    @Override
    public String toString() {
        if (value != null) {
            return Text.format("%s(%s)", type, value);
        }

        return Text.format("%s(%s)",
                             type,
                             Stream.of(aggregators).map(Objects::toString).collect(Collectors.joining(" ")));
    }
}