service/library/contentaudit/video_moderation/client.go

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
}