From 29c674fdfffc9fba867d6aebfa647233e1fff8f7 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期二, 16 七月 2024 20:15:55 +0800
Subject: [PATCH] 禁用服务端的keepalive

---
 main.go |   24 +++++++++++++-----------
 1 files changed, 13 insertions(+), 11 deletions(-)

diff --git a/main.go b/main.go
index 8130d67..5d71a57 100644
--- a/main.go
+++ b/main.go
@@ -37,6 +37,18 @@
 		Handler: r,
 	}
 
+	srv.SetKeepAlivesEnabled(config.ServeConf.Keepalive)
+
+	// Initializing the server in a goroutine so that
+	// it won't block the graceful shutdown handling below
+	go func() {
+		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+			logger.Error("listen: %s", err)
+		}
+	}()
+
+	logger.Info("Server is running at %s:%s", host, port)
+
 	// 鍚姩1400瀹㈡埛绔�
 	ctx, cancel := context.WithCancel(context.Background())
 	go client.Init1400Client(ctx)
@@ -48,16 +60,6 @@
 
 	// 鍚姩瀹氭椂浠诲姟
 	cron.Init()
-
-	// Initializing the server in a goroutine so that
-	// it won't block the graceful shutdown handling below
-	go func() {
-		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
-			logger.Error("listen: %s", err)
-		}
-	}()
-
-	logger.Info("Server is running at %s:%s", host, port)
 
 	// Wait for interrupt signal to gracefully shutdown the server with
 	// a timeout of 5 seconds.
@@ -72,7 +74,7 @@
 	// The context is used to inform the server it has 5 seconds to finish
 	// the request it is currently handling
 	//ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
-	defer cancel()
+	cancel()
 	if err := srv.Shutdown(ctx); err != nil {
 		logger.Error("Server forced to shutdown:", err)
 	}

--
Gitblit v1.8.0