142 lines
4.3 KiB
Go
142 lines
4.3 KiB
Go
package service
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/config"
|
|
"github.com/google/wire"
|
|
)
|
|
|
|
// BuildInfo contains build information
|
|
type BuildInfo struct {
|
|
Version string
|
|
BuildType string
|
|
}
|
|
|
|
// ProvidePricingService creates and initializes PricingService
|
|
func ProvidePricingService(cfg *config.Config, remoteClient PricingRemoteClient) (*PricingService, error) {
|
|
svc := NewPricingService(cfg, remoteClient)
|
|
if err := svc.Initialize(); err != nil {
|
|
// Pricing service initialization failure should not block startup, use fallback prices
|
|
println("[Service] Warning: Pricing service initialization failed:", err.Error())
|
|
}
|
|
return svc, nil
|
|
}
|
|
|
|
// ProvideUpdateService creates UpdateService with BuildInfo
|
|
func ProvideUpdateService(cache UpdateCache, githubClient GitHubReleaseClient, buildInfo BuildInfo) *UpdateService {
|
|
return NewUpdateService(cache, githubClient, buildInfo.Version, buildInfo.BuildType)
|
|
}
|
|
|
|
// ProvideEmailQueueService creates EmailQueueService with default worker count
|
|
func ProvideEmailQueueService(emailService *EmailService) *EmailQueueService {
|
|
return NewEmailQueueService(emailService, 3)
|
|
}
|
|
|
|
// ProvideTokenRefreshService creates and starts TokenRefreshService
|
|
func ProvideTokenRefreshService(
|
|
accountRepo AccountRepository,
|
|
oauthService *OAuthService,
|
|
openaiOAuthService *OpenAIOAuthService,
|
|
geminiOAuthService *GeminiOAuthService,
|
|
antigravityOAuthService *AntigravityOAuthService,
|
|
cfg *config.Config,
|
|
) *TokenRefreshService {
|
|
svc := NewTokenRefreshService(accountRepo, oauthService, openaiOAuthService, geminiOAuthService, antigravityOAuthService, cfg)
|
|
svc.Start()
|
|
return svc
|
|
}
|
|
|
|
// ProvideDashboardAggregationService 创建并启动仪表盘聚合服务
|
|
func ProvideDashboardAggregationService(repo DashboardAggregationRepository, timingWheel *TimingWheelService, cfg *config.Config) *DashboardAggregationService {
|
|
svc := NewDashboardAggregationService(repo, timingWheel, cfg)
|
|
svc.Start()
|
|
return svc
|
|
}
|
|
|
|
// ProvideAccountExpiryService creates and starts AccountExpiryService.
|
|
func ProvideAccountExpiryService(accountRepo AccountRepository) *AccountExpiryService {
|
|
svc := NewAccountExpiryService(accountRepo, time.Minute)
|
|
svc.Start()
|
|
return svc
|
|
}
|
|
|
|
// ProvideTimingWheelService creates and starts TimingWheelService
|
|
func ProvideTimingWheelService() *TimingWheelService {
|
|
svc := NewTimingWheelService()
|
|
svc.Start()
|
|
return svc
|
|
}
|
|
|
|
// ProvideDeferredService creates and starts DeferredService
|
|
func ProvideDeferredService(accountRepo AccountRepository, timingWheel *TimingWheelService) *DeferredService {
|
|
svc := NewDeferredService(accountRepo, timingWheel, 10*time.Second)
|
|
svc.Start()
|
|
return svc
|
|
}
|
|
|
|
// ProvideConcurrencyService creates ConcurrencyService and starts slot cleanup worker.
|
|
func ProvideConcurrencyService(cache ConcurrencyCache, accountRepo AccountRepository, cfg *config.Config) *ConcurrencyService {
|
|
svc := NewConcurrencyService(cache)
|
|
if cfg != nil {
|
|
svc.StartSlotCleanupWorker(accountRepo, cfg.Gateway.Scheduling.SlotCleanupInterval)
|
|
}
|
|
return svc
|
|
}
|
|
|
|
// ProvideAPIKeyAuthCacheInvalidator 提供 API Key 认证缓存失效能力
|
|
func ProvideAPIKeyAuthCacheInvalidator(apiKeyService *APIKeyService) APIKeyAuthCacheInvalidator {
|
|
return apiKeyService
|
|
}
|
|
|
|
// ProviderSet is the Wire provider set for all services
|
|
var ProviderSet = wire.NewSet(
|
|
// Core services
|
|
NewAuthService,
|
|
NewUserService,
|
|
NewAPIKeyService,
|
|
ProvideAPIKeyAuthCacheInvalidator,
|
|
NewGroupService,
|
|
NewAccountService,
|
|
NewProxyService,
|
|
NewRedeemService,
|
|
NewPromoService,
|
|
NewUsageService,
|
|
NewDashboardService,
|
|
ProvidePricingService,
|
|
NewBillingService,
|
|
NewBillingCacheService,
|
|
NewAdminService,
|
|
NewGatewayService,
|
|
NewOpenAIGatewayService,
|
|
NewOAuthService,
|
|
NewOpenAIOAuthService,
|
|
NewGeminiOAuthService,
|
|
NewGeminiQuotaService,
|
|
NewAntigravityOAuthService,
|
|
NewGeminiTokenProvider,
|
|
NewGeminiMessagesCompatService,
|
|
NewAntigravityTokenProvider,
|
|
NewAntigravityGatewayService,
|
|
NewRateLimitService,
|
|
NewAccountUsageService,
|
|
NewAccountTestService,
|
|
NewSettingService,
|
|
NewEmailService,
|
|
ProvideEmailQueueService,
|
|
NewTurnstileService,
|
|
NewSubscriptionService,
|
|
ProvideConcurrencyService,
|
|
NewIdentityService,
|
|
NewCRSSyncService,
|
|
ProvideUpdateService,
|
|
ProvideTokenRefreshService,
|
|
ProvideAccountExpiryService,
|
|
ProvideTimingWheelService,
|
|
ProvideDashboardAggregationService,
|
|
ProvideDeferredService,
|
|
NewAntigravityQuotaFetcher,
|
|
NewUserAttributeService,
|
|
NewUsageCache,
|
|
)
|