126 lines
2.5 KiB
Go
126 lines
2.5 KiB
Go
//go:build wireinject
|
|
// +build wireinject
|
|
|
|
package main
|
|
|
|
import (
|
|
"sub2api/internal/config"
|
|
"sub2api/internal/handler"
|
|
"sub2api/internal/infrastructure"
|
|
"sub2api/internal/repository"
|
|
"sub2api/internal/server"
|
|
"sub2api/internal/service"
|
|
|
|
"context"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"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(
|
|
// 基础设施层 ProviderSets
|
|
config.ProviderSet,
|
|
infrastructure.ProviderSet,
|
|
|
|
// 业务层 ProviderSets
|
|
repository.ProviderSet,
|
|
service.ProviderSet,
|
|
handler.ProviderSet,
|
|
|
|
// 服务器层 ProviderSet
|
|
server.ProviderSet,
|
|
|
|
// BuildInfo provider
|
|
provideServiceBuildInfo,
|
|
|
|
// 清理函数提供者
|
|
provideCleanup,
|
|
|
|
// 应用程序结构体
|
|
wire.Struct(new(Application), "Server", "Cleanup"),
|
|
)
|
|
return nil, nil
|
|
}
|
|
|
|
func provideServiceBuildInfo(buildInfo handler.BuildInfo) service.BuildInfo {
|
|
return service.BuildInfo{
|
|
Version: buildInfo.Version,
|
|
BuildType: buildInfo.BuildType,
|
|
}
|
|
}
|
|
|
|
func provideCleanup(
|
|
db *gorm.DB,
|
|
rdb *redis.Client,
|
|
services *service.Services,
|
|
) func() {
|
|
return func() {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
|
|
// Cleanup steps in reverse dependency order
|
|
cleanupSteps := []struct {
|
|
name string
|
|
fn func() error
|
|
}{
|
|
{"TokenRefreshService", func() error {
|
|
services.TokenRefresh.Stop()
|
|
return nil
|
|
}},
|
|
{"PricingService", func() error {
|
|
services.Pricing.Stop()
|
|
return nil
|
|
}},
|
|
{"EmailQueueService", func() error {
|
|
services.EmailQueue.Stop()
|
|
return nil
|
|
}},
|
|
{"OAuthService", func() error {
|
|
services.OAuth.Stop()
|
|
return nil
|
|
}},
|
|
{"OpenAIOAuthService", func() error {
|
|
services.OpenAIOAuth.Stop()
|
|
return nil
|
|
}},
|
|
{"Redis", func() error {
|
|
return rdb.Close()
|
|
}},
|
|
{"Database", func() error {
|
|
sqlDB, err := db.DB()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return sqlDB.Close()
|
|
}},
|
|
}
|
|
|
|
for _, step := range cleanupSteps {
|
|
if err := step.fn(); err != nil {
|
|
log.Printf("[Cleanup] %s failed: %v", step.name, err)
|
|
// Continue with remaining cleanup steps even if one fails
|
|
} else {
|
|
log.Printf("[Cleanup] %s succeeded", step.name)
|
|
}
|
|
}
|
|
|
|
// Check if context timed out
|
|
select {
|
|
case <-ctx.Done():
|
|
log.Printf("[Cleanup] Warning: cleanup timed out after 10 seconds")
|
|
default:
|
|
log.Printf("[Cleanup] All cleanup steps completed")
|
|
}
|
|
}
|
|
}
|