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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
// 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.config.ConfigInstance.Builder;
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.lang.reflect.Method;
import java.util.List;
/**
* Superclass of all derived configurations
*
* @author bratseth
*/
public abstract class Derived implements Exportable {
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);
}
/**
* Exports this derived configuration to its .cfg file
* in toDirectory
*
* @param toDirectory the directory to export to, or null
*
*/
public void export(String toDirectory) throws IOException {
Writer writer = null;
try {
String fileName = getDerivedName() + ".cfg";
if (toDirectory != null)
writer = IOUtils.createWriter(toDirectory + "/" + fileName,false);
try {
exportBuilderConfig(writer);
} catch (ReflectiveOperationException | SecurityException | IllegalArgumentException e) {
throw new RuntimeException(e);
}
}
finally {
if (writer != null) IOUtils.closeWriter(writer);
}
}
/**
* Checks what this is a producer of, instantiate that and export to writer
*/
// TODO move to ReflectionUtil, and move that to unexported pkg
private void exportBuilderConfig(Writer writer) throws ReflectiveOperationException, SecurityException, IllegalArgumentException, IOException {
for (Class<?> intf : getClass().getInterfaces()) {
if (ConfigInstance.Producer.class.isAssignableFrom(intf)) {
Class<?> configClass = intf.getEnclosingClass();
String builderClassName = configClass.getCanonicalName()+"$Builder";
Class<?> builderClass = Class.forName(builderClassName);
ConfigInstance.Builder builder = (Builder) builderClass.getDeclaredConstructor().newInstance();
Method getConfig = getClass().getMethod("getConfig", builderClass);
getConfig.invoke(this, builder);
ConfigInstance inst = (ConfigInstance) configClass.getConstructor(builderClass).newInstance(builder);
List<String> payloadL = ConfigInstance.serialize(inst);
String payload = StringUtilities.implodeMultiline(payloadL);
writer.write(payload);
}
}
}
@Override
public String getFileName() {
return getDerivedName() + ".cfg";
}
}
|
