From 631005a40ce0fd0171bea69d41bc42fea3c38b5d Mon Sep 17 00:00:00 2001
From: xuzhiwei <jsrdxzw@163.com>
Date: Mon, 23 May 2022 11:47:41 +0800
Subject: [PATCH] add notify channel when exit

---
 main.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index 1d8bb86..d8e25dd 100644
--- a/main.go
+++ b/main.go
@@ -15,7 +15,9 @@ import (
 	"net/http/httputil"
 	"net/url"
 	"os"
+	"os/signal"
 	"strings"
+	"syscall"
 
 	"github.com/dtm-labs/dtm/dtmcli/logger"
 	"github.com/dtm-labs/dtm/dtmsvr/config"
@@ -31,7 +33,10 @@ func main() {
 	app, conf := entry.Main(&Version)
 	if app != nil {
 		addAdmin(app, conf)
-		select {}
+		q := make(chan os.Signal)
+		signal.Notify(q, syscall.SIGINT, syscall.SIGTERM)
+		<-q
+		logger.Infof("Shutdown dtm server...")
 	}
 }