first commit
This commit is contained in:
@@ -0,0 +1,349 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"log/slog"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
"git.lzcwy.cn/zhengchiliu/zt-closet-backend/global"
|
||||
"git.lzcwy.cn/zhengchiliu/zt-closet-backend/internal/mysqlmodel"
|
||||
"git.lzcwy.cn/zhengchiliu/zt-closet-backend/internal/redismodel"
|
||||
"git.lzcwy.cn/zhengchiliu/zt-closet-backend/internal/routers"
|
||||
"git.lzcwy.cn/zhengchiliu/zt-closet-backend/internal/service"
|
||||
|
||||
"git.lzcwy.cn/zhengchiliu/zt-closet-backend/pkg/cos"
|
||||
"git.lzcwy.cn/zhengchiliu/zt-closet-backend/pkg/setting"
|
||||
"git.lzcwy.cn/zhengchiliu/zt-closet-backend/pkg/tokenhub"
|
||||
"git.lzcwy.cn/zhengchiliu/zt-closet-backend/pkg/tracing"
|
||||
"git.lzcwy.cn/zhengchiliu/zt-closet-backend/pkg/wechat"
|
||||
"github.com/gin-gonic/gin"
|
||||
"go.opentelemetry.io/contrib/bridges/otelslog"
|
||||
"go.opentelemetry.io/otel"
|
||||
"gopkg.in/natefinch/lumberjack.v2"
|
||||
)
|
||||
|
||||
func init() {
|
||||
if err := setupSetting(); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
// OTel 需在 logger 之前初始化，使 LoggerProvider 就绪后 slog 可桥接到 OTel Logs。
|
||||
setupOTel()
|
||||
|
||||
if err := setupSlogLogger(); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
if err := setupMysqlEngine(); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
if err := setupRedisEngine(); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
setupCosClient()
|
||||
setupWechatPayClient()
|
||||
setupVirtualPayClient()
|
||||
}
|
||||
|
||||
func main() {
|
||||
gin.SetMode(global.ServerSetting.RunMode)
|
||||
|
||||
// 启动运势后台任务：每日 0 点生成 + 启动兜底。
|
||||
tokenhub.EnsureTodayFortunes()
|
||||
tokenhub.StartFortuneScheduler()
|
||||
|
||||
// 启动权益发放补偿与订单超时过期后台任务。
|
||||
service.StartBenefitCompensationScheduler()
|
||||
service.StartOrderExpireScheduler()
|
||||
|
||||
router := routers.NewRouter()
|
||||
|
||||
s := &http.Server{
|
||||
Addr: ":" + global.ServerSetting.HttpPort,
|
||||
Handler: router,
|
||||
ReadTimeout: time.Duration(global.ServerSetting.ReadTimeout) * time.Second,
|
||||
WriteTimeout: time.Duration(global.ServerSetting.WriteTimeout) * time.Second,
|
||||
|
||||
// 限制http请求头的大小为1MB
|
||||
MaxHeaderBytes: 1 << 20,
|
||||
}
|
||||
|
||||
// 后台启动 HTTP 服务；启动失败（如端口占用）通过 channel 反馈主流程，避免进程「假存活」。
|
||||
serverErr := make(chan error, 1)
|
||||
go func() {
|
||||
if err := s.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
|
||||
serverErr <- err
|
||||
}
|
||||
}()
|
||||
slog.Info("server started", "port", global.ServerSetting.HttpPort)
|
||||
|
||||
// 等待退出信号或服务启动错误。
|
||||
quit := make(chan os.Signal, 1)
|
||||
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
||||
|
||||
select {
|
||||
case <-quit:
|
||||
slog.Info("shutting down server...")
|
||||
case err := <-serverErr:
|
||||
slog.Error("http server start failed", "error", err)
|
||||
}
|
||||
|
||||
// 优雅关闭：HTTP 与 OTel 使用独立 context，避免 HTTP 等待耗尽 OTel flush 的超时。
|
||||
shutdownCtx, shutdownCancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer shutdownCancel()
|
||||
if err := s.Shutdown(shutdownCtx); err != nil {
|
||||
slog.Error("http server shutdown error", "error", err)
|
||||
}
|
||||
|
||||
if global.OTelShutdown != nil {
|
||||
otelCtx, otelCancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer otelCancel()
|
||||
if err := global.OTelShutdown(otelCtx); err != nil {
|
||||
slog.Error("otel shutdown error", "error", err)
|
||||
}
|
||||
}
|
||||
slog.Info("server exited")
|
||||
}
|
||||
|
||||
func setupSetting() error {
|
||||
setting, err := setting.NewSetting()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = setting.ReadSection("Server", &global.ServerSetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = setting.ReadSection("Log", &global.LogSetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = setting.ReadSection("Mysql", &global.MysqlSetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = setting.ReadSection("Redis", &global.RedisSetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = setting.ReadSection("Otel", &global.OtelSetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = setting.ReadSection("JWT", &global.JWTSetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
// 校验 JWT 关键配置，防止密钥缺失导致可用空密钥伪造 token。
|
||||
if global.JWTSetting == nil || global.JWTSetting.SecretKey == "" {
|
||||
return fmt.Errorf("JWT secret key must not be empty")
|
||||
}
|
||||
|
||||
err = setting.ReadSection("Cos", &global.CosSetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = setting.ReadSection("Upload", &global.UploadSetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if global.UploadSetting.MaxSizeMB <= 0 {
|
||||
global.UploadSetting.MaxSizeMB = 20
|
||||
}
|
||||
|
||||
err = setting.ReadSection("Tokenhub", &global.TokenhubSetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if global.TokenhubSetting.Model == "" {
|
||||
global.TokenhubSetting.Model = "hy4"
|
||||
}
|
||||
|
||||
err = setting.ReadSection("MattingModel", &global.MattingModelSetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if global.MattingModelSetting.MaxRetry <= 0 {
|
||||
global.MattingModelSetting.MaxRetry = 3
|
||||
}
|
||||
|
||||
err = setting.ReadSection("ClassificationModel", &global.ClassificationModelSetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if global.ClassificationModelSetting.MaxRetry <= 0 {
|
||||
global.ClassificationModelSetting.MaxRetry = 3
|
||||
}
|
||||
|
||||
err = setting.ReadSection("WechatMini", &global.WechatMiniSetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = setting.ReadSection("WechatPay", &global.WechatPaySetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = setting.ReadSection("UserDefaultQuota", &global.UserDefaultQuotaSetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = setting.ReadSection("AdReward", &global.AdRewardSetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = setting.ReadSection("WechatVirtualPay", &global.VirtualPaySetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
|
||||
// 校验 Server 配置（时间字段已按秒解析，此处仅做存在性校验）。
|
||||
if global.ServerSetting == nil {
|
||||
return fmt.Errorf("Server config section is missing")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func setupSlogLogger() error {
|
||||
// 确保日志目录存在（lumberjack 不会自动创建父目录，目录缺失时文件日志会静默失效）。
|
||||
if err := os.MkdirAll(global.LogSetting.LogSavePath, 0o755); err != nil {
|
||||
return fmt.Errorf("create log dir: %w", err)
|
||||
}
|
||||
|
||||
writer := &lumberjack.Logger{
|
||||
Filename: global.LogSetting.LogSavePath + "/" + global.LogSetting.LogFileName + global.LogSetting.LogFileExt,
|
||||
MaxSize: global.LogSetting.MaxSize,
|
||||
MaxBackups: global.LogSetting.MaxBackups,
|
||||
MaxAge: global.LogSetting.MaxAge,
|
||||
Compress: true,
|
||||
}
|
||||
|
||||
multiWriter := io.MultiWriter(writer, os.Stdout)
|
||||
|
||||
level := slog.LevelInfo
|
||||
switch global.LogSetting.LogLevel {
|
||||
case "Debug":
|
||||
level = slog.LevelDebug
|
||||
case "Warn":
|
||||
level = slog.LevelWarn
|
||||
case "Error":
|
||||
level = slog.LevelError
|
||||
default:
|
||||
level = slog.LevelInfo
|
||||
}
|
||||
|
||||
// 本地 JSON 日志（文件 + stdout），并注入 trace_id/span_id。
|
||||
localHandler := tracing.OtelContextHandler{
|
||||
Handler: slog.NewJSONHandler(multiWriter, &slog.HandlerOptions{
|
||||
AddSource: true,
|
||||
Level: level,
|
||||
}),
|
||||
}
|
||||
|
||||
// OTel 日志桥接：将 slog 记录导出到 OTel Logs（LoggerProvider 已在 setupOTel 就绪；
|
||||
// 未启用 OTel 时为 no-op，不产生开销）。
|
||||
otelHandler := otelslog.NewHandler("zt-closet-backend")
|
||||
|
||||
// 扇出：本地日志 + OTel 日志同时输出。
|
||||
slog.SetDefault(slog.New(tracing.NewFanoutHandler(localHandler, otelHandler)))
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// setupMysqlEngine 初始化 MySQL 数据库连接。
|
||||
func setupMysqlEngine() error {
|
||||
var err error
|
||||
global.MysqlEngine, err = mysqlmodel.NewMysqlEngine(global.MysqlSetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// setupRedisEngine 初始化 Redis 连接。
|
||||
func setupRedisEngine() error {
|
||||
var err error
|
||||
global.RedisEngine, err = redismodel.NewRedisEngine(global.RedisSetting)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func setupOTel() {
|
||||
tracer, shutdown, err := tracing.NewOTelProvider(global.OtelSetting)
|
||||
// NewOTelProvider 在任何错误路径都会返回可用的 no-op tracer 与安全 shutdown。
|
||||
global.Tracer = tracer
|
||||
global.OTelShutdown = shutdown
|
||||
if err != nil {
|
||||
// 此时 logger 尚未就绪，用标准输出告警；已降级为 no-op。
|
||||
fmt.Printf("[WARN] init otel failed, downgraded to no-op: %v\n", err)
|
||||
}
|
||||
|
||||
name := "zt-closet-backend"
|
||||
if global.OtelSetting != nil && global.OtelSetting.ServiceName != "" {
|
||||
name = global.OtelSetting.ServiceName
|
||||
}
|
||||
global.Meter = otel.Meter(name)
|
||||
}
|
||||
|
||||
// setupCosClient 初始化腾讯云 COS 客户端（衣物图片上传依赖）。
|
||||
// 初始化失败仅告警，不阻断服务启动。
|
||||
func setupCosClient() {
|
||||
client, err := cos.NewCosClient()
|
||||
if err != nil {
|
||||
slog.Warn("init cos client failed, upload features may fail", "error", err)
|
||||
return
|
||||
}
|
||||
global.CosClient = client
|
||||
}
|
||||
|
||||
// setupWechatPayClient 初始化微信支付客户端。
|
||||
// 未启用（Enable=false）或未配置时仅告警，不阻断服务启动（便于本地/未开通支付的开发环境）。
|
||||
func setupWechatPayClient() {
|
||||
if global.WechatPaySetting == nil || !global.WechatPaySetting.Enable {
|
||||
slog.Warn("wechat pay not enabled, pay features disabled")
|
||||
return
|
||||
}
|
||||
if err := wechat.InitPayClient(global.WechatPaySetting); err != nil {
|
||||
slog.Warn("init wechat pay client failed, pay features disabled", "error", err)
|
||||
return
|
||||
}
|
||||
slog.Info("wechat pay client initialized")
|
||||
}
|
||||
|
||||
// setupVirtualPayClient 初始化微信小程序虚拟支付（XPay）客户端。
|
||||
// 未开通（Enable=false）或配置不完整时仅告警，不阻断服务启动；虚拟商品下单时会返回「虚拟支付未开通」。
|
||||
func setupVirtualPayClient() {
|
||||
if global.VirtualPaySetting == nil || !global.VirtualPaySetting.Enable {
|
||||
slog.Warn("virtual pay not enabled, virtual goods purchase disabled")
|
||||
return
|
||||
}
|
||||
if err := wechat.InitVirtualPayClient(global.VirtualPaySetting); err != nil {
|
||||
slog.Warn("init virtual pay client failed, virtual goods purchase disabled", "error", err)
|
||||
return
|
||||
}
|
||||
env, _ := wechat.GetVirtualPayEnv()
|
||||
slog.Info("virtual pay client initialized", "env", env, "env_name", map[int]string{0: "prod", 1: "sandbox"}[env])
|
||||
}
|
||||
Reference in New Issue
Block a user