fork-it/gpt-load
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 并发池功能

Browse Source
This commit is contained in:
tbphp
2025-07-06 02:00:21 +08:00
parent e6fe973ea4
commit 4dc17c12d5
9 changed files with 298 additions and 137 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
cmd/gpt-load/main.go
View File

@@ -46,7 +46,7 @@ func main() {
setupLogger(configManager)
// Initialize database
database, err := db.InitDB()
database, err := db.InitDB(configManager)
if err != nil {
logrus.Fatalf("Failed to initialize database: %v", err)
}
@@ -88,8 +88,14 @@ func main() {
channelFactory := channel.NewFactory(settingsManager)
keyValidatorService := services.NewKeyValidatorService(database, channelFactory, settingsManager)
keyManualValidationService := services.NewKeyManualValidationService(database, keyValidatorService, taskService, settingsManager)
keyCronService := services.NewKeyCronService(database, keyValidatorService, settingsManager)
// --- Global Key Validation Pool ---
KeyValidationPool := services.NewKeyValidationPool(keyValidatorService, configManager)
KeyValidationPool.Start()
defer KeyValidationPool.Stop()
// ---
keyManualValidationService := services.NewKeyManualValidationService(database, keyValidatorService, taskService, settingsManager, configManager)
keyCronService := services.NewKeyCronService(database, settingsManager, KeyValidationPool)
keyCronService.Start()
defer keyCronService.Stop()