129 lines
2.9 KiB
Go
129 lines
2.9 KiB
Go
|
package main
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"ego/internal/conf"
|
|||
|
|
"ego/internal/router"
|
|||
|
|
"ego/pkg/logger"
|
|||
|
|
"errors"
|
|||
|
|
"net/http"
|
|||
|
|
"os"
|
|||
|
|
"os/signal"
|
|||
|
|
"syscall"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
"go.uber.org/zap"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// @title EGO API
|
|||
|
|
// @version 1.0
|
|||
|
|
// @description EGO 系统 API 文档
|
|||
|
|
// @termsOfService http://swagger.io/terms/
|
|||
|
|
// @contact.name API Support
|
|||
|
|
// @contact.url http://www.swagger.io/support
|
|||
|
|
// @contact.email support@swagger.io
|
|||
|
|
// @license.name Apache 2.0
|
|||
|
|
// @license.url http://www.apache.org/licenses/LICENSE-2.0.html
|
|||
|
|
// @host 127.0.0.1:3000
|
|||
|
|
// @BasePath /
|
|||
|
|
// @securityDefinitions.apikey BearerAuth
|
|||
|
|
// @in header
|
|||
|
|
// @name Authorization
|
|||
|
|
|
|||
|
|
func main() {
|
|||
|
|
// 从配置文件读取配置
|
|||
|
|
if err := conf.Init(); err != nil {
|
|||
|
|
logger.Error(nil, "配置初始化失败", zap.Error(err))
|
|||
|
|
os.Exit(1)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置 gin 模式
|
|||
|
|
ginMode := getGinMode()
|
|||
|
|
gin.SetMode(ginMode)
|
|||
|
|
logger.Info(nil, "Gin模式设置完成", zap.String("mode", ginMode))
|
|||
|
|
|
|||
|
|
// 创建路由
|
|||
|
|
engine := router.NewRouter()
|
|||
|
|
|
|||
|
|
// 获取服务端口
|
|||
|
|
port := getServerPort()
|
|||
|
|
|
|||
|
|
// 创建HTTP服务器
|
|||
|
|
server := &http.Server{
|
|||
|
|
Addr: port,
|
|||
|
|
Handler: engine,
|
|||
|
|
ReadTimeout: 15 * time.Second,
|
|||
|
|
WriteTimeout: 15 * time.Second,
|
|||
|
|
IdleTimeout: 60 * time.Second,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 在goroutine中启动服务器
|
|||
|
|
go func() {
|
|||
|
|
logger.Info(nil, "服务器启动中...", zap.String("addr", port))
|
|||
|
|
if err := server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
|
|||
|
|
logger.Error(nil, "服务器启动失败", zap.Error(err))
|
|||
|
|
os.Exit(1)
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
logger.Info(nil, "服务器启动成功", zap.String("addr", port))
|
|||
|
|
|
|||
|
|
// 等待中断信号来优雅地关闭服务器
|
|||
|
|
quit := make(chan os.Signal, 1)
|
|||
|
|
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
|||
|
|
<-quit
|
|||
|
|
|
|||
|
|
logger.Info(nil, "收到关闭信号，开始优雅关闭服务器...")
|
|||
|
|
|
|||
|
|
// 创建一个5秒超时的context
|
|||
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|||
|
|
defer cancel()
|
|||
|
|
|
|||
|
|
// 优雅关闭HTTP服务器
|
|||
|
|
if err := server.Shutdown(ctx); err != nil {
|
|||
|
|
logger.Error(nil, "服务器强制关闭", zap.Error(err))
|
|||
|
|
} else {
|
|||
|
|
logger.Info(nil, "HTTP服务器已优雅关闭")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 关闭其他资源
|
|||
|
|
conf.Close()
|
|||
|
|
|
|||
|
|
logger.Info(nil, "服务器关闭完成")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// getGinMode 获取Gin运行模式
|
|||
|
|
func getGinMode() string {
|
|||
|
|
mode := os.Getenv("GIN_MODE")
|
|||
|
|
if mode == "" {
|
|||
|
|
mode = gin.ReleaseMode // 默认为生产模式
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证模式是否有效
|
|||
|
|
switch mode {
|
|||
|
|
case gin.DebugMode, gin.ReleaseMode, gin.TestMode:
|
|||
|
|
return mode
|
|||
|
|
default:
|
|||
|
|
logger.Warn(nil, "无效的GIN_MODE，使用默认值",
|
|||
|
|
zap.String("invalid_mode", mode),
|
|||
|
|
zap.String("default_mode", gin.ReleaseMode))
|
|||
|
|
return gin.ReleaseMode
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// getServerPort 获取服务器端口
|
|||
|
|
func getServerPort() string {
|
|||
|
|
port := os.Getenv("SERVER_PORT")
|
|||
|
|
if port == "" {
|
|||
|
|
port = ":3000" // 默认端口
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 确保端口格式正确
|
|||
|
|
if port[0] != ':' {
|
|||
|
|
port = ":" + port
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return port
|
|||
|
|
}
|