122 lines
3.2 KiB
Go
122 lines
3.2 KiB
Go
package imageaudit
|
||
|
||
import (
|
||
"service/api/consts"
|
||
imageauditproto "service/api/proto/imageaudit/proto"
|
||
imageaudittaskproto "service/api/proto/imageaudittask/proto"
|
||
"service/bizcommon/util"
|
||
"service/dbstruct"
|
||
"service/library/logger"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
goproto "google.golang.org/protobuf/proto"
|
||
)
|
||
|
||
func AddTasks(tasks []*dbstruct.ImageAuditTask) error {
|
||
for _, task := range tasks {
|
||
err := AddTask(task)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func AddTask(task *dbstruct.ImageAuditTask) error {
|
||
// 若非对齐任务，则进行插入前预处理
|
||
task.BatchId = goproto.String(defaultImageAuditTaskScheduler.batchId)
|
||
if !task.IsForAlignment() {
|
||
if task == nil || task.AuditedMedia == nil {
|
||
return nil
|
||
}
|
||
fragmentNum := len(task.AuditedMedia.GetImageIds())
|
||
if fragmentNum == 0 {
|
||
return nil
|
||
}
|
||
|
||
task.Status = goproto.Int64(consts.ImageAudit_Created)
|
||
|
||
if fragmentNum == 1 {
|
||
task.IsFragmented = goproto.Int64(0)
|
||
task.FragmentsNum = goproto.Int64(1)
|
||
|
||
// 写入图像审核表
|
||
if err := prepareNotFragmentedImageAuditTask(task); err != nil {
|
||
return err
|
||
}
|
||
} else {
|
||
task.IsFragmented = goproto.Int64(1)
|
||
task.FragmentsNum = goproto.Int64(int64(fragmentNum))
|
||
|
||
// 写入图像审核表
|
||
if err := prepareFragmentedImageAuditTask(task); err != nil {
|
||
return err
|
||
}
|
||
}
|
||
}
|
||
|
||
// 写入图像审核任务表
|
||
if err := _DefaultImageAuditTask.OpCreate(&gin.Context{}, &imageaudittaskproto.OpCreateReq{
|
||
ImageAuditTask: task,
|
||
}); err != nil {
|
||
logger.Error("Imageaudittask OpCreate failed: %v", err)
|
||
return err
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// 根据任务信息，创建单个图片的图片审核，并存入数据库，返回images
|
||
func prepareFragmentedImageAuditTask(task *dbstruct.ImageAuditTask) (err error) {
|
||
ctx := &gin.Context{}
|
||
|
||
fragmentsNum := util.DerefInt64(task.FragmentsNum)
|
||
imageAudits := make([]*dbstruct.ImageAudit, fragmentsNum)
|
||
|
||
// 将媒体拆分为单个图像并存入图片审核表
|
||
imageIds := task.AuditedMedia.GetImageIds()
|
||
for i, imageId := range imageIds {
|
||
image := &dbstruct.MediaComponent{
|
||
ImageIds: util.Int64Slice([]int64{imageId}),
|
||
}
|
||
imageAudits[i] = &dbstruct.ImageAudit{
|
||
AuditedMedia: image,
|
||
BatchId: task.BatchId,
|
||
Status: goproto.Int64(consts.ImageAudit_Created),
|
||
}
|
||
}
|
||
if err = _DefaultImageAudit.OpCreateFragmentGroup(ctx, &imageauditproto.OpCreateBatchReq{
|
||
ImageAudits: imageAudits,
|
||
FragmentsNum: fragmentsNum,
|
||
}); err != nil {
|
||
logger.Error("Imageaudit OpCreateFragmentGroup failed: %v", err)
|
||
return
|
||
}
|
||
|
||
auditFragmentIds := make([]string, fragmentsNum)
|
||
for i, imageaudit := range imageAudits {
|
||
auditFragmentIds[i] = util.DerefString(imageaudit.Id)
|
||
}
|
||
task.ImageAuditFragmentIds = &auditFragmentIds
|
||
return
|
||
}
|
||
|
||
func prepareNotFragmentedImageAuditTask(task *dbstruct.ImageAuditTask) (err error) {
|
||
ctx := &gin.Context{}
|
||
|
||
imageAudit := &dbstruct.ImageAudit{
|
||
AuditedMedia: task.AuditedMedia,
|
||
BatchId: task.BatchId,
|
||
Status: goproto.Int64(consts.ImageAudit_Created),
|
||
}
|
||
if err = _DefaultImageAudit.OpCreate(ctx, &imageauditproto.OpCreateReq{
|
||
ImageAudit: imageAudit,
|
||
}); err != nil {
|
||
logger.Error("Imageaudit OpCreate failed: %v", err)
|
||
return
|
||
}
|
||
|
||
task.ImageAuditId = imageAudit.Id
|
||
return
|
||
}
|