fix: fixed bug for mj service pool config pointer

This commit is contained in:
RockYang
2023-12-15 22:52:57 +08:00
parent 58f3dd5336
commit 3bea8f9706
8 changed files with 77 additions and 53 deletions

View File

@@ -3,18 +3,19 @@ package mj
import (
"chatplus/core/types"
"fmt"
"github.com/imroc/req/v3"
"time"
"github.com/imroc/req/v3"
)
// MidJourney client
type Client struct {
client *req.Client
config *types.MidJourneyConfig
config types.MidJourneyConfig
}
func NewClient(config *types.MidJourneyConfig, proxy string) *Client {
func NewClient(config types.MidJourneyConfig, proxy string) *Client {
client := req.C().SetTimeout(10 * time.Second)
// set proxy URL
if proxy != "" {

View File

@@ -5,6 +5,7 @@ import (
"chatplus/service/oss"
"chatplus/store"
"fmt"
"github.com/go-redis/redis/v8"
"gorm.io/gorm"
)
@@ -24,7 +25,7 @@ func NewServicePool(db *gorm.DB, redisCli *redis.Client, manager *oss.UploaderMa
continue
}
// create mj client
client := NewClient(&config, appConfig.ProxyURL)
client := NewClient(config, appConfig.ProxyURL)
name := fmt.Sprintf("MjService-%d", k)
// create mj service