Fix windows build

Signed-off-by: Joeky <joeky5888@gmail.com>

main.go

@@ -12,7 +12,6 @@ import (
 	"time"
 	"unsafe"
 
-	"github.com/appleboy/graceful"
 	"github.com/valyala/fasthttp"
 	"golang.org/x/exp/slog"
 )
@@ -175,17 +174,5 @@ func main() {
 		}
 	}()
 
-	graceful.NewManager().AddRunningJob(func(ctx context.Context) error {
-		<-ctx.Done()
-		server.DisableKeepalive = true
-		if err := server.Shutdown(); err != nil {
-			Warn("Shutdown err: %s", err)
-			defer os.Exit(1)
-		} else {
-			Info("gracefully stopped")
-		}
-		return nil
-	})
-
-	<-graceful.NewManager().Done()
+	<-wait(server)
 }

wait_unix.go

@@ -0,0 +1,28 @@
+//go:build !windows
+// +build !windows
+
+package main
+
+import (
+	"context"
+	"os"
+
+	"github.com/appleboy/graceful"
+	"github.com/valyala/fasthttp"
+)
+
+func wait(server *fasthttp.Server) <-chan struct{} {
+	graceful.NewManager().AddRunningJob(func(ctx context.Context) error {
+		<-ctx.Done()
+		server.DisableKeepalive = true
+		if err := server.Shutdown(); err != nil {
+			Warn("Shutdown err: %s", err)
+			defer os.Exit(1)
+		} else {
+			Info("gracefully stopped")
+		}
+		return nil
+	})
+
+	return graceful.NewManager().Done()
+}

wait_windows.go

@@ -0,0 +1,31 @@
+//go:build windows
+// +build windows
+
+package main
+
+import (
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/valyala/fasthttp"
+)
+
+func wait(server *fasthttp.Server) <-chan struct{} {
+	ctx := make(chan os.Signal)
+	sig := make(chan struct{})
+	signal.Notify(ctx, os.Interrupt, syscall.SIGTERM)
+	go func() {
+		<-ctx
+		server.DisableKeepalive = true
+		if err := server.Shutdown(); err != nil {
+			Warn("Shutdown err: %s", err)
+			defer os.Exit(1)
+		} else {
+			Info("gracefully stopped")
+		}
+		sig <- struct{}{}
+	}()
+
+	return sig
+}