zicla
7 years ago
24 changed files with 371 additions and 339 deletions
@ -0,0 +1,27 @@ |
|||
package core |
|||
|
|||
const ( |
|||
//用户身份的cookie字段名
|
|||
COOKIE_AUTH_KEY = "_ak" |
|||
|
|||
//数据库表前缀 tank200表示当前应用版本是tank:2.0.x版，数据库结构发生变化必然是中型升级
|
|||
TABLE_PREFIX = "tank20_" |
|||
|
|||
//当前版本
|
|||
VERSION = "2.0.0" |
|||
) |
|||
|
|||
type Config interface { |
|||
|
|||
//是否已经安装
|
|||
IsInstalled() bool |
|||
//启动端口
|
|||
GetServerPort() int |
|||
//获取mysql链接
|
|||
GetMysqlUrl() string |
|||
|
|||
//文件存放路径
|
|||
GetMatterPath() string |
|||
//完成安装过程，主要是要将配置写入到文件中
|
|||
FinishInstall(mysqlPort int, mysqlHost string, mysqlSchema string, mysqlUsername string, mysqlPassword string) |
|||
} |
|||
@ -1,220 +0,0 @@ |
|||
package rest |
|||
|
|||
import ( |
|||
"fmt" |
|||
"github.com/eyebluecn/tank/code/config" |
|||
"github.com/eyebluecn/tank/code/core" |
|||
"github.com/eyebluecn/tank/code/tool/cache" |
|||
"github.com/jinzhu/gorm" |
|||
"net/http" |
|||
"reflect" |
|||
) |
|||
|
|||
//上下文，管理数据库连接，管理所有路由请求，管理所有的单例component.
|
|||
type Context struct { |
|||
//数据库连接
|
|||
db *gorm.DB |
|||
//session缓存
|
|||
SessionCache *cache.Table |
|||
//各类的Bean Map。这里面是包含ControllerMap中所有元素
|
|||
BeanMap map[string]core.IBean |
|||
//只包含了Controller的map
|
|||
ControllerMap map[string]core.IController |
|||
//处理所有路由请求
|
|||
Router *Router |
|||
} |
|||
|
|||
//初始化上下文
|
|||
func (this *Context) Init() { |
|||
|
|||
//创建一个用于存储session的缓存。
|
|||
this.SessionCache = cache.NewTable() |
|||
|
|||
//初始化Map
|
|||
this.BeanMap = make(map[string]core.IBean) |
|||
this.ControllerMap = make(map[string]core.IController) |
|||
|
|||
//注册各类Beans.在这个方法里面顺便把Controller装入ControllerMap中去。
|
|||
this.registerBeans() |
|||
|
|||
//初始化每个bean.
|
|||
this.initBeans() |
|||
|
|||
//初始化Router. 这个方法要在Bean注册好了之后才能。
|
|||
this.Router = NewRouter() |
|||
|
|||
//如果数据库信息配置好了，就直接打开数据库连接 同时执行Bean的ConfigPost方法
|
|||
this.InstallOk() |
|||
|
|||
} |
|||
|
|||
//获取数据库对象
|
|||
func (this *Context) GetDB() *gorm.DB { |
|||
return this.db |
|||
} |
|||
|
|||
func (this *Context) GetSessionCache() *cache.Table { |
|||
return this.SessionCache |
|||
} |
|||
|
|||
func (this *Context) GetControllerMap() map[string]core.IController { |
|||
return this.ControllerMap |
|||
} |
|||
|
|||
func (this *Context) Cleanup() { |
|||
for _, bean := range this.BeanMap { |
|||
bean.Cleanup() |
|||
} |
|||
} |
|||
|
|||
//响应http的能力
|
|||
func (this *Context) ServeHTTP(writer http.ResponseWriter, request *http.Request) { |
|||
this.Router.ServeHTTP(writer, request) |
|||
} |
|||
|
|||
func (this *Context) OpenDb() { |
|||
|
|||
var err error = nil |
|||
this.db, err = gorm.Open("mysql", config.CONFIG.MysqlUrl) |
|||
|
|||
if err != nil { |
|||
core.LOGGER.Panic("failed to connect mysql database") |
|||
} |
|||
|
|||
//是否打开sql日志(在调试阶段可以打开，以方便查看执行的SQL)
|
|||
this.db.LogMode(false) |
|||
} |
|||
|
|||
func (this *Context) CloseDb() { |
|||
|
|||
if this.db != nil { |
|||
err := this.db.Close() |
|||
if err != nil { |
|||
core.LOGGER.Error("关闭数据库连接出错 %s", err.Error()) |
|||
} |
|||
} |
|||
} |
|||
|
|||
//注册一个Bean
|
|||
func (this *Context) registerBean(bean core.IBean) { |
|||
|
|||
typeOf := reflect.TypeOf(bean) |
|||
typeName := typeOf.String() |
|||
|
|||
if element, ok := bean.(core.IBean); ok { |
|||
|
|||
err := fmt.Sprintf("【%s】已经被注册了，跳过。", typeName) |
|||
if _, ok := this.BeanMap[typeName]; ok { |
|||
core.LOGGER.Error(fmt.Sprintf(err)) |
|||
} else { |
|||
this.BeanMap[typeName] = element |
|||
|
|||
//看看是不是controller类型，如果是，那么单独放在ControllerMap中。
|
|||
if controller, ok1 := bean.(core.IController); ok1 { |
|||
this.ControllerMap[typeName] = controller |
|||
} |
|||
|
|||
} |
|||
|
|||
} else { |
|||
core.LOGGER.Panic("注册的【%s】不是Bean类型。", typeName) |
|||
} |
|||
|
|||
} |
|||
|
|||
//注册各个Beans
|
|||
func (this *Context) registerBeans() { |
|||
|
|||
//alien
|
|||
this.registerBean(new(AlienController)) |
|||
this.registerBean(new(AlienService)) |
|||
|
|||
//dashboard
|
|||
this.registerBean(new(DashboardController)) |
|||
this.registerBean(new(DashboardDao)) |
|||
this.registerBean(new(DashboardService)) |
|||
|
|||
//downloadToken
|
|||
this.registerBean(new(DownloadTokenDao)) |
|||
|
|||
//imageCache
|
|||
this.registerBean(new(ImageCacheController)) |
|||
this.registerBean(new(ImageCacheDao)) |
|||
this.registerBean(new(ImageCacheService)) |
|||
|
|||
//install
|
|||
this.registerBean(new(InstallController)) |
|||
|
|||
//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)) |
|||
|
|||
//footprint
|
|||
this.registerBean(new(FootprintController)) |
|||
this.registerBean(new(FootprintDao)) |
|||
this.registerBean(new(FootprintService)) |
|||
|
|||
//session
|
|||
this.registerBean(new(SessionDao)) |
|||
this.registerBean(new(SessionService)) |
|||
|
|||
//uploadToken
|
|||
this.registerBean(new(UploadTokenDao)) |
|||
|
|||
//user
|
|||
this.registerBean(new(UserController)) |
|||
this.registerBean(new(UserDao)) |
|||
this.registerBean(new(UserService)) |
|||
|
|||
//webdav
|
|||
this.registerBean(new(DavController)) |
|||
this.registerBean(new(DavService)) |
|||
|
|||
} |
|||
|
|||
//从Map中获取某个Bean.
|
|||
func (this *Context) GetBean(bean core.IBean) core.IBean { |
|||
|
|||
typeOf := reflect.TypeOf(bean) |
|||
typeName := typeOf.String() |
|||
|
|||
if val, ok := this.BeanMap[typeName]; ok { |
|||
return val |
|||
} else { |
|||
core.LOGGER.Panic("【%s】没有注册。", typeName) |
|||
return nil |
|||
} |
|||
} |
|||
|
|||
//初始化每个Bean
|
|||
func (this *Context) initBeans() { |
|||
|
|||
for key, bean := range this.BeanMap { |
|||
core.LOGGER.Info("init %s", key) |
|||
bean.Init() |
|||
} |
|||
} |
|||
|
|||
//系统如果安装好了就调用这个方法。
|
|||
func (this *Context) InstallOk() { |
|||
|
|||
if config.CONFIG.Installed { |
|||
this.OpenDb() |
|||
|
|||
for _, bean := range this.BeanMap { |
|||
bean.Bootstrap() |
|||
} |
|||
} |
|||
|
|||
} |
|||
|
|||
//销毁的方法
|
|||
func (this *Context) Destroy() { |
|||
this.CloseDb() |
|||
} |
|||
@ -1 +1,220 @@ |
|||
package support |
|||
|
|||
import ( |
|||
"fmt" |
|||
|
|||
"github.com/eyebluecn/tank/code/core" |
|||
"github.com/eyebluecn/tank/code/rest" |
|||
"github.com/eyebluecn/tank/code/tool/cache" |
|||
"github.com/jinzhu/gorm" |
|||
"net/http" |
|||
"reflect" |
|||
) |
|||
|
|||
//上下文，管理数据库连接，管理所有路由请求，管理所有的单例component.
|
|||
type TankContext struct { |
|||
//数据库连接
|
|||
db *gorm.DB |
|||
//session缓存
|
|||
SessionCache *cache.Table |
|||
//各类的Bean Map。这里面是包含ControllerMap中所有元素
|
|||
BeanMap map[string]core.IBean |
|||
//只包含了Controller的map
|
|||
ControllerMap map[string]core.IController |
|||
//处理所有路由请求
|
|||
Router *TankRouter |
|||
} |
|||
|
|||
//初始化上下文
|
|||
func (this *TankContext) Init() { |
|||
|
|||
//创建一个用于存储session的缓存。
|
|||
this.SessionCache = cache.NewTable() |
|||
|
|||
//初始化Map
|
|||
this.BeanMap = make(map[string]core.IBean) |
|||
this.ControllerMap = make(map[string]core.IController) |
|||
|
|||
//注册各类Beans.在这个方法里面顺便把Controller装入ControllerMap中去。
|
|||
this.registerBeans() |
|||
|
|||
//初始化每个bean.
|
|||
this.initBeans() |
|||
|
|||
//初始化Router. 这个方法要在Bean注册好了之后才能。
|
|||
this.Router = NewRouter() |
|||
|
|||
//如果数据库信息配置好了，就直接打开数据库连接 同时执行Bean的ConfigPost方法
|
|||
this.InstallOk() |
|||
|
|||
} |
|||
|
|||
//获取数据库对象
|
|||
func (this *TankContext) GetDB() *gorm.DB { |
|||
return this.db |
|||
} |
|||
|
|||
func (this *TankContext) GetSessionCache() *cache.Table { |
|||
return this.SessionCache |
|||
} |
|||
|
|||
func (this *TankContext) GetControllerMap() map[string]core.IController { |
|||
return this.ControllerMap |
|||
} |
|||
|
|||
func (this *TankContext) Cleanup() { |
|||
for _, bean := range this.BeanMap { |
|||
bean.Cleanup() |
|||
} |
|||
} |
|||
|
|||
//响应http的能力
|
|||
func (this *TankContext) ServeHTTP(writer http.ResponseWriter, request *http.Request) { |
|||
this.Router.ServeHTTP(writer, request) |
|||
} |
|||
|
|||
func (this *TankContext) OpenDb() { |
|||
|
|||
var err error = nil |
|||
this.db, err = gorm.Open("mysql", core.CONFIG.GetMysqlUrl()) |
|||
|
|||
if err != nil { |
|||
core.LOGGER.Panic("failed to connect mysql database") |
|||
} |
|||
|
|||
//是否打开sql日志(在调试阶段可以打开，以方便查看执行的SQL)
|
|||
this.db.LogMode(false) |
|||
} |
|||
|
|||
func (this *TankContext) CloseDb() { |
|||
|
|||
if this.db != nil { |
|||
err := this.db.Close() |
|||
if err != nil { |
|||
core.LOGGER.Error("关闭数据库连接出错 %s", err.Error()) |
|||
} |
|||
} |
|||
} |
|||
|
|||
//注册一个Bean
|
|||
func (this *TankContext) registerBean(bean core.IBean) { |
|||
|
|||
typeOf := reflect.TypeOf(bean) |
|||
typeName := typeOf.String() |
|||
|
|||
if element, ok := bean.(core.IBean); ok { |
|||
|
|||
err := fmt.Sprintf("【%s】已经被注册了，跳过。", typeName) |
|||
if _, ok := this.BeanMap[typeName]; ok { |
|||
core.LOGGER.Error(fmt.Sprintf(err)) |
|||
} else { |
|||
this.BeanMap[typeName] = element |
|||
|
|||
//看看是不是controller类型，如果是，那么单独放在ControllerMap中。
|
|||
if controller, ok1 := bean.(core.IController); ok1 { |
|||
this.ControllerMap[typeName] = controller |
|||
} |
|||
|
|||
} |
|||
|
|||
} else { |
|||
core.LOGGER.Panic("注册的【%s】不是Bean类型。", typeName) |
|||
} |
|||
|
|||
} |
|||
|
|||
//注册各个Beans
|
|||
func (this *TankContext) registerBeans() { |
|||
|
|||
//alien
|
|||
this.registerBean(new(rest.AlienController)) |
|||
this.registerBean(new(rest.AlienService)) |
|||
|
|||
//dashboard
|
|||
this.registerBean(new(rest.DashboardController)) |
|||
this.registerBean(new(rest.DashboardDao)) |
|||
this.registerBean(new(rest.DashboardService)) |
|||
|
|||
//downloadToken
|
|||
this.registerBean(new(rest.DownloadTokenDao)) |
|||
|
|||
//imageCache
|
|||
this.registerBean(new(rest.ImageCacheController)) |
|||
this.registerBean(new(rest.ImageCacheDao)) |
|||
this.registerBean(new(rest.ImageCacheService)) |
|||
|
|||
//install
|
|||
this.registerBean(new(rest.InstallController)) |
|||
|
|||
//matter
|
|||
this.registerBean(new(rest.MatterController)) |
|||
this.registerBean(new(rest.MatterDao)) |
|||
this.registerBean(new(rest.MatterService)) |
|||
|
|||
//preference
|
|||
this.registerBean(new(rest.PreferenceController)) |
|||
this.registerBean(new(rest.PreferenceDao)) |
|||
this.registerBean(new(rest.PreferenceService)) |
|||
|
|||
//footprint
|
|||
this.registerBean(new(rest.FootprintController)) |
|||
this.registerBean(new(rest.FootprintDao)) |
|||
this.registerBean(new(rest.FootprintService)) |
|||
|
|||
//session
|
|||
this.registerBean(new(rest.SessionDao)) |
|||
this.registerBean(new(rest.SessionService)) |
|||
|
|||
//uploadToken
|
|||
this.registerBean(new(rest.UploadTokenDao)) |
|||
|
|||
//user
|
|||
this.registerBean(new(rest.UserController)) |
|||
this.registerBean(new(rest.UserDao)) |
|||
this.registerBean(new(rest.UserService)) |
|||
|
|||
//webdav
|
|||
this.registerBean(new(rest.DavController)) |
|||
this.registerBean(new(rest.DavService)) |
|||
|
|||
} |
|||
|
|||
//从Map中获取某个Bean.
|
|||
func (this *TankContext) GetBean(bean core.IBean) core.IBean { |
|||
|
|||
typeOf := reflect.TypeOf(bean) |
|||
typeName := typeOf.String() |
|||
|
|||
if val, ok := this.BeanMap[typeName]; ok { |
|||
return val |
|||
} else { |
|||
core.LOGGER.Panic("【%s】没有注册。", typeName) |
|||
return nil |
|||
} |
|||
} |
|||
|
|||
//初始化每个Bean
|
|||
func (this *TankContext) initBeans() { |
|||
|
|||
for _, bean := range this.BeanMap { |
|||
bean.Init() |
|||
} |
|||
} |
|||
|
|||
//系统如果安装好了就调用这个方法。
|
|||
func (this *TankContext) InstallOk() { |
|||
|
|||
if core.CONFIG.IsInstalled() { |
|||
this.OpenDb() |
|||
|
|||
for _, bean := range this.BeanMap { |
|||
bean.Bootstrap() |
|||
} |
|||
} |
|||
|
|||
} |
|||
|
|||
//销毁的方法
|
|||
func (this *TankContext) Destroy() { |
|||
this.CloseDb() |
|||
} |
|||
|
|||
Loading…
Reference in new issue