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
|
// Copyright 2020 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.provision.provisioning;
import com.google.common.base.Supplier;
import com.google.common.base.Suppliers;
import com.yahoo.config.provision.DockerImage;
import com.yahoo.config.provision.NodeType;
import com.yahoo.vespa.curator.Lock;
import com.yahoo.vespa.hosted.provision.persistence.CuratorDatabaseClient;
import java.time.Duration;
import java.util.Collections;
import java.util.Map;
import java.util.Optional;
import java.util.concurrent.TimeUnit;
import java.util.logging.Logger;
/**
* Multithread safe class to get and set docker images for given host types.
*
* @author freva
*/
public class DockerImages {
private static final Duration defaultCacheTtl = Duration.ofMinutes(1);
private static final Logger log = Logger.getLogger(DockerImages.class.getName());
private final CuratorDatabaseClient db;
private final DockerImage defaultImage;
private final Duration cacheTtl;
/**
* Docker image is read on every request to /nodes/v2/node/[fqdn]. Cache current getDockerImages to avoid
* unnecessary ZK reads. When getDockerImages change, some nodes may need to wait for TTL until they see the new target,
* this is fine.
*/
private volatile Supplier<Map<NodeType, DockerImage>> dockerImages;
public DockerImages(CuratorDatabaseClient db, DockerImage defaultImage) {
this(db, defaultImage, defaultCacheTtl);
}
DockerImages(CuratorDatabaseClient db, DockerImage defaultImage, Duration cacheTtl) {
this.db = db;
this.defaultImage = defaultImage;
this.cacheTtl = cacheTtl;
createCache();
}
private void createCache() {
this.dockerImages = Suppliers.memoizeWithExpiration(() -> Collections.unmodifiableMap(db.readDockerImages()),
cacheTtl.toMillis(), TimeUnit.MILLISECONDS);
}
/** Returns the current docker images for each node type */
public Map<NodeType, DockerImage> getDockerImages() {
return dockerImages.get();
}
/** Returns the current docker image for given node type, or default */
public DockerImage dockerImageFor(NodeType type) {
return getDockerImages().getOrDefault(type, defaultImage);
}
/** Set the docker image for nodes of given type */
public void setDockerImage(NodeType nodeType, Optional<DockerImage> dockerImage) {
if (nodeType.isDockerHost()) {
throw new IllegalArgumentException("Setting docker image for " + nodeType + " nodes is unsupported");
}
try (Lock lock = db.lockDockerImages()) {
Map<NodeType, DockerImage> dockerImages = db.readDockerImages();
dockerImage.ifPresentOrElse(image -> dockerImages.put(nodeType, image),
() -> dockerImages.remove(nodeType));
db.writeDockerImages(dockerImages);
createCache(); // Throw away current cache
log.info("Set docker image for " + nodeType + " nodes to " + dockerImage.map(DockerImage::asString).orElse(null));
}
}
}
|