light-apiengine/routers/router.go

package routers

import (
	"git.qianqiusoft.com/qianqiusoft/light-apiengine/engine"
	"git.qianqiusoft.com/qianqiusoft/light-apiengine/models"
	"sync"
)

var (
	_registerHandlers []func(e *engine.ApiEngine) = nil
	_registerHandlerMutex sync.Mutex
)

func addRegisterHandler(handler func(e *engine.ApiEngine)){
	_registerHandlerMutex.Lock()
	_registerHandlerMutex.Unlock()
	if _registerHandlers == nil{
		_registerHandlers = make([]func(e *engine.ApiEngine), 0)
	}
	_registerHandlers = append(_registerHandlers, handler)
}

func Register(e *engine.ApiEngine){
	models.SyncDb(e.OrmEngine)

	_registerHandlerMutex.Lock()
	_registerHandlerMutex.Unlock()
	for i := range _registerHandlers{
		_registerHandlers[i](e)
	}
}

func InitRouter(e *engine.ApiEngine){
	Register(e)
}