1 files changed, 119 insertions, 0 deletions
diff --git a/service-monitor/src/main/java/com/yahoo/vespa/service/monitor/ModelGenerator.java b/service-monitor/src/main/java/com/yahoo/vespa/service/monitor/ModelGenerator.java
new file mode 100644
index 00000000000..ef418f99d47
--- /dev/null
+++ b/service-monitor/src/main/java/com/yahoo/vespa/service/monitor/ModelGenerator.java
@@ -0,0 +1,119 @@
+// 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.monitor;
+
+import com.yahoo.config.model.api.ApplicationInfo;
+import com.yahoo.config.model.api.HostInfo;
+import com.yahoo.config.model.api.ServiceInfo;
+import com.yahoo.config.model.api.SuperModel;
+import com.yahoo.config.provision.Zone;
+import com.yahoo.vespa.applicationmodel.ApplicationInstance;
+import com.yahoo.vespa.applicationmodel.ApplicationInstanceId;
+import com.yahoo.vespa.applicationmodel.ApplicationInstanceReference;
+import com.yahoo.vespa.applicationmodel.ClusterId;
+import com.yahoo.vespa.applicationmodel.ConfigId;
+import com.yahoo.vespa.applicationmodel.HostName;
+import com.yahoo.vespa.applicationmodel.ServiceCluster;
+import com.yahoo.vespa.applicationmodel.ServiceClusterKey;
+import com.yahoo.vespa.applicationmodel.ServiceInstance;
+import com.yahoo.vespa.applicationmodel.ServiceType;
+import com.yahoo.vespa.applicationmodel.TenantId;
+
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import java.util.stream.Collectors;
+
+/**
+ * Util to convert SuperModel to ServiceModel and application model classes
+ */
+public class ModelGenerator {
+    public static final String CLUSTER_ID_PROPERTY_NAME = "clustername";
+
+    ServiceModel toServiceModel(
+            SuperModel superModel,
+            Zone zone,
+            List<String> configServerHosts,
+            SlobrokMonitor2 slobrokMonitor) {
+        Map<ApplicationInstanceReference, ApplicationInstance<ServiceMonitorStatus>> applicationInstances = new HashMap<>();
+
+        for (ApplicationInfo applicationInfo : superModel.getAllApplicationInfos()) {
+            ApplicationInstance<ServiceMonitorStatus> applicationInstance = toApplicationInstance(
+                    applicationInfo,
+                    zone,
+                    slobrokMonitor);
+            applicationInstances.put(applicationInstance.reference(), applicationInstance);
+        }
+
+        // The config server is part of the service model (but not super model)
+        ConfigServerApplication configServerApplication = new ConfigServerApplication();
+        ApplicationInstance<ServiceMonitorStatus> configServerApplicationInstance =
+                configServerApplication.toApplicationInstance(configServerHosts);
+        applicationInstances.put(configServerApplicationInstance.reference(), configServerApplicationInstance);
+
+        return new ServiceModel(applicationInstances);
+    }
+
+    ApplicationInstance<ServiceMonitorStatus> toApplicationInstance(
+            ApplicationInfo applicationInfo,
+            Zone zone,
+            SlobrokMonitor2 slobrokMonitor) {
+        Map<ServiceClusterKey, Set<ServiceInstance<ServiceMonitorStatus>>> groupedServiceInstances = new HashMap<>();
+
+        for (HostInfo host : applicationInfo.getModel().getHosts()) {
+            HostName hostName = new HostName(host.getHostname());
+            for (ServiceInfo serviceInfo : host.getServices()) {
+                ServiceClusterKey serviceClusterKey = toServiceClusterKey(serviceInfo);
+                ServiceInstance<ServiceMonitorStatus> serviceInstance =
+                        toServiceInstance(serviceInfo, hostName, slobrokMonitor);
+
+                if (!groupedServiceInstances.containsKey(serviceClusterKey)) {
+                    groupedServiceInstances.put(serviceClusterKey, new HashSet<>());
+                }
+                groupedServiceInstances.get(serviceClusterKey).add(serviceInstance);
+            }
+        }
+
+        Set<ServiceCluster<ServiceMonitorStatus>> serviceClusters = groupedServiceInstances.entrySet().stream()
+                .map(entry -> new ServiceCluster<>(
+                        entry.getKey().clusterId(),
+                        entry.getKey().serviceType(),
+                        entry.getValue()))
+                .collect(Collectors.toSet());
+
+        ApplicationInstance<ServiceMonitorStatus> applicationInstance = new ApplicationInstance<>(
+                new TenantId(applicationInfo.getApplicationId().tenant().toString()),
+                toApplicationInstanceId(applicationInfo, zone),
+                serviceClusters);
+
+        return applicationInstance;
+    }
+
+    ServiceClusterKey toServiceClusterKey(ServiceInfo serviceInfo) {
+        ClusterId clusterId = new ClusterId(serviceInfo.getProperty(CLUSTER_ID_PROPERTY_NAME).orElse(""));
+        ServiceType serviceType = toServiceType(serviceInfo);
+        return new ServiceClusterKey(clusterId, serviceType);
+    }
+
+    ServiceInstance<ServiceMonitorStatus> toServiceInstance(
+            ServiceInfo serviceInfo,
+            HostName hostName,
+            SlobrokMonitor2 slobrokMonitor) {
+        ConfigId configId = new ConfigId(serviceInfo.getConfigId());
+        ServiceMonitorStatus serviceStatus = slobrokMonitor.getStatus(toServiceType(serviceInfo), configId);
+        return new ServiceInstance<>(configId, hostName,serviceStatus);
+    }
+
+    ApplicationInstanceId toApplicationInstanceId(ApplicationInfo applicationInfo, Zone zone) {
+        return new ApplicationInstanceId(String.format("%s:%s:%s:%s",
+                applicationInfo.getApplicationId().application().value(),
+                zone.environment().value(),
+                zone.region().value(),
+                applicationInfo.getApplicationId().instance().value()));
+    }
+
+    ServiceType toServiceType(ServiceInfo serviceInfo) {
+        return new ServiceType(serviceInfo.getServiceType());
+    }
+}