cbfce49aa1
Implement deferred batch update mechanism to reduce database load:
- Add DeferredService for batching account last_used_at updates
- Add TimingWheelService for efficient recurring task scheduling
- Integrate with GatewayService and OpenAIGatewayService
- Implement BatchUpdateLastUsed repository method using CASE...WHEN SQL
- Fix golangci-lint error: Replace interface{} with any
Benefits:
- Reduces database writes by batching updates (10-second intervals)
- Improves request throughput by deferring non-critical updates
- Maintains accurate account usage tracking for scheduling
102 lines
2.8 KiB
Go
102 lines
2.8 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 {
|
|
// 价格服务初始化失败不应阻止启动,使用回退价格
|
|
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,
|
|
cfg *config.Config,
|
|
) *TokenRefreshService {
|
|
svc := NewTokenRefreshService(accountRepo, oauthService, openaiOAuthService, geminiOAuthService, cfg)
|
|
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
|
|
}
|
|
|
|
// ProviderSet is the Wire provider set for all services
|
|
var ProviderSet = wire.NewSet(
|
|
// Core services
|
|
NewAuthService,
|
|
NewUserService,
|
|
NewApiKeyService,
|
|
NewGroupService,
|
|
NewAccountService,
|
|
NewProxyService,
|
|
NewRedeemService,
|
|
NewUsageService,
|
|
NewDashboardService,
|
|
ProvidePricingService,
|
|
NewBillingService,
|
|
NewBillingCacheService,
|
|
NewAdminService,
|
|
NewGatewayService,
|
|
NewOpenAIGatewayService,
|
|
NewOAuthService,
|
|
NewOpenAIOAuthService,
|
|
NewGeminiOAuthService,
|
|
NewGeminiTokenProvider,
|
|
NewGeminiMessagesCompatService,
|
|
NewRateLimitService,
|
|
NewAccountUsageService,
|
|
NewAccountTestService,
|
|
NewSettingService,
|
|
NewEmailService,
|
|
ProvideEmailQueueService,
|
|
NewTurnstileService,
|
|
NewSubscriptionService,
|
|
NewConcurrencyService,
|
|
NewIdentityService,
|
|
NewCRSSyncService,
|
|
ProvideUpdateService,
|
|
ProvideTokenRefreshService,
|
|
ProvideTimingWheelService,
|
|
ProvideDeferredService,
|
|
)
|