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 { if task == nil || task.AuditedMedia == nil { return nil } fragmentNum := len(task.AuditedMedia.GetImageIds()) if fragmentNum == 0 { return nil } task.BatchId = goproto.String(defaultImageAuditTaskScheduler.batchId) 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 }