54 lines
1.9 KiB
Go
54 lines
1.9 KiB
Go
package yidun
|
||
|
||
import (
|
||
"context"
|
||
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/yidun/yidun-golang-sdk/yidun/service/antispam/image/v5"
|
||
text "github.com/yidun/yidun-golang-sdk/yidun/service/antispam/text"
|
||
"github.com/yidun/yidun-golang-sdk/yidun/service/antispam/videosolution"
|
||
)
|
||
|
||
// YidunClients 易盾客户端集合（直接使用SDK客户端）
|
||
type YidunClients struct {
|
||
TextClient *text.TextClient
|
||
ImageClient *image.ImageClient
|
||
VideoClient *videosolution.VideoSolutionClient
|
||
}
|
||
|
||
var DefaultClients *YidunClients
|
||
|
||
// InitYidunClients 初始化所有易盾客户端
|
||
func InitYidunClients(ctx context.Context) error {
|
||
clients := &YidunClients{}
|
||
|
||
// 文本检测
|
||
secretId := g.Cfg().MustGet(ctx, "yidun.text.secret_id").String()
|
||
secretKey := g.Cfg().MustGet(ctx, "yidun.text.secret_key").String()
|
||
if secretId != "" && secretKey != "" {
|
||
clients.TextClient = text.NewTextClientWithAccessKey(secretId, secretKey)
|
||
g.Log().Info(ctx, "文本检测客户端初始化成功")
|
||
}
|
||
|
||
// 图片检测
|
||
secretId = g.Cfg().MustGet(ctx, "yidun.image.secret_id").String()
|
||
secretKey = g.Cfg().MustGet(ctx, "yidun.image.secret_key").String()
|
||
businessId := g.Cfg().MustGet(ctx, "yidun.image.business_id").String()
|
||
if secretId != "" && secretKey != "" {
|
||
clients.ImageClient = image.NewImageClientWithAccessKey(secretId, secretKey)
|
||
g.Log().Infof(ctx, "图片检测客户端初始化成功, business_id: %s", businessId)
|
||
}
|
||
|
||
// 视频检测
|
||
secretId = g.Cfg().MustGet(ctx, "yidun.video.secret_id").String()
|
||
secretKey = g.Cfg().MustGet(ctx, "yidun.video.secret_key").String()
|
||
businessId = g.Cfg().MustGet(ctx, "yidun.video.business_id").String()
|
||
if secretId != "" && secretKey != "" {
|
||
clients.VideoClient = videosolution.NewVideoSolutionClientWithAccessKey(secretId, secretKey)
|
||
g.Log().Infof(ctx, "视频检测客户端初始化成功, business_id: %s", businessId)
|
||
}
|
||
|
||
DefaultClients = clients
|
||
return nil
|
||
}
|