lsm/mindoc
1
0
Fork 0
mirror of https://github.com/mindoc-org/mindoc.git synced 2025-10-21 11:27:26 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

实现文档历史功能

Browse Source
This commit is contained in:
Minho
2017-05-25 15:19:17 +08:00
parent 2b1ba118c4
commit 1fdb2c5df6
16 changed files with 491 additions and 190 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
commands/install.go
View File

@@ -8,6 +8,7 @@ import (
"github.com/astaxie/beego/orm"
"github.com/lifei6671/godoc/conf"
"github.com/lifei6671/godoc/models"
"io/ioutil"
)
//系统安装.
@@ -32,26 +33,15 @@ func initialization() {
o := orm.NewOrm()
_, err := o.Raw(`INSERT INTO md_options (option_title, option_name, option_value) SELECT '是否启用注册','ENABLED_REGISTER','false' WHERE NOT exists(SELECT * FROM md_options WHERE option_name = 'ENABLED_REGISTER');`).Exec()
b,err := ioutil.ReadFile("./data/data.sql")
if err != nil {
panic("ENABLED_REGISTER => " + err.Error())
panic(err.Error())
os.Exit(1)
}
_, err = o.Raw(`INSERT INTO md_options (option_title, option_name, option_value) SELECT '是否启用验证码','ENABLED_CAPTCHA','false' WHERE NOT exists(SELECT * FROM md_options WHERE option_name = 'ENABLED_CAPTCHA');`).Exec()
if err != nil {
panic("ENABLED_CAPTCHA => " + err.Error())
os.Exit(1)
}
_, err = o.Raw(`INSERT INTO md_options (option_title, option_name, option_value) SELECT '启用匿名访问','ENABLE_ANONYMOUS','true' WHERE NOT exists(SELECT * FROM md_options WHERE option_name = 'ENABLE_ANONYMOUS');`).Exec()
if err != nil {
panic("ENABLE_ANONYMOUS => " + err.Error())
os.Exit(1)
}
_, err = o.Raw(`INSERT INTO md_options (option_title, option_name, option_value) SELECT '站点名称','SITE_NAME','MinDoc' WHERE NOT exists(SELECT * FROM md_options WHERE option_name = 'SITE_NAME');`).Exec()
sql := string(b)
_,err = o.Raw(sql).Exec()
if err != nil {
panic("SITE_NAME => " + err.Error())
os.Exit(1)

46
commands/update.go
View File

@@ -9,12 +9,20 @@ import (
"github.com/astaxie/beego"
"github.com/lifei6671/godoc/conf"
"github.com/astaxie/beego/orm"
)
//系统升级.
func Update() {
if len(os.Args) >= 2 && os.Args[1] == "update" {
adapter := beego.AppConfig.String("db_adapter")
if adapter == "mysql" {
mysqlUpdate()
}else if adapter == "sqlite3" {
sqliteUpdate()
}
fmt.Println("update successed.")
os.Exit(0)
@@ -55,3 +63,41 @@ func CheckUpdate() {
os.Exit(0)
}
}
//MySQL 数据库更新表结构.
func mysqlUpdate() {
sql := `
IF NOT EXISTS (SELECT * FROM information_schema.columns WHERE table_schema=CurrentDatabase AND table_name = 'md_members' AND column_name = 'auth_method') THEN
ALTER TABLE md_members ADD auth_method VARCHAR(50) DEFAULT 'local' NULL;
END IF; `
o := orm.NewOrm()
_,err := o.Raw(sql).Exec()
if err != nil {
panic(fmt.Sprintf("error : 6001 => %s",err.Error()))
os.Exit(1)
}
}
//sqlite 数据库更新表结构.
func sqliteUpdate() {
o := orm.NewOrm()
var sqlite_master struct{
Name string
}
err := o.Raw("select * from sqlite_master where name='md_members' and sql like '%auth_method%' limit 1").QueryRow(&sqlite_master)
//查询是否已经存在 auth_method 列
if err == nil && sqlite_master.Name == ""{
_,err = o.Raw("ALTER TABLE md_members ADD auth_method VARCHAR(50) DEFAULT 'local' NULL;").Exec()
if err != nil {
panic(fmt.Sprintf("error : 6001 => %s",err.Error()))
os.Exit(1)
}
}
os.Exit(1)
}