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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.provision.restapi;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.config.provision.ZoneEndpoint.AllowedUrn;
import com.yahoo.container.jdisc.HttpRequest;
import com.yahoo.restapi.SlimeJsonResponse;
import com.yahoo.slime.Cursor;
import com.yahoo.vespa.hosted.provision.NodeRepository;
import com.yahoo.vespa.hosted.provision.lb.LoadBalancer;
import com.yahoo.vespa.hosted.provision.lb.LoadBalancerInstance;
import com.yahoo.vespa.hosted.provision.lb.LoadBalancerList;
import java.util.List;
import java.util.Optional;
/**
* @author mpolden
*/
public class LoadBalancersResponse extends SlimeJsonResponse {
private final NodeRepository nodeRepository;
private final HttpRequest request;
public LoadBalancersResponse(HttpRequest request, NodeRepository nodeRepository) {
this.request = request;
this.nodeRepository = nodeRepository;
Cursor root = slime.setObject();
toSlime(loadBalancers(), root);
}
private Optional<ApplicationId> application() {
return Optional.ofNullable(request.getProperty("application"))
.map(ApplicationId::fromFullString);
}
private List<LoadBalancer> loadBalancers() {
LoadBalancerList loadBalancers;
var application = application();
if (application.isPresent()) {
loadBalancers = nodeRepository.loadBalancers().list(application.get());
} else {
loadBalancers = nodeRepository.loadBalancers().list();
}
return loadBalancers.asList();
}
private void toSlime(List<LoadBalancer> loadBalancers, Cursor object) {
Cursor loadBalancerArray = object.setArray("loadBalancers");
loadBalancers.forEach(lb -> {
Cursor lbObject = loadBalancerArray.addObject();
lbObject.setString("id", lb.id().serializedForm());
lbObject.setString("state", lb.state().name());
lbObject.setLong("changedAt", lb.changedAt().toEpochMilli());
lbObject.setString("application", lb.id().application().application().value());
lbObject.setString("tenant", lb.id().application().tenant().value());
lbObject.setString("instance", lb.id().application().instance().value());
lbObject.setString("cluster", lb.id().cluster().value());
lb.instance().flatMap(LoadBalancerInstance::hostname).ifPresent(hostname -> lbObject.setString("hostname", hostname.value()));
lb.instance().flatMap(LoadBalancerInstance::ip4Address).ifPresent(ip -> lbObject.setString("ipAddress", ip));
lb.instance().flatMap(LoadBalancerInstance::ip6Address).ifPresent(ip -> lbObject.setString("ip6Address", ip));
lb.instance().flatMap(LoadBalancerInstance::dnsZone).ifPresent(dnsZone -> lbObject.setString("dnsZone", dnsZone.id()));
Cursor networkArray = lbObject.setArray("networks");
lb.instance().ifPresent(instance -> instance.networks().forEach(networkArray::addString));
Cursor portArray = lbObject.setArray("ports");
lb.instance().ifPresent(instance -> instance.ports().forEach(portArray::addLong));
Cursor realArray = lbObject.setArray("reals");
lb.instance().ifPresent(instance -> {
instance.reals().forEach(real -> {
Cursor realObject = realArray.addObject();
realObject.setString("hostname", real.hostname().value());
realObject.setString("ipAddress", real.ipAddress());
realObject.setLong("port", real.port());
});
});
lb.instance().ifPresent(instance -> {
if ( ! instance.settings().isDefault()) {
Cursor urnsArray = lbObject.setObject("settings").setArray("allowedUrns");
for (AllowedUrn urn : instance.settings().allowedUrns()) {
Cursor urnObject = urnsArray.addObject();
urnObject.setString("type", switch (urn.type()) {
case awsPrivateLink -> "aws-private-link";
case gcpServiceConnect -> "gcp-service-connect";
});
urnObject.setString("urn", urn.urn());
}
}
instance.serviceId().ifPresent(serviceId -> lbObject.setString("serviceId", serviceId.value()));
lbObject.setBool("public", instance.settings().isPublicEndpoint());
});
lb.instance()
.map(LoadBalancerInstance::cloudAccount)
.filter(cloudAccount -> !cloudAccount.isUnspecified())
.ifPresent(cloudAccount -> lbObject.setString("cloudAccount", cloudAccount.value()));
});
}
}
|
