Ma/origin
1
0
Fork 0
mirror of https://github.com/duanhf2012/origin.git synced 2026-03-05 13:47:35 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

整理module

Browse Source
This commit is contained in:
boyce
2019-02-22 14:17:04 +08:00
parent ee61b6145e
commit 29607777f8
4 changed files with 291 additions and 205 deletions
Download Patch File Download Diff File Expand all files Collapse all files

204
service/Service.go
View File

@@ -7,7 +7,6 @@ import (
"os"
"reflect"
"strings"
"sync"
"time"
)
@@ -17,26 +16,6 @@ type MethodInfo struct {
types reflect.Type
}
type IModule interface {
SetModuleId(moduleId uint32) bool
GetModuleId() uint32
GetModuleById(moduleId uint32) IModule
AddModule(module IModule) uint32
RunModule(module IModule)
InitModule(exit chan bool, pwaitGroup *sync.WaitGroup) error
OnInit() error
OnRun() bool
GetOwnerService() IService
SetOwnerService(iservice IService)
SetOwner(module IModule)
GetOwner() IModule
getBaseModule() *BaseModule
}
type IService interface {
Init(Iservice IService) error
OnInit() error
@@ -75,22 +54,6 @@ type BaseService struct {
Status int
}
type BaseModule struct {
moduleId uint32
ownerService IService
tickTime int64
ExitChan chan bool
WaitGroup *sync.WaitGroup
mapModule map[uint32]IModule
CurrMaxModuleId uint32
rwModuleLocker sync.RWMutex
ownerModule IModule
}
func (slf *BaseService) GetServiceId() int {
return slf.serviceid
}
@@ -142,170 +105,3 @@ func (slf *BaseService) IsTimeOutTick(microSecond int64) bool {
nowtm := time.Now().UnixNano() / 1e6
return nowtm-slf.tickTime >= microSecond
}
func (slf *BaseModule) SetModuleId(moduleId uint32) bool {
slf.moduleId = moduleId
return true
}
func (slf *BaseModule) GetModuleId() uint32 {
return slf.moduleId
}
func (slf *BaseModule) GetModuleById(moduleId uint32) IModule {
slf.rwModuleLocker.RLock()
ret, ok := slf.mapModule[moduleId]
if ok == false {
slf.rwModuleLocker.RUnlock()
return nil
}
slf.rwModuleLocker.RUnlock()
return ret
}
func (slf *BaseModule) genModuleId() uint32 {
slf.rwModuleLocker.Lock()
slf.CurrMaxModuleId++
moduleId := slf.CurrMaxModuleId
slf.rwModuleLocker.Unlock()
return moduleId
}
func (slf *BaseModule) RemoveModule(moduleId uint32) bool {
slf.rwModuleLocker.Lock()
_, ok := slf.mapModule[moduleId]
if ok == false {
GetLogger().Printf(LEVER_WARN, "%T RemoveModule fail %d...", slf.GetOwner().GetModuleById(slf.GetModuleId()), moduleId)
slf.rwModuleLocker.Unlock()
return false
}
delete(slf.mapModule, moduleId)
slf.rwModuleLocker.Unlock()
return true
}
func (slf *BaseModule) IsRoot() bool {
return slf.GetOwner().GetModuleById(slf.GetModuleId()) == nil
}
const (
//ModuleNone ...
MAX_ALLOW_SET_MODULE_ID = iota + 100000000
INIT_AUTO_INCREMENT
)
func (slf *BaseModule) AddModule(module IModule) uint32 {
if slf.WaitGroup == nil {
GetLogger().Printf(LEVER_FATAL, "AddModule error wait group is nil:%T...", module)
return 0
}
//用户设置的id不允许大于MAX_ALLOW_SET_MODULE_ID
if module.GetModuleId() > MAX_ALLOW_SET_MODULE_ID {
return 0
}
//如果没有设置，自动生成ModuleId
if module.GetModuleId() == 0 {
module.SetModuleId(slf.genModuleId())
}
if slf.GetOwner() != nil {
if slf.IsRoot() {
module.SetOwner(slf.GetOwner())
} else {
module.SetOwner(slf.GetOwner().GetModuleById(slf.GetModuleId()))
}
}
//设置模块退出信号捕获
module.InitModule(slf.ExitChan, slf.WaitGroup)
//存入父模块中
slf.rwModuleLocker.Lock()
if slf.mapModule == nil {
slf.mapModule = make(map[uint32]IModule)
}
_, ok := slf.mapModule[module.GetModuleId()]
if ok == true {
slf.rwModuleLocker.Unlock()
return 0
}
slf.mapModule[module.GetModuleId()] = module
slf.rwModuleLocker.Unlock()
//运行模块
go module.RunModule(module)
return module.GetModuleId()
}
func (slf *BaseModule) OnInit() error {
return nil
}
func (slf *BaseModule) OnRun() bool {
return false
}
func (slf *BaseModule) SetOwner(ownerModule IModule) {
slf.ownerModule = ownerModule
}
func (slf *BaseModule) GetOwner() IModule {
return slf.ownerModule
}
func (slf *BaseModule) GetOwnerService() IService {
return slf.ownerService
}
func (slf *BaseModule) SetOwnerService(iservice IService) {
slf.ownerService = iservice
}
func (slf *BaseModule) InitModule(exit chan bool, pwaitGroup *sync.WaitGroup) error {
slf.CurrMaxModuleId = INIT_AUTO_INCREMENT
slf.WaitGroup = pwaitGroup
slf.ExitChan = exit
return nil
}
func (slf *BaseModule) getBaseModule() *BaseModule {
return slf
}
func (slf *BaseModule) GetSelf() IModule {
if slf.IsRoot() {
return slf.GetOwner()
}
return slf.GetOwner().GetModuleById(slf.GetModuleId())
}
func (slf *BaseModule) RunModule(module IModule) {
module.OnInit()
//运行所有子模块
slf.WaitGroup.Add(1)
defer slf.WaitGroup.Done()
for {
select {
case <-slf.ExitChan:
GetLogger().Printf(LEVER_WARN, "Stopping module %T...", slf.GetSelf())
fmt.Println("Stopping module %T...", slf.GetSelf())
return
}
if module.OnRun() == false {
break
}
slf.tickTime = time.Now().UnixNano() / 1e6
}
}