aboutsummaryrefslogtreecommitdiffstats
path: root/config-model/src/main/java/com/yahoo/schema/derived/Juniperrc.java
blob: 58e77cedaf15aafe6b779468e9c74ce3a8c58bcb (plain) (blame)
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.schema.derived;

import com.yahoo.schema.Schema;
import com.yahoo.vespa.documentmodel.SummaryField;
import com.yahoo.vespa.documentmodel.SummaryTransform;
import com.yahoo.vespa.config.search.summary.JuniperrcConfig;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * Generated juniperrc-config for controlling juniper.
 *
 * @author Simon Thoresen Hult
 */
public class Juniperrc extends Derived {

    private static final int Mb = 1024 * 1024;

    /** List of all fields that should be bolded. */
    private final Set<String> boldingFields = new java.util.LinkedHashSet<>();

    /**
     * Constructs a new juniper rc instance for a given search object.
     * This will derive the configuration automatically,
     * so there is no need to call {@link #derive(Schema)}.
     *
     * @param schema the search model to use for deriving
     */
    public Juniperrc(Schema schema) {
        derive(schema);
    }

    @Override
    protected void derive(Schema schema) {
        super.derive(schema);
        for (SummaryField summaryField : schema.getUniqueNamedSummaryFields().values()) {
            if (summaryField.getTransform() == SummaryTransform.BOLDED) {
                boldingFields.add(summaryField.getName());
            }
        }
    }

    public void export(String toDirectory) throws IOException {
        var builder = new JuniperrcConfig.Builder();
        getConfig(builder);
        export(toDirectory, builder.build());
    }

    @Override protected String getDerivedName() { return "juniperrc"; }

    private static JuniperrcConfig.Override.Builder createOverride(String name) {
        return new JuniperrcConfig.Override.Builder()
                .fieldname(name)
                .length(64*Mb)
                .max_matches(1)
                .min_length(8192)
                .surround_max(64*Mb);
    }

    public void getConfig(JuniperrcConfig.Builder builder) {
        // Replace
        if (!boldingFields.isEmpty()) {
            builder.prefix(true);
            // Needs an modifiable list for config overrides ....
            builder.override(boldingFields.stream().map(Juniperrc::createOverride).collect(Collectors.toCollection(() -> new ArrayList<>())));
        }
    }

}