refactor(backend): 引入 Wire 重构服务启动与依赖组装

2025-12-18 21:16:00 +08:00
parent 9b4fc42457
commit e5aa676853
7 changed files with 279 additions and 61 deletions
--- a/backend/cmd/server/main.go
+++ b/backend/cmd/server/main.go
@@ -1,8 +1,11 @@
 package main

+//go:generate go run github.com/google/wire/cmd/wire
+
 import (
 	"context"
 	_ "embed"
+	"errors"
 	"flag"
 	"log"
 	"net/http"
@@ -110,78 +113,25 @@ func runSetupServer() {
 }

 func runMainServer() {
-	// 加载配置
-	cfg, err := config.Load()
-	if err != nil {
-		log.Fatalf("Failed to load config: %v", err)
-	}
-
-	// 初始化时区（类似 PHP 的 date_default_timezone_set）
-	if err := timezone.Init(cfg.Timezone); err != nil {
-		log.Fatalf("Failed to initialize timezone: %v", err)
-	}
-
-	// 初始化数据库
-	db, err := initDB(cfg)
-	if err != nil {
-		log.Fatalf("Failed to connect to database: %v", err)
-	}
-
-	// 初始化Redis
-	rdb := initRedis(cfg)
-
-	// 初始化Repository
-	repos := repository.NewRepositories(db)
-
-	// 初始化Service
-	services := service.NewServices(repos, rdb, cfg)
-
-	// 初始化Handler
 	buildInfo := handler.BuildInfo{
 		Version:   Version,
 		BuildType: BuildType,
 	}
-	handlers := handler.NewHandlers(services, repos, rdb, buildInfo)

-	// 设置Gin模式
-	if cfg.Server.Mode == "release" {
-		gin.SetMode(gin.ReleaseMode)
-	}
-
-	// 创建路由
-	r := gin.New()
-	r.Use(gin.Recovery())
-	r.Use(middleware.Logger())
-	r.Use(middleware.CORS())
-
-	// 注册路由
-	registerRoutes(r, handlers, services, repos)
-
-	// Serve embedded frontend if available
-	if web.HasEmbeddedFrontend() {
-		r.Use(web.ServeEmbeddedFrontend())
+	app, err := initializeApplication(buildInfo)
+	if err != nil {
+		log.Fatalf("Failed to initialize application: %v", err)
 	}
+	defer app.Cleanup()

 	// 启动服务器
-	srv := &http.Server{
-		Addr:    cfg.Server.Address(),
-		Handler: r,
-		// ReadHeaderTimeout: 读取请求头的超时时间，防止慢速请求头攻击
-		ReadHeaderTimeout: time.Duration(cfg.Server.ReadHeaderTimeout) * time.Second,
-		// IdleTimeout: 空闲连接超时时间，释放不活跃的连接资源
-		IdleTimeout: time.Duration(cfg.Server.IdleTimeout) * time.Second,
-		// 注意：不设置 WriteTimeout，因为流式响应可能持续十几分钟
-		// 不设置 ReadTimeout，因为大请求体可能需要较长时间读取
-	}
-
-	// 优雅关闭
 	go func() {
-		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+		if err := app.Server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
 			log.Fatalf("Failed to start server: %v", err)
 		}
 	}()

-	log.Printf("Server started on %s", cfg.Server.Address())
+	log.Printf("Server started on %s", app.Server.Addr)

 	// 等待中断信号
 	quit := make(chan os.Signal, 1)
@@ -193,7 +143,7 @@ func runMainServer() {
 	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
 	defer cancel()

-	if err := srv.Shutdown(ctx); err != nil {
+	if err := app.Server.Shutdown(ctx); err != nil {
 		log.Fatalf("Server forced to shutdown: %v", err)
 	}

@@ -201,6 +151,11 @@ func runMainServer() {
 }

 func initDB(cfg *config.Config) (*gorm.DB, error) {
+	// 初始化时区（在数据库连接之前，确保时区设置正确）
+	if err := timezone.Init(cfg.Timezone); err != nil {
+		return nil, err
+	}
+
 	gormConfig := &gorm.Config{}
 	if cfg.Server.Mode == "debug" {
 		gormConfig.Logger = logger.Default.LogMode(logger.Info)
@@ -479,3 +434,34 @@ func registerRoutes(r *gin.Engine, h *handler.Handlers, s *service.Services, rep
 		gateway.GET("/usage", h.Gateway.Usage)
 	}
 }
+
+// setupRouter 配置路由器中间件和路由
+func setupRouter(r *gin.Engine, cfg *config.Config, handlers *handler.Handlers, services *service.Services, repos *repository.Repositories) *gin.Engine {
+	// 应用中间件
+	r.Use(middleware.Logger())
+	r.Use(middleware.CORS())
+
+	// 注册路由
+	registerRoutes(r, handlers, services, repos)
+
+	// Serve embedded frontend if available
+	if web.HasEmbeddedFrontend() {
+		r.Use(web.ServeEmbeddedFrontend())
+	}
+
+	return r
+}
+
+// createHTTPServer 创建HTTP服务器
+func createHTTPServer(cfg *config.Config, router *gin.Engine) *http.Server {
+	return &http.Server{
+		Addr:    cfg.Server.Address(),
+		Handler: router,
+		// ReadHeaderTimeout: 读取请求头的超时时间，防止慢速请求头攻击
+		ReadHeaderTimeout: time.Duration(cfg.Server.ReadHeaderTimeout) * time.Second,
+		// IdleTimeout: 空闲连接超时时间，释放不活跃的连接资源
+		IdleTimeout: time.Duration(cfg.Server.IdleTimeout) * time.Second,
+		// 注意：不设置 WriteTimeout，因为流式响应可能持续十几分钟
+		// 不设置 ReadTimeout，因为大请求体可能需要较长时间读取
+	}
+}
--- a/backend/cmd/server/wire.go
+++ b/backend/cmd/server/wire.go
@@ -0,0 +1,103 @@
+//go:build wireinject
+// +build wireinject
+
+package main
+
+import (
+	"sub2api/internal/config"
+	"sub2api/internal/handler"
+	"sub2api/internal/repository"
+	"sub2api/internal/service"
+
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+	"github.com/google/wire"
+	"github.com/redis/go-redis/v9"
+	"gorm.io/gorm"
+)
+
+type Application struct {
+	Server  *http.Server
+	Cleanup func()
+}
+
+func initializeApplication(buildInfo handler.BuildInfo) (*Application, error) {
+	wire.Build(
+		// Config provider
+		provideConfig,
+
+		// Database provider
+		provideDB,
+
+		// Redis provider
+		provideRedis,
+
+		// Repository provider
+		provideRepositories,
+
+		// Service provider
+		provideServices,
+
+		// Handler provider
+		provideHandlers,
+
+		// Router provider
+		provideRouter,
+
+		// HTTP Server provider
+		provideHTTPServer,
+
+		// Cleanup provider
+		provideCleanup,
+
+		// Application provider
+		wire.Struct(new(Application), "Server", "Cleanup"),
+	)
+	return nil, nil
+}
+
+func provideConfig() (*config.Config, error) {
+	return config.Load()
+}
+
+func provideDB(cfg *config.Config) (*gorm.DB, error) {
+	return initDB(cfg)
+}
+
+func provideRedis(cfg *config.Config) *redis.Client {
+	return initRedis(cfg)
+}
+
+func provideRepositories(db *gorm.DB) *repository.Repositories {
+	return repository.NewRepositories(db)
+}
+
+func provideServices(repos *repository.Repositories, rdb *redis.Client, cfg *config.Config) *service.Services {
+	return service.NewServices(repos, rdb, cfg)
+}
+
+func provideHandlers(services *service.Services, repos *repository.Repositories, rdb *redis.Client, buildInfo handler.BuildInfo) *handler.Handlers {
+	return handler.NewHandlers(services, repos, rdb, buildInfo)
+}
+
+func provideRouter(cfg *config.Config, handlers *handler.Handlers, services *service.Services, repos *repository.Repositories) *gin.Engine {
+	if cfg.Server.Mode == "release" {
+		gin.SetMode(gin.ReleaseMode)
+	}
+
+	r := gin.New()
+	r.Use(gin.Recovery())
+
+	return setupRouter(r, cfg, handlers, services, repos)
+}
+
+func provideHTTPServer(cfg *config.Config, router *gin.Engine) *http.Server {
+	return createHTTPServer(cfg, router)
+}
+
+func provideCleanup() func() {
+	return func() {
+		//	@todo
+	}
+}
--- a/backend/cmd/server/wire_gen.go
+++ b/backend/cmd/server/wire_gen.go
@@ -0,0 +1,99 @@
+// Code generated by Wire. DO NOT EDIT.
+
+//go:generate go run -mod=mod github.com/google/wire/cmd/wire
+//go:build !wireinject
+// +build !wireinject
+
+package main
+
+import (
+	"github.com/gin-gonic/gin"
+	"github.com/redis/go-redis/v9"
+	"gorm.io/gorm"
+	"net/http"
+	"sub2api/internal/config"
+	"sub2api/internal/handler"
+	"sub2api/internal/repository"
+	"sub2api/internal/service"
+)
+
+import (
+	_ "embed"
+)
+
+// Injectors from wire.go:
+
+func initializeApplication(buildInfo handler.BuildInfo) (*Application, error) {
+	config, err := provideConfig()
+	if err != nil {
+		return nil, err
+	}
+	db, err := provideDB(config)
+	if err != nil {
+		return nil, err
+	}
+	repositories := provideRepositories(db)
+	client := provideRedis(config)
+	services := provideServices(repositories, client, config)
+	handlers := provideHandlers(services, repositories, client, buildInfo)
+	engine := provideRouter(config, handlers, services, repositories)
+	server := provideHTTPServer(config, engine)
+	v := provideCleanup()
+	application := &Application{
+		Server:  server,
+		Cleanup: v,
+	}
+	return application, nil
+}
+
+// wire.go:
+
+type Application struct {
+	Server  *http.Server
+	Cleanup func()
+}
+
+func provideConfig() (*config.Config, error) {
+	return config.Load()
+}
+
+func provideDB(cfg *config.Config) (*gorm.DB, error) {
+	return initDB(cfg)
+}
+
+func provideRedis(cfg *config.Config) *redis.Client {
+	return initRedis(cfg)
+}
+
+func provideRepositories(db *gorm.DB) *repository.Repositories {
+	return repository.NewRepositories(db)
+}
+
+func provideServices(repos *repository.Repositories, rdb *redis.Client, cfg *config.Config) *service.Services {
+	return service.NewServices(repos, rdb, cfg)
+}
+
+func provideHandlers(services *service.Services, repos *repository.Repositories, rdb *redis.Client, buildInfo handler.BuildInfo) *handler.Handlers {
+	return handler.NewHandlers(services, repos, rdb, buildInfo)
+}
+
+func provideRouter(cfg *config.Config, handlers *handler.Handlers, services *service.Services, repos *repository.Repositories) *gin.Engine {
+	if cfg.Server.Mode == "release" {
+		gin.SetMode(gin.ReleaseMode)
+	}
+
+	r := gin.New()
+	r.Use(gin.Recovery())
+
+	return setupRouter(r, cfg, handlers, services, repos)
+}
+
+func provideHTTPServer(cfg *config.Config, router *gin.Engine) *http.Server {
+	return createHTTPServer(cfg, router)
+}
+
+func provideCleanup() func() {
+	return func() {
+
+	}
+}