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
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.controller.versions;
import com.google.common.collect.ImmutableList;
import com.yahoo.collections.ListMap;
import com.yahoo.component.Version;
import com.yahoo.component.Vtag;
import com.yahoo.vespa.hosted.controller.api.integration.github.GitSha;
import com.yahoo.vespa.hosted.controller.Application;
import com.yahoo.vespa.hosted.controller.Controller;
import com.yahoo.vespa.hosted.controller.application.ApplicationList;
import com.yahoo.vespa.hosted.controller.application.Deployment;
import com.yahoo.vespa.hosted.controller.application.DeploymentJobs;
import com.yahoo.vespa.hosted.controller.application.JobStatus;
import java.net.URI;
import java.time.Instant;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.stream.Collectors;
/**
* Information about the current platform versions in use.
* The versions in use are the set of all versions running in current applications, versions
* of config servers in all zones, and the version of this controller itself.
*
* This is immutable.
*
* @author bratseth
*/
public class VersionStatus {
private static final Logger log = Logger.getLogger(VersionStatus.class.getName());
private static final String VESPA_REPO = "vespa-yahoo";
private static final String VESPA_REPO_OWNER = "vespa";
private final ImmutableList<VespaVersion> versions;
/** Create a version status. DO NOT USE: Public for testing only */
public VersionStatus(List<VespaVersion> versions) {
this.versions = ImmutableList.copyOf(versions);
}
/**
* Returns the current Vespa version of the system controlled by this,
* or empty if we have not currently determined what the system version is in this status.
*/
public Optional<VespaVersion> systemVersion() {
return versions().stream().filter(VespaVersion::isCurrentSystemVersion).findAny();
}
/**
* Lists all currently active Vespa versions, with deployment statistics,
* sorted from lowest to highest version number.
* The returned list is immutable.
* Calling this is free, but the returned status is slightly out of date.
*/
public List<VespaVersion> versions() { return versions; }
/** Returns the given version, or null if it is not present */
public VespaVersion version(Version version) {
return versions.stream().filter(v -> v.versionNumber().equals(version)).findFirst().orElse(null);
}
/** Create the empty version status */
public static VersionStatus empty() { return new VersionStatus(ImmutableList.of()); }
/** Create a full, updated version status. This is expensive and should be done infrequently */
public static VersionStatus compute(Controller controller) {
return compute(controller, Vtag.currentVersion);
}
/** Compute version status using the given current version. This is useful for testing. */
public static VersionStatus compute(Controller controller, Version currentVersion) {
ListMap<Version, String> configServerVersions = findConfigServerVersions(controller);
Set<Version> infrastructureVersions = new HashSet<>();
infrastructureVersions.add(currentVersion);
infrastructureVersions.addAll(configServerVersions.keySet());
// The system version is the oldest infrastructure version
Version systemVersion = infrastructureVersions.stream().sorted().findFirst().get();
Collection<DeploymentStatistics> deploymentStatistics = computeDeploymentStatistics(infrastructureVersions,
controller.applications().asList());
List<VespaVersion> versions = new ArrayList<>();
for (DeploymentStatistics statistics : deploymentStatistics) {
if (statistics.version().isEmpty()) continue;
try {
VespaVersion vespaVersion = createVersion(statistics,
statistics.version().equals(systemVersion),
configServerVersions.getList(statistics.version()),
controller);
versions.add(vespaVersion);
} catch (IllegalArgumentException e) {
log.log(Level.WARNING, "Unable to create VespaVersion for version " +
statistics.version().toFullString(), e);
}
}
Collections.sort(versions);
return new VersionStatus(versions);
}
private static ListMap<Version, String> findConfigServerVersions(Controller controller) {
List<URI> configServers = controller.zoneRegistry().zones().stream()
.flatMap(zone -> controller.getConfigServerUris(zone.environment(), zone.region()).stream())
.collect(Collectors.toList());
ListMap<Version, String> versions = new ListMap<>();
for (URI configServer : configServers)
versions.put(controller.applications().configserverClient().version(configServer), configServer.getHost());
return versions;
}
private static Collection<DeploymentStatistics> computeDeploymentStatistics(Set<Version> infrastructureVersions,
List<Application> applications) {
Map<Version, DeploymentStatistics> versionMap = new HashMap<>();
for (Version infrastructureVersion : infrastructureVersions) {
versionMap.put(infrastructureVersion, DeploymentStatistics.empty(infrastructureVersion));
}
for (Application application : ApplicationList.from(applications).notPullRequest().asList()) {
DeploymentJobs jobs = application.deploymentJobs();
// Note that each version deployed on this application exists
for (Deployment deployment : application.deployments().values()) {
versionMap.computeIfAbsent(deployment.version(), DeploymentStatistics::empty);
}
// List versions which have failing jobs, and versions which are in production
// Failing versions
Map<Version, List<JobStatus>> failingJobsByVersion = jobs.jobStatus().values().stream()
.filter(jobStatus -> jobStatus.lastCompleted().isPresent())
.filter(jobStatus -> jobStatus.lastCompleted().get().upgrade())
.filter(jobStatus -> jobStatus.jobError().isPresent())
.filter(jobStatus -> jobStatus.jobError().get() != DeploymentJobs.JobError.outOfCapacity)
.collect(Collectors.groupingBy(jobStatus -> jobStatus.lastCompleted().get().version()));
for (Version v : failingJobsByVersion.keySet()) {
versionMap.compute(v, (version, statistics) -> emptyIfMissing(version, statistics).withFailing(application.id()));
}
// Succeeding versions
Map<Version, List<JobStatus>> succeedingJobsByVersions = jobs.jobStatus().values().stream()
.filter(jobStatus -> jobStatus.lastSuccess().isPresent())
.filter(jobStatus -> jobStatus.type().isProduction())
.collect(Collectors.groupingBy(jobStatus -> jobStatus.lastSuccess().get().version()));
for (Version v : succeedingJobsByVersions.keySet()) {
versionMap.compute(v, (version, statistics) -> emptyIfMissing(version, statistics).withProduction(application.id()));
}
}
return versionMap.values();
}
private static DeploymentStatistics emptyIfMissing(Version version, DeploymentStatistics statistics) {
return statistics == null ? DeploymentStatistics.empty(version) : statistics;
}
private static VespaVersion createVersion(DeploymentStatistics statistics,
boolean isSystemVersion,
Collection<String> configServerHostnames,
Controller controller) {
GitSha gitSha = controller.gitHub().getCommit(VESPA_REPO_OWNER, VESPA_REPO, statistics.version().toFullString());
return new VespaVersion(statistics,
gitSha.sha, Instant.ofEpochMilli(gitSha.commit.author.date.getTime()),
isSystemVersion,
configServerHostnames,
controller);
}
}
|
