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
|
// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.provision.restapi.v2;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.container.jdisc.HttpRequest;
import com.yahoo.container.jdisc.HttpResponse;
import com.yahoo.slime.Cursor;
import com.yahoo.slime.JsonFormat;
import com.yahoo.slime.Slime;
import com.yahoo.vespa.hosted.provision.NodeRepository;
import com.yahoo.vespa.hosted.provision.lb.LoadBalancer;
import com.yahoo.vespa.hosted.provision.lb.LoadBalancerList;
import com.yahoo.vespa.hosted.provision.node.filter.ApplicationFilter;
import java.io.IOException;
import java.io.OutputStream;
import java.util.List;
import java.util.Optional;
/**
* @author mpolden
*/
public class LoadBalancersResponse extends HttpResponse {
private final NodeRepository nodeRepository;
private final HttpRequest request;
public LoadBalancersResponse(HttpRequest request, NodeRepository nodeRepository) {
super(200);
this.request = request;
this.nodeRepository = nodeRepository;
}
private Optional<ApplicationId> application() {
return Optional.ofNullable(request.getProperty("application"))
.map(ApplicationFilter::toApplicationId);
}
private List<LoadBalancer> loadBalancers() {
LoadBalancerList loadBalancers = nodeRepository.loadBalancers();
return application().map(loadBalancers::owner)
.map(LoadBalancerList::asList)
.orElseGet(loadBalancers::asList);
}
@Override
public String getContentType() { return "application/json"; }
@Override
public void render(OutputStream stream) throws IOException {
Slime slime = new Slime();
Cursor root = slime.setObject();
Cursor loadBalancerArray = root.setArray("loadBalancers");
loadBalancers().forEach(lb -> {
Cursor lbObject = loadBalancerArray.addObject();
lbObject.setString("id", lb.id().serializedForm());
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());
lbObject.setString("hostname", lb.instance().hostname().value());
lb.instance().dnsZone().ifPresent(dnsZone -> lbObject.setString("dnsZone", dnsZone.id()));
Cursor networkArray = lbObject.setArray("networks");
lb.instance().networks().forEach(networkArray::addString);
Cursor portArray = lbObject.setArray("ports");
lb.instance().ports().forEach(portArray::addLong);
Cursor realArray = lbObject.setArray("reals");
lb.instance().reals().forEach(real -> {
Cursor realObject = realArray.addObject();
realObject.setString("hostname", real.hostname().value());
realObject.setString("ipAddress", real.ipAddress());
realObject.setLong("port", real.port());
});
Cursor rotationArray = lbObject.setArray("rotations");
lb.rotations().forEach(rotation -> {
Cursor rotationObject = rotationArray.addObject();
rotationObject.setString("name", rotation.value());
});
lbObject.setBool("inactive", lb.inactive());
});
new JsonFormat(true).encode(stream, slime);
}
}
|
