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 2017 Yahoo Holdings. 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.provision.Zone;
import com.yahoo.vespa.applicationmodel.ApplicationInstance;
import com.yahoo.vespa.applicationmodel.ApplicationInstanceReference;
import com.yahoo.vespa.applicationmodel.HostName;
import com.yahoo.vespa.applicationmodel.ServiceInstance;
import com.yahoo.vespa.service.monitor.ServiceModel;
import com.yahoo.vespa.service.monitor.ServiceStatusProvider;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import java.util.function.Function;
import java.util.stream.Collectors;
/**
* Util to make ServiceModel and its related application model classes
*/
public class ModelGenerator {
public static final String CLUSTER_ID_PROPERTY_NAME = "clustername";
private final Zone zone;
public ModelGenerator(Zone zone) {
this.zone = zone;
}
/**
* Create service model based primarily on super model.
*
* If the configServerhosts is non-empty, a config server application is added.
*/
public ServiceModel toServiceModel(List<ApplicationInfo> allApplicationInfos,
ServiceStatusProvider serviceStatusProvider) {
Map<ApplicationInstanceReference, ApplicationInstance> applicationInstances =
allApplicationInfos.stream()
.map(info -> new ApplicationInstanceGenerator(info, zone)
.makeApplicationInstance(serviceStatusProvider))
.collect(Collectors.toMap(ApplicationInstance::reference, Function.identity()));
return new ServiceModel(applicationInstances);
}
public Set<ApplicationInstanceReference> toApplicationInstanceReferenceSet(List<ApplicationInfo> infos) {
return infos.stream()
.map(info -> new ApplicationInstanceGenerator(info, zone).toApplicationInstanceReference())
.collect(Collectors.toSet());
}
public Optional<ApplicationInstance> toApplication(List<ApplicationInfo> applicationInfos,
HostName hostname,
ServiceStatusProvider serviceStatusProvider) {
for (var applicationInfo : applicationInfos) {
var generator = new ApplicationInstanceGenerator(applicationInfo, zone);
if (generator.containsHostname(hostname)) {
return Optional.of(generator.makeApplicationInstance(serviceStatusProvider));
}
}
return Optional.empty();
}
public ApplicationInstance toApplication(ApplicationInfo applicationInfo,
ServiceStatusProvider serviceStatusProvider) {
var generator = new ApplicationInstanceGenerator(applicationInfo, zone);
return generator.makeApplicationInstance(serviceStatusProvider);
}
public List<ServiceInstance> toServices(List<ApplicationInfo> applicationInfos,
HostName hostname,
ServiceStatusProvider serviceStatusProvider) {
for (var applicationInfo : applicationInfos) {
var generator = new ApplicationInstanceGenerator(applicationInfo, zone);
ApplicationInstance applicationInstance = generator.makeApplicationInstanceLimitedTo(
hostname, serviceStatusProvider);
List<ServiceInstance> serviceInstances = applicationInstance.serviceClusters().stream()
.flatMap(cluster -> cluster.serviceInstances().stream())
.collect(Collectors.toList());
if (serviceInstances.size() > 0) {
return serviceInstances;
}
}
return List.of();
}
}
|