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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.config.model;
import com.yahoo.config.application.api.ApplicationPackage;
import com.yahoo.config.application.api.DeployLogger;
import com.yahoo.config.model.api.ModelContext;
import com.yahoo.config.model.deploy.DeployState;
import com.yahoo.config.model.producer.AbstractConfigProducer;
import com.yahoo.vespa.model.VespaModel;
import java.util.stream.Stream;
/**
* This class contains a context that is passed to a model builder, and can be used to retrieve the application package,
* logger etc.
*
* @author Ulf Lilleengen
*/
public final class ConfigModelContext {
private final AbstractConfigProducer<?> parent;
private final String producerId;
private final DeployState deployState;
private final VespaModel vespaModel;
private final ConfigModelRepoAdder configModelRepoAdder;
private final ApplicationType applicationType;
private ConfigModelContext(ApplicationType applicationType,
DeployState deployState,
VespaModel vespaModel,
ConfigModelRepoAdder configModelRepoAdder,
AbstractConfigProducer<?> parent,
String producerId) {
this.applicationType = applicationType;
this.deployState = deployState;
this.vespaModel = vespaModel;
this.configModelRepoAdder = configModelRepoAdder;
this.parent = parent;
this.producerId = producerId;
}
public ApplicationPackage getApplicationPackage() { return deployState.getApplicationPackage(); }
public String getProducerId() { return producerId; }
public AbstractConfigProducer<?> getParentProducer() { return parent; }
public DeployLogger getDeployLogger() { return deployState.getDeployLogger(); }
public DeployState getDeployState() { return deployState; }
public ApplicationType getApplicationType() { return applicationType; }
public VespaModel vespaModel() { return vespaModel; }
public ModelContext.Properties properties() { return deployState.getProperties(); }
public ModelContext.FeatureFlags featureFlags() { return deployState.featureFlags(); }
/** Returns write access to the config model repo, or null (only) if this is improperly initialized during testing */
public ConfigModelRepoAdder getConfigModelRepoAdder() { return configModelRepoAdder; }
/** Create a new context with a different parent */
public ConfigModelContext withParent(AbstractConfigProducer<?> newParent) {
return ConfigModelContext.create(deployState, vespaModel, configModelRepoAdder, newParent, producerId);
}
/** Create a new context with a different config model producer id */
public ConfigModelContext withId(String producerId) {
return ConfigModelContext.create(deployState, vespaModel, configModelRepoAdder, parent, producerId);
}
public ConfigModelContext with(VespaModel vespaModel) {
return ConfigModelContext.create(deployState, vespaModel, configModelRepoAdder, parent, producerId);
}
/**
* Create an application context from a parent producer and an id.
*
* @param deployState the global deploy state for this model
* @param parent the parent to be used for the config model
* @param producerId the id to be used for the config model
* @return a model context that can be passed to a model
*/
public static ConfigModelContext create(DeployState deployState,
VespaModel vespaModel,
ConfigModelRepoAdder configModelRepoAdder,
AbstractConfigProducer<?> parent,
String producerId) {
return new ConfigModelContext(ApplicationType.DEFAULT, deployState, vespaModel, configModelRepoAdder, parent, producerId);
}
/**
* Create an application context from an application type, a parent producer and an id.
*
* @param applicationType the application type
* @param deployState the global deploy state for this model
* @param parent the parent to be used for the config model
* @param producerId the id to be used for the config model
* @return a model context that can be passed to a model
*/
public static ConfigModelContext create(ApplicationType applicationType,
DeployState deployState,
VespaModel vespaModel,
ConfigModelRepoAdder configModelRepoAdder,
AbstractConfigProducer<?> parent,
String producerId) {
return new ConfigModelContext(applicationType, deployState, vespaModel, configModelRepoAdder, parent, producerId);
}
public enum ApplicationType {
DEFAULT("default"),
HOSTED_INFRASTRUCTURE("hosted-infrastructure");
private final String type;
ApplicationType(String type) {
this.type = type;
}
public static ApplicationType fromString(String value) {
return Stream.of(ApplicationType.values())
.filter(applicationType -> applicationType.type.equals(value))
.findFirst()
.orElse(DEFAULT);
}
}
}
|