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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.config.server;
import com.yahoo.component.annotation.Inject;
import com.yahoo.cloud.config.ConfigserverConfig;
import com.yahoo.config.model.api.ApplicationInfo;
import com.yahoo.config.model.api.SuperModel;
import com.yahoo.config.model.api.SuperModelListener;
import com.yahoo.config.model.api.SuperModelProvider;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.config.provision.Zone;
import com.yahoo.vespa.config.GenerationCounter;
import com.yahoo.vespa.config.server.application.ApplicationVersions;
import com.yahoo.vespa.config.server.model.SuperModelConfigProvider;
import com.yahoo.vespa.flags.FlagSource;
import java.time.Instant;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import java.util.Set;
import java.util.logging.Level;
import java.util.logging.Logger;
/**
* Provides a SuperModel - a model of all application instances, and makes it stays
* up to date as applications are added, redeployed, and removed.
*/
public class SuperModelManager implements SuperModelProvider {
private static final Logger logger = Logger.getLogger(SuperModelManager.class.getName());
private final Zone zone;
private final Object monitor = new Object();
private final FlagSource flagSource;
private SuperModelConfigProvider superModelConfigProvider; // Guarded by 'this' monitor
private final List<SuperModelListener> listeners = new ArrayList<>(); // Guarded by 'this' monitor
// Generation of the super model
private long generation;
private final long masterGeneration; // ConfigserverConfig's generation
private final GenerationCounter generationCounter;
// The initial set of applications to be deployed on bootstrap.
private final Optional<Set<ApplicationId>> bootstrapApplicationSet = Optional.empty();
@Inject
public SuperModelManager(ConfigserverConfig configserverConfig,
Zone zone,
GenerationCounter generationCounter,
FlagSource flagSource) {
this.flagSource = flagSource;
this.zone = zone;
this.generationCounter = generationCounter;
this.masterGeneration = configserverConfig.masterGeneration();
makeNewSuperModelConfigProvider(new SuperModel());
}
@Override
public SuperModel getSuperModel() {
synchronized (monitor) {
return superModelConfigProvider.getSuperModel();
}
}
public SuperModelConfigProvider getSuperModelConfigProvider() {
synchronized (monitor) {
return superModelConfigProvider;
}
}
public long getGeneration() {
synchronized (monitor) {
return generation;
}
}
@Override
public void registerListener(SuperModelListener listener) {
synchronized (monitor) {
listeners.add(listener);
SuperModel superModel = superModelConfigProvider.getSuperModel();
superModel.getAllApplicationInfos().forEach(application -> listener.applicationActivated(superModel, application));
if (superModel.isComplete()) {
listener.notifyOfCompleteness(superModel);
}
}
}
public void configActivated(ApplicationVersions applicationVersions) {
synchronized (monitor) {
// TODO: Should supermodel care about multiple versions?
ApplicationInfo applicationInfo = applicationVersions
.getForVersionOrLatest(Optional.empty(), Instant.now())
.toApplicationInfo();
SuperModel newSuperModel = superModelConfigProvider.getSuperModel()
.cloneAndSetApplication(applicationInfo);
generationCounter.increment();
makeNewSuperModelConfigProvider(newSuperModel);
listeners.forEach(listener -> listener.applicationActivated(newSuperModel, applicationInfo));
}
}
public void applicationRemoved(ApplicationId applicationId) {
synchronized (monitor) {
bootstrapApplicationSet.ifPresent(set -> set.remove(applicationId));
SuperModel newSuperModel = this.superModelConfigProvider
.getSuperModel()
.cloneAndRemoveApplication(applicationId);
generationCounter.increment();
makeNewSuperModelConfigProvider(newSuperModel);
listeners.forEach(listener -> listener.applicationRemoved(newSuperModel, applicationId));
}
}
public void markAsComplete() {
// Invoked on component graph bootstrap (even before ConfigServerBootstrap),
// there is no need to bump generation counter.
logger.log(Level.FINE, "Super model is complete");
SuperModel newSuperModel = getSuperModel().cloneAsComplete();
superModelConfigProvider = new SuperModelConfigProvider(newSuperModel, zone, flagSource);
listeners.forEach(listener -> listener.notifyOfCompleteness(newSuperModel));
}
private void makeNewSuperModelConfigProvider(SuperModel newSuperModel) {
generation = masterGeneration + generationCounter.get();
superModelConfigProvider = new SuperModelConfigProvider(newSuperModel, zone, flagSource);
}
}
|
