- 提取 App 公共逻辑到 app_common.go,消除 app.go/app_windows.go 重复定义
- 提取 Server 生命周期/中间件/路由逻辑到独立文件(lifecycle.go/middleware_builder.go/router.go)
- 提取 Proxy 缓存处理/头部修改/目标选择到独立模块
- 提取 CheckIPAccess/CheckTokenAuth 到 utils/httperror.go,消除 status/purge 重复实现
- 修复 stream 双向转发:任一方向完成立即关闭双端,避免连接泄漏
- 修复 SSL/TLS 中静默忽略错误的问题,添加日志记录
- 统一日志消息为英文
💘 Generated with Crush
Assisted-by: GLM 5.1 via Crush <crush@charm.land>
115 lines
2.4 KiB
Go
115 lines
2.4 KiB
Go
//go:build windows
|
|
|
|
// Windows lacks POSIX signals (SIGUSR1, SIGUSR2, SIGHUP, SIGQUIT);
|
|
// this file provides stub implementations for those Unix-specific signals.
|
|
package app
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"rua.plus/lolly/internal/server"
|
|
)
|
|
|
|
// Run starts the application: loads config, creates servers, and handles signals (Windows version).
|
|
func (a *App) Run() int {
|
|
if err := a.loadAndValidateConfig(); err != nil {
|
|
return 1
|
|
}
|
|
|
|
a.initVariables()
|
|
a.logServerAddresses()
|
|
a.initResolver()
|
|
a.initServer()
|
|
a.initStreamServers()
|
|
a.initHTTP3()
|
|
a.initHTTP2()
|
|
|
|
a.upgradeMgr = server.NewUpgradeManager(a.srv)
|
|
if a.pidFile != "" {
|
|
a.upgradeMgr.SetPidFile(a.pidFile)
|
|
_ = a.upgradeMgr.WritePid()
|
|
}
|
|
|
|
sigChan := make(chan os.Signal, 1)
|
|
a.setupSignalHandlers(sigChan)
|
|
|
|
errChan := make(chan error, 1)
|
|
go func() {
|
|
a.logger.LogStartup("Starting HTTP server", nil)
|
|
if err := a.srv.Start(); err != nil {
|
|
errChan <- err
|
|
}
|
|
}()
|
|
|
|
sigintCount := 0
|
|
|
|
for {
|
|
select {
|
|
case err := <-errChan:
|
|
a.logger.Error().Err(err).Msg("Server failed to start")
|
|
return 1
|
|
case sig := <-sigChan:
|
|
if sig == syscall.SIGINT {
|
|
sigintCount++
|
|
if sigintCount >= 3 {
|
|
a.logger.LogShutdown("Received 3 SIGINT, forcing exit")
|
|
return 1
|
|
}
|
|
}
|
|
if !a.handleSignal(sig) {
|
|
a.logger.LogShutdown("Server stopped")
|
|
return 0
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (a *App) setupSignalHandlers(sigChan chan<- os.Signal) {
|
|
signal.Notify(sigChan,
|
|
syscall.SIGTERM,
|
|
syscall.SIGINT,
|
|
)
|
|
}
|
|
|
|
// handleSignal returns false to indicate the app should exit (Windows version).
|
|
func (a *App) handleSignal(sig os.Signal) bool {
|
|
switch sig {
|
|
case syscall.SIGTERM, syscall.SIGINT:
|
|
timeout := a.cfg.Shutdown.FastTimeout
|
|
if timeout <= 0 {
|
|
timeout = 5 * time.Second
|
|
}
|
|
a.logger.LogSignal(sigName(sig.(syscall.Signal)), "Stopping server")
|
|
a.shutdownHTTP2()
|
|
a.shutdownHTTP3()
|
|
_ = a.srv.StopWithTimeout(timeout)
|
|
return false
|
|
default:
|
|
a.logger.Info().Str("signal", sig.String()).Msg("Received signal (ignored on Windows)")
|
|
return true
|
|
}
|
|
}
|
|
|
|
func (a *App) reloadConfig() {
|
|
// Windows stub - functionality limited
|
|
}
|
|
|
|
func (a *App) gracefulUpgrade() {
|
|
a.logger.Info().Msg("Graceful upgrade is not supported on Windows")
|
|
}
|
|
|
|
func sigName(sig syscall.Signal) string {
|
|
switch sig {
|
|
case syscall.SIGTERM:
|
|
return "SIGTERM"
|
|
case syscall.SIGINT:
|
|
return "SIGINT"
|
|
default:
|
|
return fmt.Sprintf("Signal(%d)", sig)
|
|
}
|
|
}
|