1 files changed, 89 insertions, 0 deletions
diff --git a/service-monitor/src/main/java/com/yahoo/vespa/service/monitor/SuperModelListenerImpl.java b/service-monitor/src/main/java/com/yahoo/vespa/service/monitor/SuperModelListenerImpl.java
new file mode 100644
index 00000000000..506a2f7d649
--- /dev/null
+++ b/service-monitor/src/main/java/com/yahoo/vespa/service/monitor/SuperModelListenerImpl.java
@@ -0,0 +1,89 @@
+// 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.SuperModel;
+import com.yahoo.config.model.api.SuperModelListener;
+import com.yahoo.config.model.api.SuperModelProvider;
+import com.yahoo.config.provision.ApplicationId;
+import com.yahoo.config.provision.Zone;
+import com.yahoo.vespa.service.monitor.internal.LatencyMeasurement;
+import com.yahoo.vespa.service.monitor.internal.ServiceMonitorMetrics;
+
+import java.util.List;
+import java.util.function.Supplier;
+import java.util.logging.Logger;
+
+public class SuperModelListenerImpl implements SuperModelListener, Supplier<ServiceModel> {
+    private static final Logger logger = Logger.getLogger(SuperModelListenerImpl.class.getName());
+
+    private final ServiceMonitorMetrics metrics;
+    private final ModelGenerator modelGenerator;
+    private final Zone zone;
+    private final List<String> configServerHosts;
+
+    // superModel and slobrokMonitorManager are always updated together
+    // and atomically using this monitor.
+    private final Object monitor = new Object();
+    private final SlobrokMonitorManagerImpl slobrokMonitorManager;
+    private SuperModel superModel;
+
+    SuperModelListenerImpl(SlobrokMonitorManagerImpl slobrokMonitorManager,
+                           ServiceMonitorMetrics metrics,
+                           ModelGenerator modelGenerator,
+                           Zone zone,
+                           List<String> configServerHosts) {
+        this.slobrokMonitorManager = slobrokMonitorManager;
+        this.metrics = metrics;
+        this.modelGenerator = modelGenerator;
+        this.zone = zone;
+        this.configServerHosts = configServerHosts;
+    }
+
+    void start(SuperModelProvider superModelProvider) {
+        synchronized (monitor) {
+            // This snapshot() call needs to be within the synchronized block,
+            // since applicationActivated()/applicationRemoved() may be called
+            // asynchronously even before snapshot() returns.
+            SuperModel snapshot = superModelProvider.snapshot(this);
+
+            snapshot.getAllApplicationInfos().stream().forEach(application ->
+                    applicationActivated(snapshot, application));
+        }
+    }
+
+    @Override
+    public void applicationActivated(SuperModel superModel, ApplicationInfo application) {
+        synchronized (monitor) {
+            this.superModel = superModel;
+            slobrokMonitorManager.applicationActivated(superModel, application);
+        }
+    }
+
+    @Override
+    public void applicationRemoved(SuperModel superModel, ApplicationId id) {
+        synchronized (monitor) {
+            this.superModel = superModel;
+            slobrokMonitorManager.applicationRemoved(superModel, id);
+        }
+    }
+
+    @Override
+    public ServiceModel get() {
+        try (LatencyMeasurement measurement = metrics.startServiceModelSnapshotLatencyMeasurement()) {
+            // Reference 'measurement' in a dummy statement, otherwise the compiler
+            // complains about "auto-closeable resource is never referenced in body of
+            // corresponding try statement". Why hasn't javac fixed this!?
+            dummy(measurement);
+
+            // WARNING: The slobrok monitor manager may be out-of-sync with super model (no locking)
+            return modelGenerator.toServiceModel(
+                    superModel,
+                    zone,
+                    configServerHosts,
+                    slobrokMonitorManager);
+        }
+    }
+
+    private void dummy(LatencyMeasurement measurement) {}
+}
+\ No newline at end of file