diff --git a/web/web.go b/web/web.go index ef57a71a..5607d18c 100644 --- a/web/web.go +++ b/web/web.go @@ -49,6 +49,32 @@ type wrapAssetsFS struct { embed.FS } +// Keep-Alive 监听器包装器:用于拦截新连接并设置 Keep-Alive 选项 +type keepAliveListener struct { + *net.TCPListener + KeepAlivePeriod time.Duration +} + +// Accept 方法:拦截连接并设置 Keep-Alive +func (l keepAliveListener) Accept() (net.Conn, error) { + // 1. 接受底层 TCP 连接 + tc, err := l.TCPListener.AcceptTCP() + if err != nil { + return nil, err + } + + // 2. 在 *net.TCPConn 上设置 Keep-Alive 属性 (这里的方法是正确的) + if err := tc.SetKeepAlive(true); err != nil { + logger.Warning("Failed to set KeepAlive:", err) + } + // 设置心跳包周期为 5 秒 + if err := tc.SetKeepAlivePeriod(l.KeepAlivePeriod); err != nil { + logger.Warning("Failed to set KeepAlivePeriod:", err) + } + + return tc, nil +} + func (f *wrapAssetsFS) Open(name string) (fs.File, error) { file, err := f.FS.Open("assets/" + name) if err != nil { @@ -93,6 +119,8 @@ type Server struct { xrayService service.XrayService settingService service.SettingService tgbotService service.Tgbot + // 〔中文注释〕: 添加这个字段,用来“持有”从 main.go 传递过来的 serverService 实例。 + serverService service.ServerService cron *cron.Cron @@ -100,14 +128,23 @@ type Server struct { cancel context.CancelFunc } -func NewServer() *Server { +// 【新增方法】:用于 main.go 将创建好的 tgBotService 注入进来 +func (s *Server) SetTelegramService(tgService service.TelegramService) { + s.tgbotService = tgService +} + +// 〔中文注释〕: 1. 让 NewServer 能够接收一个 serverService 实例作为参数。 +func NewServer(serverService service.ServerService) *Server { ctx, cancel := context.WithCancel(context.Background()) return &Server{ ctx: ctx, cancel: cancel, + // 〔中文注释〕: 2. 将传入的 serverService 存储到 Server 结构体的字段中。 + serverService: serverService, } } + func (s *Server) getHtmlFiles() ([]string, error) { files := make([]string, 0) dir, _ := os.Getwd() @@ -243,7 +280,8 @@ func (s *Server) initRouter() (*gin.Engine, error) { g := engine.Group(basePath) s.index = controller.NewIndexController(g) - s.server = controller.NewServerController(g) + // 〔中文注释〕: 调用我们刚刚改造过的 NewServerController,并将 s.serverService 作为参数传进去。 + s.server = controller.NewServerController(g, s.serverService) s.panel = controller.NewXUIController(g) s.api = controller.NewAPIController(g) @@ -367,10 +405,30 @@ func (s *Server) Start() (err error) { listenAddr = net.JoinHostPort(listen, strconv.Itoa(port)) } - listener, err := net.Listen("tcp", listenAddr) - if err != nil { - return err - } + // 1. 使用 baseListener 临时变量接收 net.Listen 的结果,这是底层的 TCP 监听器 + baseListener, err := net.Listen("tcp", listenAddr) + if err != nil { + return err + } + + // 声明最终要使用的 listener 变量 + var listener net.Listener + + // 2. 尝试将 net.Listener 断言为 *net.TCPListener,以便进行更底层的设置 + tcpListener, ok := baseListener.(*net.TCPListener) + if !ok { + // 如果断言失败 (例如在某些特殊环境或测试中),则直接使用原始的 listener,不设置 Keep-Alive + logger.Warning("监听器不是 TCPListener 类型, 无法设置 Keep-Alive。") + listener = baseListener + } else { + // 3. 【核心功能】: 使用自定义的包装器为每一个新的连接设置 Keep-Alive 属性 + kaListener := &keepAliveListener{ + TCPListener: tcpListener, + KeepAlivePeriod: 5 * time.Second, // 将 Keep-Alive 探测周期设置为 5 秒 + } + // 将包装后的监听器赋值给最终的 listener 变量,后续流程将使用这个新的 listener + listener = net.Listener(kaListener) + } // 再次检查证书,配置 TLS Listener if certFile != "" && keyFile != "" { @@ -386,21 +444,32 @@ func (s *Server) Start() (err error) { } s.listener = listener + // 修改 s.httpServer 的初始化代码 s.httpServer = &http.Server{ Handler: engine, + // 【新增】:设置 120 秒的读写超时,确保 ufw 命令有足够的时间完成 + ReadTimeout: 120 * time.Second, + WriteTimeout: 120 * time.Second, } + go func() { s.httpServer.Serve(listener) }() s.startTask() - isTgbotenabled, err := s.settingService.GetTgbotEnabled() - if (err == nil) && (isTgbotenabled) { - tgBot := s.tgbotService.NewTgbot() - tgBot.Start(i18nFS) - } + // 启动 TG Bot + isTgbotenabled, err := s.settingService.GetTgbotEnabled() + if (err == nil) && (isTgbotenabled) { + // 现在直接在注入的实例上调用 Start 方法,而不是 NewTgbot() + // 因为 main.go 已经注入了完整的实例 + if tgbot, ok := s.tgbotService.(*service.Tgbot); ok { + tgbot.Start(i18nFS) + } else { + logger.Warning("Telegram Bot 已启用,但注入的实例类型不正确或为 nil,无法启动。") + } + } return nil } @@ -408,8 +477,11 @@ func (s *Server) Start() (err error) { func (s *Server) Stop() error { s.cancel() s.xrayService.StopXray() - if s.cron != nil { - s.cron.Stop() + // 只有在断言成功后,才能调用只在 *service.Tgbot 上定义的 Stop() 和 IsRunning() 方法。 + if tgBot, ok := s.tgbotService.(*service.Tgbot); ok { + if tgBot.IsRunning() { + tgBot.Stop() + } } if s.tgbotService.IsRunning() { s.tgbotService.Stop()