f5/app.go

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
	uuid q5.Uuid
	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.uuid.SetMachineId((this.nodeId - 1) * MAX_NODE_ID + this.instanceId)
	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 this.uuid.Generate()
}

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
			}
		})
}