浏览代码

fix: catch process termination signal to clear socket file

Hintay 8 月之前
父节点
当前提交
e634c75804
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      main.go

+ 2 - 2
main.go

@@ -10,8 +10,8 @@ import (
 	"fmt"
 	"fmt"
 	"net"
 	"net"
 	"net/http"
 	"net/http"
-	"os"
 	"os/signal"
 	"os/signal"
+	"syscall"
 
 
 	"github.com/0xJacky/Nginx-UI/internal/cert"
 	"github.com/0xJacky/Nginx-UI/internal/cert"
 	"github.com/0xJacky/Nginx-UI/internal/cmd"
 	"github.com/0xJacky/Nginx-UI/internal/cmd"
@@ -108,7 +108,7 @@ func main() {
 	confPath := appCmd.String("config")
 	confPath := appCmd.String("config")
 	settings.Init(confPath)
 	settings.Init(confPath)
 
 
-	ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt)
+	ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
 	defer cancel()
 	defer cancel()
 
 
 	err := risefront.New(ctx, risefront.Config{
 	err := risefront.New(ctx, risefront.Config{