refactor: refactor stable diffusion service, add service pool support

2025-11-06 17:23:42 +08:00 · 2023-12-14 16:48:54 +08:00
parent 10ba1430f9
commit d2991e60b6
14 changed files with 266 additions and 253 deletions
--- a/api/main.go
+++ b/api/main.go
@@ -167,14 +167,7 @@ func main() {
 		fx.Provide(mj.NewServicePool),

 		// Stable Diffusion 机器人
-		fx.Provide(sd.NewService),
-		fx.Invoke(func(config *types.AppConfig, service *sd.Service) {
-			if config.SdConfig.Enabled {
-				go func() {
-					service.Run()
-				}()
-			}
-		}),
+		fx.Provide(sd.NewServicePool),

 		fx.Provide(payment.NewAlipayService),
 		fx.Provide(payment.NewHuPiPay),