118 lines
5.0 KiB
Go
118 lines
5.0 KiB
Go
package imageaudit
|
||
|
||
import (
|
||
"fmt"
|
||
"service/api/consts"
|
||
"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()
|
||
// 将此前批次对该元素未成功，待处理的审核全部置为已失效
|
||
failStatuses := []int64{consts.ImageAudit_Rollbacked, consts.ImageAudit_ServiceFailed}
|
||
if err := _DefaultImageAuditTask.OpHandleOverdue(ctx, taskCtrlBlock.ImageAuditTask, util.DerefString(taskCtrlBlock.ImageAuditTask.BatchId), failStatuses); 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
|
||
}
|