tank/rest/context.go

package rest

import (
	"fmt"
	"github.com/jinzhu/gorm"
	"reflect"
)

//全局唯一的上下文(在main函数中初始化)
var CONTEXT = &Context{}

//上下文，管理数据库连接，管理所有路由请求，管理所有的单例component.
type Context struct {
	//数据库连接
	DB *gorm.DB
	//session缓存
	SessionCache *CacheTable
	//TODO:日志相关内容

	//各类的Bean Map。这里面是包含ControllerMap中所有元素
	BeanMap map[string]IBean
	//只包含了Controller的map
	ControllerMap map[string]IController
	//处理所有路由请求
	Router *Router
}

func (this *Context) OpenDb() {

	var err error = nil
	this.DB, err = gorm.Open("mysql", CONFIG.MysqlUrl)

	//是否打开sql日志
	this.DB.LogMode(false)
	if err != nil {
		panic("failed to connect mysql database")
	}
}

func (this *Context) CloseDb() {

	if this.DB != nil {
		err := this.DB.Close()
		if err != nil {
			fmt.Println("关闭数据库连接出错", err)
		}
	}
}

//构造方法
func (this *Context) Init()  {

	//处理数据库连接的开关。
	this.OpenDb()

	//创建一个用于存储session的缓存。
	this.SessionCache = NewCacheTable()

	//初始化Map
	this.BeanMap = make(map[string]IBean)
	this.ControllerMap = make(map[string]IController)

	//注册各类Beans.在这个方法里面顺便把Controller装入ControllerMap中去。
	this.registerBeans()

	//初始化每个bean.
	this.initBeans()

	//初始化Router. 这个方法要在Bean注册好了之后才能。
	this.Router = NewRouter(this)
}

//注册一个Bean
func (this *Context) registerBean(bean IBean) {

	typeOf := reflect.TypeOf(bean)
	typeName := typeOf.String()

	if element, ok := bean.(IBean); ok {

		err := fmt.Sprintf("【%s】已经被注册了，跳过。", typeName)
		if _, ok := this.BeanMap[typeName]; ok {
			LogError(fmt.Sprintf(err))
		} else {
			this.BeanMap[typeName] = element

			//看看是不是controller类型，如果是，那么单独放在ControllerMap中。
			if controller, ok1 := bean.(IController); ok1 {
				this.ControllerMap[typeName] = controller
			}

		}

	} else {
		err := fmt.Sprintf("注册的【%s】不是Bean类型。", typeName)
		panic(err)
	}

}

//注册各个Beans
func (this *Context) registerBeans() {

	//alien
	this.registerBean(new(AlienController))
	this.registerBean(new(AlienService))

	//downloadToken
	this.registerBean(new(DownloadTokenDao))

	//imageCache
	this.registerBean(new(ImageCacheController))
	this.registerBean(new(ImageCacheDao))
	this.registerBean(new(ImageCacheService))

	//matter
	this.registerBean(new(MatterController))
	this.registerBean(new(MatterDao))
	this.registerBean(new(MatterService))

	//preference
	this.registerBean(new(PreferenceController))
	this.registerBean(new(PreferenceDao))
	this.registerBean(new(PreferenceService))

	//securityVisit
	this.registerBean(new(SecurityVisitController))
	this.registerBean(new(SecurityVisitDao))
	this.registerBean(new(SecurityVisitService))

	//session
	this.registerBean(new(SessionDao))

	//uploadToken
	this.registerBean(new(UploadTokenDao))

	//user
	this.registerBean(new(UserController))
	this.registerBean(new(UserDao))
	this.registerBean(new(UserService))

}

//从Map中获取某个Bean.
func (this *Context) GetBean(bean IBean) IBean {

	typeOf := reflect.TypeOf(bean)
	typeName := typeOf.String()

	if val, ok := this.BeanMap[typeName]; ok {
		return val
	} else {
		err := fmt.Sprintf("【%s】没有注册。", typeName)
		panic(err)
	}
}

//初始化每个Bean
func (this *Context) initBeans() {

	for _, bean := range this.BeanMap {
		bean.Init(this)
	}

}

//销毁的方法
func (this *Context) Destroy() {
	this.CloseDb()

}