package f5 import ( "q5" ) type taskLock struct { key string entry q5.ListHead } type LockAsyncTask struct { status int32 debugInfo string cb func(*LockAsyncTask) succCb func(*LockAsyncTask) failCb func(*LockAsyncTask) exitCb func(*LockAsyncTask) execTimes int64 preExecTimes int64 lockKeys map[string]*taskLock } func (this *LockAsyncTask) init(cb func(*LockAsyncTask)) *LockAsyncTask { this.cb = cb return this } func (this *LockAsyncTask) isRunning() bool { return this.status == 0 } func (this *LockAsyncTask) SetSucc() { if !this.isRunning() { panic("task is not runing") } this.status = 1 this.clearLocks() if this.succCb != nil { this.succCb(this) } if this.exitCb != nil { this.exitCb(this) } } func (this *LockAsyncTask) SetFail() { if !this.isRunning() { panic("task is not runing") } this.status = -1 this.clearLocks() if this.failCb != nil { this.failCb(this) } if this.exitCb != nil { this.exitCb(this) } } func (this *LockAsyncTask) checkDo() *LockAsyncTask { if !this.isRunning() { panic("task is not runing") } if this.allIsReady() { if this.preExecTimes <= 0 { this.preExecTimes++ if this.preExecTimes > 1 { panic("locktask preExectimes > 0") } GetApp().RegisterMainThreadCb( func () { this.cb(this) if this.execTimes > 0 { panic("locktask only run once") } this.execTimes += 1 }) } } return this } func (this *LockAsyncTask) OnSucc(cb func(*LockAsyncTask)) *LockAsyncTask { this.succCb = cb return this } func (this *LockAsyncTask) OnFail(cb func(*LockAsyncTask)) *LockAsyncTask { this.failCb = cb return this } func (this *LockAsyncTask) OnExit(cb func(*LockAsyncTask)) *LockAsyncTask { this.exitCb = cb return this } func (this *LockAsyncTask) allIsReady() bool { var allReady = true for _, lock := range(this.lockKeys) { if !_app.isFirstAsyncTask(lock.key, &lock.entry) { allReady = false break } } return allReady } func (this *LockAsyncTask) clearLocks() { for _, lock := range(this.lockKeys) { lock.entry.DelInit() } for _, lock := range(this.lockKeys) { _app.clearEmptyPendingAsyncTask(lock.key) } for _, lock := range(this.lockKeys) { _app.notifyNextTask(lock.key) } } func NewLockAsyncTask(keys [][]string, cb func(*LockAsyncTask)) *LockAsyncTask { p := new(LockAsyncTask) p.lockKeys = make(map[string]*taskLock) for _, val := range(keys) { if len(val) != 2 { panic("NewLockAsyncTask len error:" + q5.EncodeJson(val)) } lock := new(taskLock) lock.key = val[0] + val[1] lock.entry.Init(p) if _, ok := p.lockKeys[lock.key]; ok { panic("NewLockAsyncTask duplicate key:" + lock.key) } p.lockKeys[lock.key] = lock _app.forcePendingAsyncTask(lock.key).AddTail(&lock.entry) } if q5.IsDebug() { p.debugInfo = q5.GetCallStack() _timer.SetTimeout( 1000 * 2, func (e int32, args* q5.Args) { if e == q5.TIMER_EXEC_EVENT { if p.isRunning() { _sysLog.Warning("LockAsyncaskTimeOut %s", p.debugInfo) panic("LockAsyncaskTimeOut") } } }) } return p.init(cb).checkDo() }