service/library/contentaudit/video_moderation/client.go

101 lines
3.7 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

package videomoderation
import (
"service/bizcommon/util"
"service/dbstruct"
"service/library/configcenter"
"service/library/logger"
video_moderation_proto "service/api/proto/video_moderation/proto"
video_moderation_task_proto "service/api/proto/video_moderation_task/proto"
openapi "github.com/alibabacloud-go/darabonba-openapi/v2/client"
green20220302 "github.com/alibabacloud-go/green-20220302/client"
"github.com/alibabacloud-go/tea/tea"
"github.com/gin-gonic/gin"
)
// 数据库接口
type VideoModerationService interface {
OpCreate(ctx *gin.Context, req *video_moderation_proto.OpCreateReq) error
OpUpdate(ctx *gin.Context, req *video_moderation_proto.OpUpdateReq) error
OpDelete(ctx *gin.Context, id string) error
OpList(ctx *gin.Context, req *video_moderation_proto.OpListReq) (*dbstruct.VideoModeration, error)
OpCreateFragmentGroup(ctx *gin.Context, req *video_moderation_proto.OpCreateBatchReq) error
OpUpdateByIds(ctx *gin.Context, req *video_moderation_proto.OpUpdateByIdsReq) error
OpUpdateByBatchId(ctx *gin.Context, batchId string, imageaudit *dbstruct.VideoModeration) error
}
type VideoModerationTaskService interface {
OpCreate(ctx *gin.Context, req *video_moderation_task_proto.OpCreateReq) error
OpUpdate(ctx *gin.Context, req *video_moderation_task_proto.OpUpdateReq) error
OpDelete(ctx *gin.Context, id string) error
OpList(ctx *gin.Context, req *video_moderation_task_proto.OpListReq) ([]*dbstruct.VideoModerationTask, error)
OpUpdateByIds(ctx *gin.Context, req *video_moderation_task_proto.OpUpdateByIdsReq) error
OpUpdateByBatchId(ctx *gin.Context, batchId string, imageaudittask *dbstruct.VideoModerationTask) error
OpHandleOverdue(ctx *gin.Context, task *dbstruct.VideoModerationTask, batchId string, statuses []int64) error
}
type ContentAuditRTIService interface {
GetAndUpdateVideoModerationBatchId(ctx *gin.Context, batchId string) (string, error)
}
type VideoModerationTaskResultHandler interface {
Handle(ctx *gin.Context, task *dbstruct.VideoModerationTask, option int) error
}
var defaultVideoModerationClient *green20220302.Client
var _DefaultVideoModeration VideoModerationService
var _DefaultVideoModerationTask VideoModerationTaskService
var _DefaultResultHandler VideoModerationTaskResultHandler
var _DefaultContentAuditRTI ContentAuditRTIService
var _defaultConfig *configcenter.VideoModerationConfig
func Init(cfg *configcenter.VideoModerationConfig) (err error) {
config := &openapi.Config{
AccessKeyId: tea.String(cfg.AccessKeyId),
AccessKeySecret: tea.String(cfg.AccessKeySecret),
RegionId: tea.String("cn-shanghai"),
Endpoint: tea.String("green-cip.cn-shanghai.aliyuncs.com"),
/**
* 请设置超时时间。服务端全链路处理超时时间为10秒请做相应设置。
* 如果您设置的ReadTimeout小于服务端处理的时间程序中会获得一个ReadTimeout异常。
*/
ConnectTimeout: tea.Int(3000),
ReadTimeout: tea.Int(6000),
}
defaultVideoModerationClient, err = green20220302.NewClient(config)
if err != nil {
logger.Error("NewVideoModerationClient fail, cfg: %v, err: %v", util.ToJson(config), err)
return
}
_, err = RefreshBatchId()
if err != nil {
logger.Error("RefreshBatchId fail, cfg: %v, err: %v", util.ToJson(config), err)
return
}
// 初始化调度器
initScheduler(cfg)
_defaultConfig = cfg
return
}
func ConnectToVideoModerationService(serivce VideoModerationService) {
_DefaultVideoModeration = serivce
}
func ConnectToVideoModerationTaskService(service VideoModerationTaskService, handler VideoModerationTaskResultHandler) {
_DefaultVideoModerationTask = service
_DefaultResultHandler = handler
}
func ConnectToContentAuditRTIService(service ContentAuditRTIService) {
_DefaultContentAuditRTI = service
}