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
|
// 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.deployment;
import com.google.common.collect.ImmutableList;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.searchlib.rankingexpression.rule.Function;
import com.yahoo.vespa.curator.Lock;
import com.yahoo.vespa.hosted.controller.Controller;
import com.yahoo.vespa.hosted.controller.api.integration.BuildService.BuildJob;
import com.yahoo.vespa.hosted.controller.application.DeploymentJobs.JobType;
import com.yahoo.vespa.hosted.controller.persistence.CuratorDb;
import java.util.Deque;
import java.util.List;
import java.util.Optional;
import java.util.function.Consumer;
/**
* Stores a queue for each type of job, and offers jobs from each of these to a periodic
* polling mechanism which is responsible for triggering the offered jobs in an external build service.
*
* @author jvenstad
*/
public class DeploymentQueue {
private final Controller controller;
private final CuratorDb curator;
public DeploymentQueue(Controller controller, CuratorDb curator) {
this.controller = controller;
this.curator = curator;
}
/** Add the given application to the queue of the given job type -- in front if first, at the back otherwise. */
public void addJob(ApplicationId applicationId, JobType jobType, boolean first) {
locked(jobType, queue -> {
if ( ! queue.contains(applicationId)) {
if (first)
queue.addFirst(applicationId);
else
queue.addLast(applicationId);
}
});
}
/** List all jobs currently enqueued. */
public List<BuildJob> jobs() {
ImmutableList.Builder<BuildJob> builder = ImmutableList.builder();
for (JobType jobType : JobType.values())
for (ApplicationId id : curator.readJobQueue(jobType))
toBuildJob(id, jobType).ifPresent(builder::add);
return builder.build();
}
/** Remove and return a set of jobs to run. This set will contain only one of each job type for capacity constrained zones. */
public List<BuildJob> takeJobsToRun() {
ImmutableList.Builder<BuildJob> builder = ImmutableList.builder();
for (JobType jobType : JobType.values())
locked(jobType, queue ->
queue.stream()
.limit(isCapacityConstrained(jobType) ? 1 : Long.MAX_VALUE)
.peek(id -> toBuildJob(id, jobType).ifPresent(builder::add))
.forEach(queue::remove));
return builder.build();
}
/** Remove all enqueued jobs for the given application. */
public void removeJobs(ApplicationId applicationId) {
for (JobType jobType : JobType.values())
locked(jobType, queue -> {
while (queue.remove(applicationId)); // Keep removing until not found.
});
}
/** Lock the job queues and read, modify, and store the queue for the given job type. */
private void locked(JobType jobType, Consumer<Deque<ApplicationId>> modifications) {
try (Lock lock = curator.lockJobQueues()) {
Deque<ApplicationId> queue = curator.readJobQueue(jobType);
modifications.accept(queue);
curator.writeJobQueue(jobType, queue);
}
}
private static boolean isCapacityConstrained(JobType jobType) {
return jobType == JobType.stagingTest || jobType == JobType.systemTest;
}
private Optional<BuildJob> toBuildJob(ApplicationId applicationId, JobType jobType) {
return controller.applications().get(applicationId)
.flatMap(application -> application.deploymentJobs().projectId())
.map(projectId -> new BuildJob(projectId, jobType.jobName()));
}
}
|