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
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.config.server;
import com.google.common.io.Files;
import com.yahoo.cloud.config.ConfigserverConfig;
import com.yahoo.config.model.NullConfigModelRegistry;
import com.yahoo.config.model.api.ConfigDefinitionRepo;
import com.yahoo.config.provision.Provisioner;
import com.yahoo.config.provision.Zone;
import com.yahoo.vespa.config.server.application.PermanentApplicationPackage;
import com.yahoo.vespa.config.server.host.HostRegistries;
import com.yahoo.vespa.config.server.modelfactory.ModelFactoryRegistry;
import com.yahoo.vespa.config.server.monitoring.Metrics;
import com.yahoo.vespa.config.server.provision.HostProvisionerProvider;
import com.yahoo.vespa.config.server.session.FileDistributionFactory;
import com.yahoo.vespa.config.server.session.MockFileDistributionFactory;
import com.yahoo.vespa.config.server.session.SessionPreparer;
import com.yahoo.vespa.config.server.tenant.MockTenantListener;
import com.yahoo.vespa.config.server.tenant.TenantListener;
import com.yahoo.vespa.config.server.tenant.TenantRequestHandlerTest;
import com.yahoo.vespa.curator.Curator;
import com.yahoo.vespa.curator.mock.MockCurator;
import com.yahoo.vespa.config.server.zookeeper.ConfigCurator;
import com.yahoo.vespa.model.VespaModelFactory;
import java.time.Clock;
import java.util.Collections;
import java.util.Optional;
import static com.yahoo.vespa.config.server.SuperModelRequestHandlerTest.emptyNodeFlavors;
/**
* @author lulf
* @since 5.1
*/
public class TestComponentRegistry implements GlobalComponentRegistry {
private final Curator curator;
private final ConfigCurator configCurator;
private final Metrics metrics;
private final ConfigServerDB serverDB;
private final SessionPreparer sessionPreparer;
private final ConfigserverConfig configserverConfig;
private final SuperModelGenerationCounter superModelGenerationCounter;
private final ConfigDefinitionRepo defRepo;
private final ReloadListener reloadListener;
private final TenantListener tenantListener;
private final PermanentApplicationPackage permanentApplicationPackage;
private final HostRegistries hostRegistries;
private final FileDistributionFactory fileDistributionFactory;
private final ModelFactoryRegistry modelFactoryRegistry;
private final Optional<Provisioner> hostProvisioner;
private final Clock clock;
private TestComponentRegistry(Curator curator, ConfigCurator configCurator, Metrics metrics,
ModelFactoryRegistry modelFactoryRegistry,
PermanentApplicationPackage permanentApplicationPackage,
FileDistributionFactory fileDistributionFactory,
SuperModelGenerationCounter superModelGenerationCounter,
ConfigServerDB configServerDB,
HostRegistries hostRegistries,
ConfigserverConfig configserverConfig,
SessionPreparer sessionPreparer,
Optional<Provisioner> hostProvisioner,
ConfigDefinitionRepo defRepo,
ReloadListener reloadListener,
TenantListener tenantListener,
Clock clock) {
this.curator = curator;
this.configCurator = configCurator;
this.metrics = metrics;
this.configserverConfig = configserverConfig;
this.serverDB = configServerDB;
this.reloadListener = reloadListener;
this.tenantListener = tenantListener;
this.superModelGenerationCounter = superModelGenerationCounter;
this.defRepo = defRepo;
this.permanentApplicationPackage = permanentApplicationPackage;
this.hostRegistries = hostRegistries;
this.fileDistributionFactory = fileDistributionFactory;
this.modelFactoryRegistry = modelFactoryRegistry;
this.hostProvisioner = hostProvisioner;
this.sessionPreparer = sessionPreparer;
this.clock = clock;
}
public static class Builder {
private Curator curator = new MockCurator();
private Optional<ConfigCurator> configCurator = Optional.empty();
private Metrics metrics = Metrics.createTestMetrics();
private ConfigserverConfig configserverConfig = new ConfigserverConfig(new ConfigserverConfig.Builder()
.configServerDBDir(Files.createTempDir()
.getAbsolutePath()));
private ConfigDefinitionRepo defRepo = new StaticConfigDefinitionRepo();
private TenantRequestHandlerTest.MockReloadListener reloadListener = new TenantRequestHandlerTest.MockReloadListener();
private MockTenantListener tenantListener = new MockTenantListener();
private Optional<PermanentApplicationPackage> permanentApplicationPackage = Optional.empty();
private HostRegistries hostRegistries = new HostRegistries();
private Optional<FileDistributionFactory> fileDistributionFactory = Optional.empty();
private ModelFactoryRegistry modelFactoryRegistry = new ModelFactoryRegistry(Collections.singletonList(new VespaModelFactory(new NullConfigModelRegistry())));
private Optional<Provisioner> hostProvisioner = Optional.empty();
private Clock clock = Clock.systemUTC();
public Builder configServerConfig(ConfigserverConfig configserverConfig) {
this.configserverConfig = configserverConfig;
return this;
}
public Builder curator(Curator curator) {
this.curator = curator;
return this;
}
public Builder configCurator(ConfigCurator configCurator) {
this.configCurator = Optional.ofNullable(configCurator);
return this;
}
public Builder metrics(Metrics metrics) {
this.metrics = metrics;
return this;
}
public Builder modelFactoryRegistry(ModelFactoryRegistry modelFactoryRegistry) {
this.modelFactoryRegistry = modelFactoryRegistry;
return this;
}
public Builder permanentApplicationPackage(PermanentApplicationPackage permanentApplicationPackage) {
this.permanentApplicationPackage = Optional.ofNullable(permanentApplicationPackage);
return this;
}
public Builder provisioner(Provisioner provisioner) {
this.hostProvisioner = Optional.ofNullable(provisioner);
return this;
}
public Builder clock(Clock clock) {
this.clock = clock;
return this;
}
public TestComponentRegistry build() {
final PermanentApplicationPackage permApp = this.permanentApplicationPackage
.orElse(new PermanentApplicationPackage(configserverConfig));
FileDistributionFactory fileDistributionFactory = this.fileDistributionFactory
.orElse(new MockFileDistributionFactory(curator));
HostProvisionerProvider hostProvisionerProvider = hostProvisioner.isPresent() ?
HostProvisionerProvider.withProvisioner(hostProvisioner.get()) :
HostProvisionerProvider.empty();
SessionPreparer sessionPreparer = new SessionPreparer(modelFactoryRegistry, fileDistributionFactory,
hostProvisionerProvider, permApp,
configserverConfig, defRepo, curator,
new Zone(configserverConfig, emptyNodeFlavors()));
return new TestComponentRegistry(curator, configCurator.orElse(ConfigCurator.create(curator)),
metrics, modelFactoryRegistry,
permApp,
fileDistributionFactory,
new SuperModelGenerationCounter(curator),
new ConfigServerDB(configserverConfig),
hostRegistries, configserverConfig, sessionPreparer,
hostProvisioner, defRepo, reloadListener,
tenantListener, clock);
}
}
@Override
public Curator getCurator() { return curator; }
@Override
public ConfigCurator getConfigCurator() { return configCurator; }
@Override
public Metrics getMetrics() { return metrics; }
@Override
public ConfigServerDB getServerDB() { return serverDB; }
@Override
public SessionPreparer getSessionPreparer() { return sessionPreparer; }
@Override
public ConfigserverConfig getConfigserverConfig() { return configserverConfig; }
@Override
public TenantListener getTenantListener() { return tenantListener; }
@Override
public ReloadListener getReloadListener() { return reloadListener; }
@Override
public SuperModelGenerationCounter getSuperModelGenerationCounter() { return superModelGenerationCounter; }
@Override
public ConfigDefinitionRepo getConfigDefinitionRepo() { return defRepo; }
@Override
public PermanentApplicationPackage getPermanentApplicationPackage() { return permanentApplicationPackage; }
@Override
public HostRegistries getHostRegistries() { return hostRegistries;}
@Override
public ModelFactoryRegistry getModelFactoryRegistry() { return modelFactoryRegistry; }
@Override
public Optional<Provisioner> getHostProvisioner() {
return hostProvisioner;
}
@Override
public Zone getZone() {
return Zone.defaultZone();
}
public FileDistributionFactory getFileDistributionFactory() { return fileDistributionFactory; }
@Override
public Clock getClock() { return clock;}
}
|