package f5 import ( "flag" "sync" "sync/atomic" "time" "os" "q5" ) type App interface { GetPkgName() string NewUuid() int64 GetInstanceId() int32 GetNodeId() int32 GetZoneId() int32 HasFlag(int32) bool AddIMMsg(uint16, q5.Args) RegisterIMMsgHandle(uint16, func(q5.Args)) NotifyLoopCond() GetNowSeconds() int64 GetNowMillis() int64 GetNowNano() int64 RegisterMainThreadCb(func ()) } type UserApp interface { GetPkgName() string Init() Update() UnInit() } type app struct { zoneId int32 nodeId int32 instanceId int32 terminated bool flags map[int32]int32 servicing bool contextHash map[int32]q5.Args loopCond *sync.Cond imTopNode *IMMsgNode imBotNode *IMMsgNode imWorkNode *IMMsgNode imMsgMutex sync.RWMutex chGoLoopTimerExit chan int chGoLoopWait chan int64 nowTime time.Time nowUnixNano int64 imMsgHandlers [1024]func(q5.Args) maxRunDelay int64 maxScheduleTime int64 userApp UserApp } func (this *app) init(userApp UserApp) { this.userApp = userApp this.nowTime = time.Now() atomic.StoreInt64(&this.nowUnixNano, this.nowTime.UnixNano()) _timer = &Timer{} _timer.init() _sysLog = new(SysLog_) _sysLog.init() _tgLog = new(TGLog_) _tgLog.init() _dbFilter = new(dbFilter) _dbFilter.init() _goStyleDb = new(dbPool) _goStyleDb.init() _jsStyleDb = new(dbPool) _jsStyleDb.init() _httpCliMgr = new(HttpCliMgr) _httpCliMgr.init() { var tmpNodeId, tmpInstanceId int flag.IntVar(&tmpNodeId, "n", 0, "node id") flag.IntVar(&tmpInstanceId, "i", 0, "instance id") this.nodeId = int32(tmpNodeId) this.instanceId = int32(tmpInstanceId) flag.Parse() } this.loopCond = sync.NewCond(new(sync.Mutex)) this.chGoLoopTimerExit = make(chan int) this.chGoLoopWait = make(chan int64) this.outputRuningLog() GetSysLog().Info("node_id:%d instance_id:%d pid:%d", this.nodeId, this.instanceId, os.Getpid()) this.installTimer() go this.goLoopTimer() this.userApp.Init() } func (this *app) unInit() { this.chGoLoopTimerExit <- 1 _httpCliMgr.unInit() _goStyleDb.unInit() _jsStyleDb.unInit() _dbFilter.unInit() _timer.unInit() _timer = nil this.userApp.UnInit() } func (this *app) run() { for !this.terminated { this.nowTime = time.Now() atomic.StoreInt64(&this.nowUnixNano, this.nowTime.UnixNano()) beginTick := q5.GetTickCount() _timer.update() this.dispatchIMMsg() this.userApp.Update() this.schedule() endTick := q5.GetTickCount() if this.maxRunDelay < endTick - beginTick { this.maxRunDelay = endTick - beginTick } } } func (this *app) NewUuid() int64 { return 0 } func (this *app) GetInstanceId() int32 { return this.instanceId } func (this *app) GetNodeId() int32 { return this.nodeId } func (this *app) GetZoneId() int32 { return this.nodeId } func (this *app) GetPkgName() string { return this.userApp.GetPkgName() } func (this *app) HasFlag(flag int32) bool { _, ok := this.flags[flag] return ok } func (this *app) AddIMMsg(msgId uint16, params q5.Args) { p := new(IMMsgNode) p.msgId = msgId p.params = params this.imMsgMutex.Lock() if this.imBotNode != nil { this.imBotNode.next = p this.imBotNode = p } else { this.imTopNode = p this.imBotNode = p } this.imMsgMutex.Unlock() this.loopCond.Broadcast() } func (this *app) RegisterIMMsgHandle(msgId uint16, handle func(q5.Args)) { this.imMsgHandlers[msgId] = handle } func (this *app) RegisterMainThreadCb(func ()) { } func (this *app) goLoopTimer() { var waitMs int64 = 1000 * 10 for { select { case <-this.chGoLoopTimerExit: return case waitMs = <-this.chGoLoopWait: if waitMs < 1 { waitMs = 1 } case <-time.After(time.Millisecond * time.Duration(waitMs)): waitMs = 1000 * 10 this.loopCond.Broadcast() } } } func (this *app) schedule() { this.chGoLoopWait <- GetTimer().GetIdleTime() this.loopCond.L.Lock() this.loopCond.Wait() this.loopCond.L.Unlock() } func (this *app) NotifyLoopCond() { this.loopCond.Broadcast() } func (this *app) GetNowSeconds() int64 { return this.nowUnixNano / int64(time.Second) } func (this *app) GetNowMillis() int64 { return this.nowUnixNano / int64(time.Millisecond) } func (this *app) GetNowNano() int64 { return this.nowUnixNano } func (this *app) outputRuningLog() { } func (this *app) dispatchIMMsg() { this.imMsgMutex.Lock() this.imWorkNode = this.imTopNode this.imTopNode = nil this.imBotNode = nil this.imMsgMutex.Unlock() for this.imWorkNode != nil { currNode := this.imWorkNode this.imWorkNode = this.imWorkNode.next if this.imMsgHandlers[currNode.msgId] != nil { this.imMsgHandlers[currNode.msgId](currNode.params) } } } func (this *app) installTimer() { GetTimer().SetInterval(1000 * 60, func (ev int32, params *q5.Args) { if ev == q5.TIMER_EXEC_EVENT { GetSysLog().Info("max_run_delay:%d max_schedule_time:%d", _app.maxRunDelay, _app.maxScheduleTime) _app.maxRunDelay = 0 _app.maxScheduleTime = 0 } }) }