package base

import (
	"gitee.com/micro-tools/micro/module"
	"gitee.com/micro-tools/wf/os/glog"
)

// NewModuleManager 新建模块管理器
func NewModuleManager() (m *ModuleManager) {
	m = new(ModuleManager)
	return
}

// ModuleManager 模块管理器
type ModuleManager struct {
	app     module.App
	mods    []*DefaultModule
	runMods []*DefaultModule
}

// Register 注册模块
func (mer *ModuleManager) Register(mi module.Module, logger *glog.Logger) {
	md := new(DefaultModule)
	md.mi = mi
	md.logger = logger
	md.closeSig = make(chan bool, 1)

	mer.mods = append(mer.mods, md)
}

// RegisterRunMod 注册需要运行的模块
func (mer *ModuleManager) RegisterRunMod(mi module.Module) {
	md := new(DefaultModule)
	md.mi = mi
	md.closeSig = make(chan bool, 1)

	mer.runMods = append(mer.runMods, md)
}

// Init 初始化
func (mer *ModuleManager) Init(app module.App) {
	mer.app = app
	for i := 0; i < len(mer.mods); i++ {
		mer.runMods = append(mer.runMods, mer.mods[i]) //这里加入能够运行的组件
	}
	for i := 0; i < len(mer.runMods); i++ {
		m := mer.runMods[i]
		m.mi.OnInit(app)

		m.wg.Add(1)
		go run(m)
	}
}

// Destroy 停止模块
func (mer *ModuleManager) Destroy() {
	for i := len(mer.runMods) - 1; i >= 0; i-- {
		m := mer.runMods[i]
		m.closeSig <- true
		m.wg.Wait()
		destroy(m)
	}
}