X-Git-Url: http://wamblee.org/gitweb/?a=blobdiff_plain;ds=inline;f=monitor%2Fsrc%2Fmain%2Fjava%2Forg%2Fwamblee%2Fupnpmonitor%2FMonitor.java;fp=monitor%2Fsrc%2Fmain%2Fjava%2Forg%2Fwamblee%2Fupnpmonitor%2FMonitor.java;h=fab9fe643b66292103c5472ce33db7d8bd2939cd;hb=c2b0bc89067fda22755e884a771b809463ea8e30;hp=0000000000000000000000000000000000000000;hpb=f69194edc31d6dce4371d6af6d3bb558cdf77757;p=upnpmonitor

diff --git a/monitor/src/main/java/org/wamblee/upnpmonitor/Monitor.java b/monitor/src/main/java/org/wamblee/upnpmonitor/Monitor.java
new file mode 100644
index 0000000..fab9fe6
--- /dev/null
+++ b/monitor/src/main/java/org/wamblee/upnpmonitor/Monitor.java
@@ -0,0 +1,101 @@
+package org.wamblee.upnpmonitor;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.concurrent.ScheduledThreadPoolExecutor;
+import java.util.concurrent.TimeUnit;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+
+import org.wamblee.io.SimpleProcess;
+
+public class Monitor implements Runnable {
+
+    private static final Logger LOGGER = Logger.getLogger(Monitor.class
+        .getName());
+
+    private ScheduledThreadPoolExecutor executor;
+    private Config config;
+    private boolean serviceFound;
+    private UpnpStack stack;
+
+    public Monitor(Config aConfig) {
+        config = aConfig;
+        executor = new ScheduledThreadPoolExecutor(1);
+    }
+
+    public void start() {
+        serviceFound = true;
+        executor.scheduleWithFixedDelay(this, 0, config.getIntervalSeconds(),
+            TimeUnit.SECONDS);
+        executeCommand(config.getStartupCommand());
+    }
+
+    public synchronized void setServiceFound(boolean aServiceFound) {
+        serviceFound = aServiceFound;
+    }
+
+    public synchronized void run() {
+
+        if (!serviceFound) {
+            LOGGER.info("UPNP service is down, executing recovery");
+            // we need to execute recovery.
+            executeCommand(config.getShutdownCommand());
+            executeCommand(config.getStartupCommand());
+            setServiceFound(true);
+            // give process a chance to startup.
+            return;
+        }
+
+        shutdownStack();
+
+        setServiceFound(false);
+
+        stack = new UpnpStack(new UpnpStack.Listener() {
+            @Override
+            public void deviceAdded(String aDeviceString) {
+                LOGGER.fine("Device added: " + aDeviceString);
+                if (aDeviceString.toLowerCase().contains(
+                    config.getPattern().toLowerCase())) {
+                    synchronized (Monitor.this) {
+                        setServiceFound(true);
+                    }
+                }
+            }
+
+            @Override
+            public void deviceRemoved(String aDeviceString) {
+                LOGGER.fine("Device removed:" + aDeviceString);
+            }
+        });
+        stack.search();
+    }
+
+    private void shutdownStack() {
+        if (stack != null) {
+            stack.shutdown();
+            stack = null;
+        }
+    }
+
+    private void executeCommand(String command) {
+        LOGGER.info("Executing command: " + command);
+        SimpleProcess process = new SimpleProcess(new File("."), new String[] {
+            "sh", "-c", command });
+
+        try {
+            int ret = process.run();
+            LOGGER.info(process.getStdout());
+            LOGGER.info(process.getStderr());
+            LOGGER.info("Exit status: " + ret);
+        } catch (IOException e) {
+            LOGGER.log(Level.INFO, "Problem executing command", e);
+        }
+    }
+
+    public void stop() {
+        executor.shutdown();
+        shutdownStack();
+        executeCommand(config.getShutdownCommand());
+    }
+}