refactor: add midjourney pool implementation, add translate prompt for mj drawing

api/service/mj/pool.go

@@ -4,35 +4,63 @@ import (
 	"chatplus/core/types"
 	"chatplus/service/oss"
 	"chatplus/store"
+	"fmt"
 	"github.com/go-redis/redis/v8"
 	"gorm.io/gorm"
 )
 
 // ServicePool Mj service pool
 type ServicePool struct {
-	services  []Service
+	services  []*Service
 	taskQueue *store.RedisQueue
 }
 
 func NewServicePool(db *gorm.DB, redisCli *redis.Client, manager *oss.UploaderManager, appConfig *types.AppConfig) *ServicePool {
+	services := make([]*Service, 0)
+	queue := store.NewRedisQueue("MidJourney_Task_Queue", redisCli)
 	// create mj client and service
-	for _, config := range appConfig.MjConfigs {
+	for k, config := range appConfig.MjConfigs {
 		if config.Enabled == false {
 			continue
 		}
 		// create mj client
 		client := NewClient(&config, appConfig.ProxyURL)
 
+		name := fmt.Sprintf("MjService-%d", k)
 		// create mj service
-		service := NewService()
+		service := NewService(name, queue, 4, 600, db, client, manager, appConfig)
+		botName := fmt.Sprintf("MjBot-%d", k)
+		bot, err := NewBot(botName, appConfig.ProxyURL, &config, service)
+		if err != nil {
+			continue
+		}
+
+		err = bot.Run()
+		if err != nil {
+			continue
+		}
+
+		// run mj service
+		go func() {
+			service.Run()
+		}()
+
+		services = append(services, service)
 	}
 
 	return &ServicePool{
-		taskQueue: store.NewRedisQueue("MidJourney_Task_Queue", redisCli),
+		taskQueue: queue,
+		services:  services,
 	}
 }
 
+// PushTask push a new mj task in to task queue
 func (p *ServicePool) PushTask(task types.MjTask) {
 	logger.Debugf("add a new MidJourney task to the task list: %+v", task)
 	p.taskQueue.RPush(task)
 }
+
+// HasAvailableService check if has available mj service in pool
+func (p *ServicePool) HasAvailableService() bool {
+	return len(p.services) > 0
+}