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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
// 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.config.ConfigInstance;
import com.yahoo.document.Field;
import com.yahoo.io.IOUtils;
import com.yahoo.schema.Index;
import com.yahoo.schema.Schema;
import com.yahoo.schema.document.ImmutableSDField;
import com.yahoo.schema.document.SDDocumentType;
import com.yahoo.schema.document.SDField;
import com.yahoo.text.StringUtilities;
import java.io.IOException;
import java.io.Writer;
import java.util.List;
/**
* Superclass of all derived configurations
*
* @author bratseth
*/
public abstract class Derived {
private String name;
public Derived() {
this("");
}
public Derived(String name) {
this.name = name;
}
public String getName() { return name; }
protected final void setName(String name) { this.name = name; }
/**
* Derives the content of this configuration. This
* default calls derive(Document) for each document
* and derive(SDField) for each search definition level field
* AND sets the name of this to the name of the input search definition
*/
protected void derive(Schema schema) {
setName(schema.getName());
derive(schema.getDocument(), schema);
for (Index index : schema.getExplicitIndices())
derive(index, schema);
for (SDField field : schema.allExtraFields())
derive(field, schema);
schema.allImportedFields().forEach(importedField -> derive(importedField, schema));
}
/**
* Derives the content of this configuration. This
* default calls derive(SDField) for each document field
*/
protected void derive(SDDocumentType document, Schema schema) {
for (Field field : document.fieldSet()) {
SDField sdField = (SDField) field;
if ( ! sdField.isExtraField()) {
derive(sdField, schema);
}
}
}
/**
* Derives the content of this configuration. This
* default does nothing.
*/
protected void derive(ImmutableSDField field, Schema schema) {}
/**
* Derives the content of this configuration. This
* default does nothing.
*/
protected void derive(Index index, Schema schema) {
}
protected abstract String getDerivedName();
/** Returns the value of getName if true, the given number as a string otherwise */
protected String getIndex(int number, boolean labels) {
return labels ? getName() : String.valueOf(number);
}
protected void export(String toDirectory, ConfigInstance cfg) throws IOException {
Writer writer = null;
try {
String fileName = getDerivedName() + ".cfg";
if (toDirectory != null) {
writer = IOUtils.createWriter(toDirectory + "/" + fileName, false);
exportConfig(writer, cfg);
}
}
finally {
if (writer != null) IOUtils.closeWriter(writer);
}
}
private void exportConfig(Writer writer, ConfigInstance cfg) throws IOException {
List<String> payloadL = ConfigInstance.serialize(cfg);
String payload = StringUtilities.implodeMultiline(payloadL);
writer.write(payload);
}
}
|
