mcp测试通过

ctx/request_context.go

@@ -12,6 +12,7 @@ type RequestContext struct {
 	InstanceName string `json:"instance_name"`
 	TenantID     string `json:"tenant_id"`
 	UserID       string `json:"user_id"`
+	ProjectID    string `json:"project_id"`
 }
 
 // 内部key，不会和其他包冲突

graceful/shutdown.go

@@ -0,0 +1,55 @@
+// graceful/shutdown.go
+package graceful
+
+import (
+	"context"
+	"log"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"git.x2erp.com/qdy/go-base/container"
+	"git.x2erp.com/qdy/go-base/logger"
+)
+
+// WaitForShutdown 带可选清理函数的优雅关闭
+func WaitForShutdown(serviceName string, server *http.Server, containerFactory *container.ContainerFactory) {
+
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+
+	log.Printf("按 Ctrl+C 停止服务 %s", serviceName)
+
+	// 等待信号
+	<-quit
+	log.Printf("接收到终止信号，正在优雅关闭服务 %s...", serviceName)
+
+	//退出注册中心
+	//consul.Deregister(s.serviceName, s.Ip, s.Port, b.Cfg.GetConsulConfig())
+
+	// 创建关闭上下文，给30秒完成当前请求
+	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+	defer cancel()
+
+	// 停止接收新请求，完成当前请求
+	if err := server.Shutdown(ctx); err != nil {
+		log.Printf("服务 %s 关闭失败: %v", serviceName, err)
+	} else {
+		log.Printf("服务 %s 已关闭", serviceName)
+	}
+
+	// 执行关闭处理
+	if containerFactory != nil {
+		containerFactory.CloseAll()
+	}
+
+	// 停止日志写入
+	logger.StopESWriter()
+	log.Printf("服务 %s 优雅关闭完成", serviceName)
+
+	// 等待一小段时间确保日志写入完成
+	time.Sleep(100 * time.Millisecond)
+	os.Exit(0)
+}

webx/webx_factory.go

@@ -1,7 +1,6 @@
 package webx
 
 import (
-	"context"
 	"fmt"
 	"log"
 	"net/http"
@@ -13,7 +12,6 @@ import (
 
 	"git.x2erp.com/qdy/go-base/client"
 	"git.x2erp.com/qdy/go-base/config/subconfigs"
-	"git.x2erp.com/qdy/go-base/container"
 	"git.x2erp.com/qdy/go-base/logger"
 )
 
@@ -202,42 +200,42 @@ func (s *WebService) RunTLS(serviceName, certFile, keyFile string) {
 	//s.waitForServiceShutdown(containerFactory)
 }
 
-// waitForServiceShutdown 等待单个服务关闭
-func (s *WebService) WaitForServiceShutdown(containerFactory *container.ContainerFactory) {
+// // waitForServiceShutdown 等待单个服务关闭
+// func (s *WebService) WaitForServiceShutdown(containerFactory *container.ContainerFactory) {
 
-	log.Printf("按 Ctrl+C 停止服务 %s", s.serviceName)
+// 	log.Printf("按 Ctrl+C 停止服务 %s", s.serviceName)
 
-	// 等待信号
-	<-s.quit
-	log.Printf("接收到终止信号，正在优雅关闭服务 %s...", s.serviceName)
+// 	// 等待信号
+// 	<-s.quit
+// 	log.Printf("接收到终止信号，正在优雅关闭服务 %s...", s.serviceName)
 
-	//退出注册中心
-	//consul.Deregister(s.serviceName, s.Ip, s.Port, b.Cfg.GetConsulConfig())
+// 	//退出注册中心
+// 	//consul.Deregister(s.serviceName, s.Ip, s.Port, b.Cfg.GetConsulConfig())
 
-	// 创建关闭上下文，给30秒完成当前请求
-	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
-	defer cancel()
+// 	// 创建关闭上下文，给30秒完成当前请求
+// 	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+// 	defer cancel()
 
-	// 停止接收新请求，完成当前请求
-	if err := s.httpServer.Shutdown(ctx); err != nil {
-		log.Printf("服务 %s 关闭失败: %v", s.serviceName, err)
-	} else {
-		log.Printf("服务 %s 已关闭", s.serviceName)
-	}
+// 	// 停止接收新请求，完成当前请求
+// 	if err := s.httpServer.Shutdown(ctx); err != nil {
+// 		log.Printf("服务 %s 关闭失败: %v", s.serviceName, err)
+// 	} else {
+// 		log.Printf("服务 %s 已关闭", s.serviceName)
+// 	}
 
-	// 执行关闭处理
-	if containerFactory != nil {
-		containerFactory.CloseAll()
-	}
+// 	// 执行关闭处理
+// 	if containerFactory != nil {
+// 		containerFactory.CloseAll()
+// 	}
 
-	// 停止日志写入
-	logger.StopESWriter()
-	log.Printf("服务 %s 优雅关闭完成", s.serviceName)
+// 	// 停止日志写入
+// 	logger.StopESWriter()
+// 	log.Printf("服务 %s 优雅关闭完成", s.serviceName)
 
-	// 等待一小段时间确保日志写入完成
-	time.Sleep(100 * time.Millisecond)
-	os.Exit(0)
-}
+// 	// 等待一小段时间确保日志写入完成
+// 	time.Sleep(100 * time.Millisecond)
+// 	os.Exit(0)
+// }
 
 // AddMiddleware 添加中间件
 func (s *WebService) AddMiddleware(middleware func(http.Handler) http.Handler) {