83 lines
2.0 KiB
Go
83 lines
2.0 KiB
Go
package imageaudit
|
|
|
|
import (
|
|
"fmt"
|
|
"service/library/configcenter"
|
|
"time"
|
|
)
|
|
|
|
var defaultImageAuditTaskScheduler *ImageAuditTaskScheduler
|
|
|
|
// 图像审核任务调度器
|
|
type ImageAuditTaskScheduler struct {
|
|
// // 同步标志
|
|
batchFlag chan bool // 批处理同步标志
|
|
|
|
// 状态记录
|
|
batchId string // 当前批次号
|
|
}
|
|
|
|
func initScheduler(cfg *configcenter.ImageAuditConfig) {
|
|
defaultImageAuditTaskScheduler = &ImageAuditTaskScheduler{
|
|
// taskBuffer: make(chan *ImageAuditTaskControlBlock, cfg.TaskBufferSize),
|
|
batchFlag: make(chan bool, 1),
|
|
// taskPacket: make([]*ImageAuditTaskControlBlock, 0),
|
|
batchId: genereteBatchId(),
|
|
}
|
|
defaultImageAuditTaskScheduler.batchFlag <- true
|
|
}
|
|
|
|
// 批处理上锁
|
|
func (s *ImageAuditTaskScheduler) lock() {
|
|
<-s.batchFlag
|
|
}
|
|
|
|
// 批处理解锁
|
|
func (s *ImageAuditTaskScheduler) unLock() {
|
|
s.batchFlag <- true
|
|
}
|
|
|
|
// // 调度方法
|
|
// func (s *ImageAuditTaskScheduler) Run() {
|
|
// go func() {
|
|
// // 1.循环取出缓冲池中的任务
|
|
// // 2.如果任务标志进行批处理，则将任务包打包，进行审核作业，最后清空任务包，生成新的批次号
|
|
// // 3.否则，将任务加入任务包中，并为任务建立索引，记录任务执行顺序
|
|
// for {
|
|
// task := <-s.taskBuffer
|
|
// if task.IsGivingNoticeToBatch {
|
|
// // 上锁
|
|
// s.lock()
|
|
|
|
// // 打包任务
|
|
// packet := s.taskPacket
|
|
// s.taskPacket = make([]*ImageAuditTaskControlBlock, 0)
|
|
|
|
// // 取出状态并重置
|
|
// batchId := s.batchId
|
|
// s.batchId = genereteBatchId()
|
|
// // 执行审核作业
|
|
// go func() {
|
|
// err := executeImageAuditTasks(packet, batchId)
|
|
// if err != nil {
|
|
// logger.Error("Batch failed : %v", err)
|
|
// }
|
|
// // 解锁
|
|
// s.unLock()
|
|
// }()
|
|
|
|
// } else {
|
|
// s.taskPacket = append(s.taskPacket, task)
|
|
// }
|
|
// }
|
|
// }()
|
|
// }
|
|
|
|
// 生成批次号
|
|
func genereteBatchId() string {
|
|
now := time.Now()
|
|
y, m, d := now.Date()
|
|
h, mi, s := now.Clock()
|
|
return fmt.Sprintf("%d%2d%2d%2d%2d%02d", y, m, d, h, mi, s)
|
|
}
|