package imageaudit import ( "fmt" "service/bizcommon/util" "service/dbstruct" "service/library/logger" "service/library/mediafiller" "github.com/gin-gonic/gin" ) // 批次图像审核任务控制块 type ImageAuditTaskBatchControlBlock struct { BatchId string // 批次号 TaskCtrlBlocks []*ImageAuditTaskControlBlock // 任务控制块 Images []*dbstruct.MediaComponent // 送审图像序列 ImageAuditIds []string // 送审图像对应的图像审核表id Img2taskIndexMap map[int]int // 图像审核序列下标->任务序列下标的映射map ActionMap map[string]*ImageAuditAction // 动作Id号-action的map } func NewImageAuditTaskBatchControlBlock(tasks []*dbstruct.ImageAuditTask, batchId string) *ImageAuditTaskBatchControlBlock { if len(tasks) == 0 { return nil } ctrlBlock := &ImageAuditTaskBatchControlBlock{ BatchId: batchId, TaskCtrlBlocks: make([]*ImageAuditTaskControlBlock, 0), Images: make([]*dbstruct.MediaComponent, 0), ImageAuditIds: make([]string, 0), Img2taskIndexMap: make(map[int]int), ActionMap: make(map[string]*ImageAuditAction), } imageIndex := 0 taskIndex := 0 mediaFillables := make([]mediafiller.MediaFillable, 0) for _, task := range tasks { taskCtrlBlock := NewImageAuditTaskControlBlock(task) ctrlBlock.TaskCtrlBlocks = append(ctrlBlock.TaskCtrlBlocks, taskCtrlBlock) ctrlBlock.RecordAction(taskCtrlBlock) imageIndex, taskIndex = ctrlBlock.RecordImage(taskCtrlBlock, &mediaFillables, imageIndex, taskIndex) } mediafiller.FillListInternal(&gin.Context{}, mediaFillables) return ctrlBlock } func (ctrlBlock *ImageAuditTaskBatchControlBlock) RecordAction(taskCtrlBlock *ImageAuditTaskControlBlock) { ctx := &gin.Context{} // 写map记录 if ctrlBlock.ActionMap[taskCtrlBlock.ActionId] == nil { // 写入ctrlBlock.ActionMap ctrlBlock.ActionMap[taskCtrlBlock.ActionId] = NewImageAuditAction() // 将此前批次对该元素未成功,待处理的审核全部置为已失效 if err := _DefaultImageAuditTask.OpHandleOverdue(ctx, taskCtrlBlock.ImageAuditTask, util.DerefString(taskCtrlBlock.ImageAuditTask.BatchId)); err != nil { logger.Error("_DefaultImageAuditTask OpHandleOverdue fail :%v", err) } } ctrlBlock.ActionMap[taskCtrlBlock.ActionId].Record(taskCtrlBlock) } func (ctrlBlock *ImageAuditTaskBatchControlBlock) RecordImage(taskCtrlBlock *ImageAuditTaskControlBlock, mediaFillables *[]mediafiller.MediaFillable, imageIndex int, taskIndex int) (newImageIndex int, newTaskIndex int) { // 未分片 if util.DerefInt64(taskCtrlBlock.ImageAuditTask.IsFragmented) == 0 { ctrlBlock.Images = append(ctrlBlock.Images, taskCtrlBlock.ImageAuditTask.AuditedMedia) *mediaFillables = append(*mediaFillables, taskCtrlBlock.ImageAuditTask.AuditedMedia) ctrlBlock.Img2taskIndexMap[imageIndex] = taskIndex ctrlBlock.ImageAuditIds = append(ctrlBlock.ImageAuditIds, util.DerefString(taskCtrlBlock.ImageAuditTask.ImageAuditId)) imageIndex++ } else { // 已分片 imageIds := taskCtrlBlock.ImageAuditTask.AuditedMedia.GetImageIds() imageAuditIds := util.DerefStringSlice(taskCtrlBlock.ImageAuditTask.ImageAuditFragmentIds) for i, imageId := range imageIds { image := &dbstruct.MediaComponent{ ImageIds: util.Int64Slice([]int64{imageId}), } ctrlBlock.Images = append(ctrlBlock.Images, image) *mediaFillables = append(*mediaFillables, image) ctrlBlock.ImageAuditIds = append(ctrlBlock.ImageAuditIds, imageAuditIds[i]) ctrlBlock.Img2taskIndexMap[imageIndex] = taskIndex imageIndex++ } } taskIndex++ return imageIndex, taskIndex } // 图像审核任务控制块 // ActionId设计初衷:由于图像审核是定时任务触发的批量作业,如果在一次作业间隔有针对同一个图像媒体的多次更新,则会提交关于它的多次审核,需要保证数据一致性 type ImageAuditTaskControlBlock struct { // 静态元素 ActionId string // 审核动作id号,由图像审核实体数据库四要素拼接而成,用于指示对数据库-表-单条数据-图像字段的审核动作 ImageAuditTask *dbstruct.ImageAuditTask // 审核任务 // 动态元素 IsTaskPassed bool // 任务状态,仅当任务已确定完成,即已完成审核任务的分片数 = 审核任务分片数时,才有意义 AuditedFragmentsNum int // 已完成审核任务的分片数 } // 新建图像审核任务块 func NewImageAuditTaskControlBlock(task *dbstruct.ImageAuditTask) (tcb *ImageAuditTaskControlBlock) { tcb = &ImageAuditTaskControlBlock{ ActionId: fmt.Sprintf("%v%v%v%v", util.DerefString(task.AssociativeDatabase), util.DerefString(task.AssociativeTableName), util.DerefInt64(task.AssociativeTableId), util.DerefString(task.AssociativeTableColumn)), ImageAuditTask: task, IsTaskPassed: true, AuditedFragmentsNum: 0, } return }