From 2fb444df86d56246a8b5cee515eb91490c8dee05 Mon Sep 17 00:00:00 2001
From: Anton Tananaev <anton.tananaev@gmail.com>
Date: Sat, 15 Sep 2018 11:29:01 +1200
Subject: More work on service implementation

---
 src/org/traccar/Main.java | 76 ++++++++++++++++++++++++++++-------------------
 1 file changed, 45 insertions(+), 31 deletions(-)

(limited to 'src/org/traccar/Main.java')

diff --git a/src/org/traccar/Main.java b/src/org/traccar/Main.java
index 986731240..992bfd8de 100644
--- a/src/org/traccar/Main.java
+++ b/src/org/traccar/Main.java
@@ -71,11 +71,16 @@ public final class Main {
             throw new RuntimeException("Configuration file is not provided");
         }
 
-        String configFile = args[args.length - 1];
+        final String configFile = args[args.length - 1];
 
-        if (args.length > 1) {
-            WindowsService windowsService = new WindowsService("traccar");
-            switch (args[1]) {
+        if (args[0].startsWith("--")) {
+            WindowsService windowsService = new WindowsService("traccar") {
+                @Override
+                public void run() {
+                    Main.run(configFile);
+                }
+            };
+            switch (args[0]) {
                 case "--install":
                     windowsService.install("traccar", null, null, null, null, configFile);
                     return;
@@ -87,40 +92,49 @@ public final class Main {
                     windowsService.init();
                     break;
             }
+        } else {
+            run(configFile);
         }
+    }
 
-        Context.init(configFile);
-        logSystemInfo();
-        LOGGER.info("Version: " + Context.getAppVersion());
-        LOGGER.info("Starting server...");
-
-        Context.getServerManager().start();
-        if (Context.getWebServer() != null) {
-            Context.getWebServer().start();
-        }
+    public static void run(String configFile) {
+        try {
+            Context.init(configFile);
+            logSystemInfo();
+            LOGGER.info("Version: " + Context.getAppVersion());
+            LOGGER.info("Starting server...");
+
+            Context.getServerManager().start();
+            if (Context.getWebServer() != null) {
+                Context.getWebServer().start();
+            }
 
-        new Timer().scheduleAtFixedRate(new TimerTask() {
-            @Override
-            public void run() {
-                try {
-                    Context.getDataManager().clearHistory();
-                } catch (SQLException error) {
-                    LOGGER.warn(null, error);
+            new Timer().scheduleAtFixedRate(new TimerTask() {
+                @Override
+                public void run() {
+                    try {
+                        Context.getDataManager().clearHistory();
+                    } catch (SQLException error) {
+                        LOGGER.warn(null, error);
+                    }
                 }
-            }
-        }, 0, CLEAN_PERIOD);
+            }, 0, CLEAN_PERIOD);
 
-        Runtime.getRuntime().addShutdownHook(new Thread() {
-            @Override
-            public void run() {
-                LOGGER.info("Shutting down server...");
+            Runtime.getRuntime().addShutdownHook(new Thread() {
+                @Override
+                public void run() {
+                    LOGGER.info("Shutting down server...");
 
-                if (Context.getWebServer() != null) {
-                    Context.getWebServer().stop();
+                    if (Context.getWebServer() != null) {
+                        Context.getWebServer().stop();
+                    }
+                    Context.getServerManager().stop();
                 }
-                Context.getServerManager().stop();
-            }
-        });
+            });
+        } catch (Exception e) {
+            LOGGER.error(null, e);
+            throw new RuntimeException(e);
+        }
     }
 
 }
-- 
cgit v1.2.3