golang/hotime
4
0
Fork 1
Code Issues Pull Requests Releases 66 Wiki Activity

Browse Source
This commit is contained in:
hoteas 2017-08-16 09:38:39 +00:00
parent feb78890a4
commit 03347e295c
1 changed files with 28 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
app.go
View File

@ -15,7 +15,7 @@ import (
type App struct { type App struct {
Router Router
Error Error
Port string //端口号 Port string //端口号
connectListener func(context Context) bool //所有的访问监听,true按原计划继续使用，false表示有监听器处理 connectListener func(context Context) bool //所有的访问监听,true按原计划继续使用，false表示有监听器处理
connectDbFunc func(err ...*Error) *sql.DB connectDbFunc func(err ...*Error) *sql.DB
configPath string configPath string
@ -30,7 +30,7 @@ type App struct {
func (this *App) Run(router Router) { func (this *App) Run(router Router) {
this.Router = router this.Router = router
//this.Port = port //this.Port = port
this.Port=this.Config.GetString("port") this.Port = this.Config.GetString("port")
if this.connectDbFunc != nil { if this.connectDbFunc != nil {
this.Db.SetConnect(this.connectDbFunc) this.Db.SetConnect(this.connectDbFunc)
@ -105,27 +105,38 @@ func (this *App) SetConfig(configPath ...string) {
for k, v := range cmap { for k, v := range cmap {
this.Config.Put(k, v) this.Config.Put(k, v)
} }
} } else {
//文件如果损坏则不写入配置防止配置文件数据丢失 //文件不存在则写入文件，存在不做处理
if this.Error.GetError() == nil {
var out bytes.Buffer var out bytes.Buffer
err = json.Indent(&out, []byte(this.Config.ToJsonString()), "", "\t") err = json.Indent(&out, []byte(this.Config.ToJsonString()), "", "\t")
os.MkdirAll(filepath.Dir(this.configPath), os.ModeDir)
os.Create(this.configPath)
err = ioutil.WriteFile(this.configPath, out.Bytes(), os.ModeAppend) err = ioutil.WriteFile(this.configPath, out.Bytes(), os.ModeAppend)
if err != nil { if err != nil {
os.MkdirAll(filepath.Dir(this.configPath), os.ModeDir) this.Error.SetError(err)
os.Create(this.configPath)
err = ioutil.WriteFile(this.configPath, out.Bytes(), os.ModeAppend)
if err != nil {
this.Error.SetError(err)
}
} }
} else {
this.Error.SetError(nil)
} }
//文件如果损坏则不写入配置防止配置文件数据丢失
//if this.Error.GetError() == nil {
// var out bytes.Buffer
//
// err = json.Indent(&out, []byte(this.Config.ToJsonString()), "", "\t")
//
// err = ioutil.WriteFile(this.configPath, out.Bytes(), os.ModeAppend)
//
// if err != nil {
// os.MkdirAll(filepath.Dir(this.configPath), os.ModeDir)
// os.Create(this.configPath)
// err = ioutil.WriteFile(this.configPath, out.Bytes(), os.ModeAppend)
//
// if err != nil {
// this.Error.SetError(err)
// }
// }
//
//} else {
// this.Error.SetError(nil)
//}
} }