blob: 60edeefeeea6624e8dcd54edfd56109fbb566e9b (
plain) (
blame)
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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.service.model;
import com.yahoo.config.model.api.ApplicationInfo;
import com.yahoo.config.model.api.HostInfo;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.vespa.applicationmodel.ApplicationInstanceReference;
import com.yahoo.vespa.applicationmodel.HostName;
import com.yahoo.vespa.service.monitor.DuperModelListener;
import com.yahoo.vespa.service.monitor.ServiceHostListener;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import java.util.Set;
import java.util.stream.Collectors;
/**
* Allows a {@link ServiceHostListener} to register with the duper model as a {@link DuperModelListener}.
*
* <p>This class is not thread-safe: As with the DuperModelListener, events from the duper model
* happens within an exclusive duper model lock.</p>
*/
public class ServiceHostListenerAdapter implements DuperModelListener {
private final ServiceHostListener listener;
private final ModelGenerator modelGenerator;
private final Map<ApplicationInstanceReference, Set<HostName>> hostnamesByReference = new HashMap<>();
public static ServiceHostListenerAdapter asDuperModelListener(ServiceHostListener listener,
ModelGenerator generator) {
return new ServiceHostListenerAdapter(listener, generator);
}
ServiceHostListenerAdapter(ServiceHostListener listener, ModelGenerator generator) {
this.listener = listener;
this.modelGenerator = generator;
}
@Override
public void applicationActivated(ApplicationInfo application) {
Set<HostName> newHostnames = application.getModel().getHosts().stream()
.map(HostInfo::getHostname)
.map(HostName::new)
.collect(Collectors.toSet());
var reference = toApplicationInstanceReference(application.getApplicationId());
Set<HostName> oldHostnames = hostnamesByReference.get(reference);
if (!Objects.equals(newHostnames, oldHostnames)) {
hostnamesByReference.put(reference, newHostnames);
listener.onApplicationActivate(reference, newHostnames);
}
}
@Override
public void applicationRemoved(ApplicationId applicationId) {
var reference = toApplicationInstanceReference(applicationId);
if (hostnamesByReference.remove(reference) != null) {
listener.onApplicationRemove(reference);
}
}
@Override
public void bootstrapComplete() {
}
private ApplicationInstanceReference toApplicationInstanceReference(ApplicationId applicationId) {
return modelGenerator.toApplicationInstanceReference(applicationId);
}
}
|