fix: Application does not terminate if severe error occurs #185 (#211)

Co-authored-by: Aliaksandr Stsiapanay <[email protected]>
epam · Feb 14, 2024 · 2e1da2b · 2e1da2b
1 parent 26105d5
commit 2e1da2b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/main/java/com/epam/aidial/core/AiDial.java b/src/main/java/com/epam/aidial/core/AiDial.java
@@ -116,7 +116,7 @@ void start() throws Exception {
 
             log.info("Proxy started on {}", server.actualPort());
         } catch (Throwable e) {
-            log.warn("Proxy failed to start:", e);
+            log.error("Proxy failed to start:", e);
             stop();
             throw e;
         }
@@ -240,7 +240,11 @@ private interface AsyncCloser<R> {
 
     public static void main(String[] args) throws Exception {
         AiDial dial = new AiDial();
-        dial.start();
+        try {
+            dial.start();
+        } catch (Throwable e) {
+            System.exit(-1);
+        }
         Runtime.getRuntime().addShutdownHook(new Thread(dial::stop, "shutdown-hook"));
     }