From 0bc22154e5c83f52bc09cf9c803c2766ddc0a5fe Mon Sep 17 00:00:00 2001
From: Darien Raymond <admin@v2ray.com>
Date: Tue, 4 Dec 2018 14:15:35 +0100
Subject: [PATCH] defer server close

---
 main/main.go | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/main/main.go b/main/main.go
index af78b83f4..f91a9d78b 100644
--- a/main/main.go
+++ b/main/main.go
@@ -109,6 +109,7 @@ func main() {
 		// Configuration error. Exit with a special value to prevent systemd from restarting.
 		os.Exit(23)
 	}
+	defer server.Close()
 
 	if *test {
 		fmt.Println("Configuration OK.")
@@ -123,9 +124,9 @@ func main() {
 	// Explicitly triggering GC to remove garbage from config loading.
 	runtime.GC()
 
-	osSignals := make(chan os.Signal, 1)
-	signal.Notify(osSignals, os.Interrupt, os.Kill, syscall.SIGTERM)
-
-	<-osSignals
-	server.Close()
+	{
+		osSignals := make(chan os.Signal, 1)
+		signal.Notify(osSignals, os.Interrupt, os.Kill, syscall.SIGTERM)
+		<-osSignals
+	}
 }