lsm/mindoc
1
0
Fork 0
mirror of https://github.com/mindoc-org/mindoc.git synced 2025-09-19 18:22:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore!(all): attempt to update beego to v2

Browse Source 
BREAKING CHANGE: beego has update to v2, and this version of mindoc IS
NOT TESTED AND STABLE!!!
This commit is contained in:
roberChen
2021-03-23 21:55:50 +08:00
parent 79f70d830d
commit 90c1329de2
56 changed files with 1021 additions and 766 deletions
Download Patch File Download Diff File Expand all files Collapse all files

191
commands/command.go
View File

@@ -15,13 +15,14 @@ import (
"encoding/json"
"net/http"
"github.com/astaxie/beego"
beegoCache "github.com/astaxie/beego/cache"
_ "github.com/astaxie/beego/cache/memcache"
"github.com/astaxie/beego/cache/redis"
_ "github.com/astaxie/beego/cache/redis"
"github.com/astaxie/beego/logs"
"github.com/astaxie/beego/orm"
"github.com/beego/beego/v2/adapter"
"github.com/beego/beego/v2/adapter/orm"
beegoCache "github.com/beego/beego/v2/client/cache"
_ "github.com/beego/beego/v2/client/cache/memcache"
"github.com/beego/beego/v2/client/cache/redis"
_ "github.com/beego/beego/v2/client/cache/redis"
"github.com/beego/beego/v2/core/logs"
"github.com/beego/beego/v2/server/web"
"github.com/howeyc/fsnotify"
"github.com/lifei6671/gocaptcha"
"github.com/mindoc-org/mindoc/cache"
@@ -32,37 +33,37 @@ import (
// RegisterDataBase 注册数据库
func RegisterDataBase() {
beego.Info("正在初始化数据库配置.")
adapter := beego.AppConfig.String("db_adapter")
logs.Info("正在初始化数据库配置.")
dbadapter := adapter.AppConfig.String("db_adapter")
orm.DefaultTimeLoc = time.Local
orm.DefaultRowsLimit = -1
if strings.EqualFold(adapter, "mysql") {
host := beego.AppConfig.String("db_host")
database := beego.AppConfig.String("db_database")
username := beego.AppConfig.String("db_username")
password := beego.AppConfig.String("db_password")
if strings.EqualFold(dbadapter, "mysql") {
host := adapter.AppConfig.String("db_host")
database := adapter.AppConfig.String("db_database")
username := adapter.AppConfig.String("db_username")
password := adapter.AppConfig.String("db_password")
timezone := beego.AppConfig.String("timezone")
timezone := adapter.AppConfig.String("timezone")
location, err := time.LoadLocation(timezone)
if err == nil {
orm.DefaultTimeLoc = location
} else {
beego.Error("加载时区配置信息失败,请检查是否存在 ZONEINFO 环境变量->", err)
logs.Error("加载时区配置信息失败,请检查是否存在 ZONEINFO 环境变量->", err)
}
port := beego.AppConfig.String("db_port")
port := adapter.AppConfig.String("db_port")
dataSource := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=true&loc=%s", username, password, host, port, database, url.QueryEscape(timezone))
if err := orm.RegisterDataBase("default", "mysql", dataSource); err != nil {
beego.Error("注册默认数据库失败->", err)
logs.Error("注册默认数据库失败->", err)
os.Exit(1)
}
} else if strings.EqualFold(adapter, "sqlite3") {
} else if strings.EqualFold(dbadapter, "sqlite3") {
database := beego.AppConfig.String("db_database")
database := adapter.AppConfig.String("db_database")
if strings.HasPrefix(database, "./") {
database = filepath.Join(conf.WorkingDirectory, string(database[1:]))
}
@@ -79,13 +80,13 @@ func RegisterDataBase() {
err := orm.RegisterDataBase("default", "sqlite3", database)
if err != nil {
beego.Error("注册默认数据库失败->", err)
logs.Error("注册默认数据库失败->", err)
}
} else {
beego.Error("不支持的数据库类型.")
logs.Error("不支持的数据库类型.")
os.Exit(1)
}
beego.Info("数据库初始化完成.")
logs.Info("数据库初始化完成.")
}
// RegisterModel 注册Model
@@ -122,11 +123,11 @@ func RegisterLogger(log string) {
_ = logs.SetLogger("console")
logs.EnableFuncCallDepth(true)
if beego.AppConfig.DefaultBool("log_is_async", true) {
if adapter.AppConfig.DefaultBool("log_is_async", true) {
logs.Async(1e3)
}
if log == "" {
logPath, err := filepath.Abs(beego.AppConfig.DefaultString("log_path", conf.WorkingDir("runtime", "logs")))
logPath, err := filepath.Abs(adapter.AppConfig.DefaultString("log_path", conf.WorkingDir("runtime", "logs")))
if err == nil {
log = logPath
} else {
@@ -146,55 +147,55 @@ func RegisterLogger(log string) {
config["perm"] = "0755"
config["rotate"] = true
if maxLines := beego.AppConfig.DefaultInt("log_maxlines", 1000000); maxLines > 0 {
if maxLines := adapter.AppConfig.DefaultInt("log_maxlines", 1000000); maxLines > 0 {
config["maxLines"] = maxLines
}
if maxSize := beego.AppConfig.DefaultInt("log_maxsize", 1<<28); maxSize > 0 {
if maxSize := adapter.AppConfig.DefaultInt("log_maxsize", 1<<28); maxSize > 0 {
config["maxsize"] = maxSize
}
if !beego.AppConfig.DefaultBool("log_daily", true) {
if !adapter.AppConfig.DefaultBool("log_daily", true) {
config["daily"] = false
}
if maxDays := beego.AppConfig.DefaultInt("log_maxdays", 7); maxDays > 0 {
if maxDays := adapter.AppConfig.DefaultInt("log_maxdays", 7); maxDays > 0 {
config["maxdays"] = maxDays
}
if level := beego.AppConfig.DefaultString("log_level", "Trace"); level != "" {
if level := adapter.AppConfig.DefaultString("log_level", "Trace"); level != "" {
switch level {
case "Emergency":
config["level"] = beego.LevelEmergency
config["level"] = logs.LevelEmergency
break
case "Alert":
config["level"] = beego.LevelAlert
config["level"] = logs.LevelAlert
break
case "Critical":
config["level"] = beego.LevelCritical
config["level"] = logs.LevelCritical
break
case "Error":
config["level"] = beego.LevelError
config["level"] = logs.LevelError
break
case "Warning":
config["level"] = beego.LevelWarning
config["level"] = logs.LevelWarning
break
case "Notice":
config["level"] = beego.LevelNotice
config["level"] = logs.LevelNotice
break
case "Informational":
config["level"] = beego.LevelInformational
config["level"] = logs.LevelInformational
break
case "Debug":
config["level"] = beego.LevelDebug
config["level"] = logs.LevelDebug
break
}
}
b, err := json.Marshal(config)
if err != nil {
beego.Error("初始化文件日志时出错 ->", err)
_ = beego.SetLogger("file", `{"filename":"`+logPath+`"}`)
logs.Error("初始化文件日志时出错 ->", err)
_ = logs.SetLogger("file", `{"filename":"`+logPath+`"}`)
} else {
_ = beego.SetLogger(logs.AdapterFile, string(b))
_ = logs.SetLogger(logs.AdapterFile, string(b))
}
beego.SetLogFuncCall(true)
logs.SetLogFuncCall(true)
}
// RunCommand 注册orm命令行工具
@@ -212,20 +213,20 @@ func RegisterCommand() {
//注册模板函数
func RegisterFunction() {
err := beego.AddFuncMap("config", models.GetOptionValue)
err := adapter.AddFuncMap("config", models.GetOptionValue)
if err != nil {
beego.Error("注册函数 config 出错 ->", err)
logs.Error("注册函数 config 出错 ->", err)
os.Exit(-1)
}
err = beego.AddFuncMap("cdn", func(p string) string {
cdn := beego.AppConfig.DefaultString("cdn", "")
err = adapter.AddFuncMap("cdn", func(p string) string {
cdn := adapter.AppConfig.DefaultString("cdn", "")
if strings.HasPrefix(p, "http://") || strings.HasPrefix(p, "https://") {
return p
}
//如果没有设置cdn，则使用baseURL拼接
if cdn == "" {
baseUrl := beego.AppConfig.DefaultString("baseurl", "")
baseUrl := adapter.AppConfig.DefaultString("baseurl", "")
if strings.HasPrefix(p, "/") && strings.HasSuffix(baseUrl, "/") {
return baseUrl + p[1:]
@@ -244,36 +245,36 @@ func RegisterFunction() {
return cdn + p
})
if err != nil {
beego.Error("注册函数 cdn 出错 ->", err)
logs.Error("注册函数 cdn 出错 ->", err)
os.Exit(-1)
}
err = beego.AddFuncMap("cdnjs", conf.URLForWithCdnJs)
err = adapter.AddFuncMap("cdnjs", conf.URLForWithCdnJs)
if err != nil {
beego.Error("注册函数 cdnjs 出错 ->", err)
logs.Error("注册函数 cdnjs 出错 ->", err)
os.Exit(-1)
}
err = beego.AddFuncMap("cdncss", conf.URLForWithCdnCss)
err = adapter.AddFuncMap("cdncss", conf.URLForWithCdnCss)
if err != nil {
beego.Error("注册函数 cdncss 出错 ->", err)
logs.Error("注册函数 cdncss 出错 ->", err)
os.Exit(-1)
}
err = beego.AddFuncMap("cdnimg", conf.URLForWithCdnImage)
err = adapter.AddFuncMap("cdnimg", conf.URLForWithCdnImage)
if err != nil {
beego.Error("注册函数 cdnimg 出错 ->", err)
logs.Error("注册函数 cdnimg 出错 ->", err)
os.Exit(-1)
}
//重写url生成，支持配置域名以及域名前缀
err = beego.AddFuncMap("urlfor", conf.URLFor)
err = adapter.AddFuncMap("urlfor", conf.URLFor)
if err != nil {
beego.Error("注册函数 urlfor 出错 ->", err)
logs.Error("注册函数 urlfor 出错 ->", err)
os.Exit(-1)
}
err = beego.AddFuncMap("date_format", func(t time.Time, format string) string {
err = adapter.AddFuncMap("date_format", func(t time.Time, format string) string {
return t.Local().Format(format)
})
if err != nil {
beego.Error("注册函数 date_format 出错 ->", err)
logs.Error("注册函数 date_format 出错 ->", err)
os.Exit(-1)
}
}
@@ -306,11 +307,11 @@ func ResolveCommand(args []string) {
log.Fatal("读取字体文件时出错 -> ", err)
}
if err := beego.LoadAppConfig("ini", conf.ConfigurationFile); err != nil {
if err := adapter.LoadAppConfig("ini", conf.ConfigurationFile); err != nil {
log.Fatal("An error occurred:", err)
}
if conf.LogFile == "" {
logPath, err := filepath.Abs(beego.AppConfig.DefaultString("log_path", conf.WorkingDir("runtime", "logs")))
logPath, err := filepath.Abs(adapter.AppConfig.DefaultString("log_path", conf.WorkingDir("runtime", "logs")))
if err == nil {
conf.LogFile = logPath
} else {
@@ -318,14 +319,14 @@ func ResolveCommand(args []string) {
}
}
conf.AutoLoadDelay = beego.AppConfig.DefaultInt("config_auto_delay", 0)
conf.AutoLoadDelay = adapter.AppConfig.DefaultInt("config_auto_delay", 0)
uploads := conf.WorkingDir("uploads")
_ = os.MkdirAll(uploads, 0666)
beego.BConfig.WebConfig.StaticDir["/static"] = filepath.Join(conf.WorkingDirectory, "static")
beego.BConfig.WebConfig.StaticDir["/uploads"] = uploads
beego.BConfig.WebConfig.ViewsPath = conf.WorkingDir("views")
web.BConfig.WebConfig.StaticDir["/static"] = filepath.Join(conf.WorkingDirectory, "static")
web.BConfig.WebConfig.StaticDir["/uploads"] = uploads
web.BConfig.WebConfig.ViewsPath = conf.WorkingDir("views")
fonts := conf.WorkingDir("static", "fonts")
@@ -347,15 +348,15 @@ func ResolveCommand(args []string) {
//注册缓存管道
func RegisterCache() {
isOpenCache := beego.AppConfig.DefaultBool("cache", false)
isOpenCache := adapter.AppConfig.DefaultBool("cache", false)
if !isOpenCache {
cache.Init(&cache.NullCache{})
return
}
beego.Info("正常初始化缓存配置.")
cacheProvider := beego.AppConfig.String("cache_provider")
logs.Info("正常初始化缓存配置.")
cacheProvider := adapter.AppConfig.String("cache_provider")
if cacheProvider == "file" {
cacheFilePath := beego.AppConfig.DefaultString("cache_file_path", "./runtime/cache/")
cacheFilePath := adapter.AppConfig.DefaultString("cache_file_path", "./runtime/cache/")
if strings.HasPrefix(cacheFilePath, "./") {
cacheFilePath = filepath.Join(conf.WorkingDirectory, string(cacheFilePath[1:]))
}
@@ -364,13 +365,13 @@ func RegisterCache() {
fileConfig := make(map[string]string, 0)
fileConfig["CachePath"] = cacheFilePath
fileConfig["DirectoryLevel"] = beego.AppConfig.DefaultString("cache_file_dir_level", "2")
fileConfig["EmbedExpiry"] = beego.AppConfig.DefaultString("cache_file_expiry", "120")
fileConfig["FileSuffix"] = beego.AppConfig.DefaultString("cache_file_suffix", ".bin")
fileConfig["DirectoryLevel"] = adapter.AppConfig.DefaultString("cache_file_dir_level", "2")
fileConfig["EmbedExpiry"] = adapter.AppConfig.DefaultString("cache_file_expiry", "120")
fileConfig["FileSuffix"] = adapter.AppConfig.DefaultString("cache_file_suffix", ".bin")
bc, err := json.Marshal(&fileConfig)
if err != nil {
beego.Error("初始化file缓存失败:", err)
logs.Error("初始化file缓存失败:", err)
os.Exit(1)
}
@@ -379,13 +380,13 @@ func RegisterCache() {
cache.Init(fileCache)
} else if cacheProvider == "memory" {
cacheInterval := beego.AppConfig.DefaultInt("cache_memory_interval", 60)
cacheInterval := adapter.AppConfig.DefaultInt("cache_memory_interval", 60)
memory := beegoCache.NewMemoryCache()
beegoCache.DefaultEvery = cacheInterval
cache.Init(memory)
} else if cacheProvider == "redis" {
//设置Redis前缀
if key := beego.AppConfig.DefaultString("cache_redis_prefix", ""); key != "" {
if key := adapter.AppConfig.DefaultString("cache_redis_prefix", ""); key != "" {
redis.DefaultKey = key
}
var redisConfig struct {
@@ -394,23 +395,23 @@ func RegisterCache() {
DbNum int `json:"dbNum"`
}
redisConfig.DbNum = 0
redisConfig.Conn = beego.AppConfig.DefaultString("cache_redis_host", "")
if pwd := beego.AppConfig.DefaultString("cache_redis_password", ""); pwd != "" {
redisConfig.Conn = adapter.AppConfig.DefaultString("cache_redis_host", "")
if pwd := adapter.AppConfig.DefaultString("cache_redis_password", ""); pwd != "" {
redisConfig.Password = pwd
}
if dbNum := beego.AppConfig.DefaultInt("cache_redis_db", 0); dbNum > 0 {
if dbNum := adapter.AppConfig.DefaultInt("cache_redis_db", 0); dbNum > 0 {
redisConfig.DbNum = dbNum
}
bc, err := json.Marshal(&redisConfig)
if err != nil {
beego.Error("初始化Redis缓存失败:", err)
logs.Error("初始化Redis缓存失败:", err)
os.Exit(1)
}
redisCache, err := beegoCache.NewCache("redis", string(bc))
if err != nil {
beego.Error("初始化Redis缓存失败:", err)
logs.Error("初始化Redis缓存失败:", err)
os.Exit(1)
}
@@ -420,17 +421,17 @@ func RegisterCache() {
var memcacheConfig struct {
Conn string `json:"conn"`
}
memcacheConfig.Conn = beego.AppConfig.DefaultString("cache_memcache_host", "")
memcacheConfig.Conn = adapter.AppConfig.DefaultString("cache_memcache_host", "")
bc, err := json.Marshal(&memcacheConfig)
if err != nil {
beego.Error("初始化 Memcache 缓存失败 ->", err)
logs.Error("初始化 Memcache 缓存失败 ->", err)
os.Exit(1)
}
memcache, err := beegoCache.NewCache("memcache", string(bc))
if err != nil {
beego.Error("初始化 Memcache 缓存失败 ->", err)
logs.Error("初始化 Memcache 缓存失败 ->", err)
os.Exit(1)
}
@@ -438,10 +439,10 @@ func RegisterCache() {
} else {
cache.Init(&cache.NullCache{})
beego.Warn("不支持的缓存管道,缓存将禁用 ->", cacheProvider)
logs.Warn("不支持的缓存管道,缓存将禁用 ->", cacheProvider)
return
}
beego.Info("缓存初始化完成.")
logs.Info("缓存初始化完成.")
}
//自动加载配置文件.修改了监听端口号和数据库配置无法自动生效.
@@ -451,7 +452,7 @@ func RegisterAutoLoadConfig() {
watcher, err := fsnotify.NewWatcher()
if err != nil {
beego.Error("创建配置文件监控器失败 ->", err)
logs.Error("创建配置文件监控器失败 ->", err)
}
go func() {
for {
@@ -459,19 +460,19 @@ func RegisterAutoLoadConfig() {
case ev := <-watcher.Event:
//如果是修改了配置文件
if ev.IsModify() {
if err := beego.LoadAppConfig("ini", conf.ConfigurationFile); err != nil {
beego.Error("An error occurred ->", err)
if err := adapter.LoadAppConfig("ini", conf.ConfigurationFile); err != nil {
logs.Error("An error occurred ->", err)
continue
}
RegisterCache()
RegisterLogger("")
beego.Info("配置文件已加载 ->", conf.ConfigurationFile)
logs.Info("配置文件已加载 ->", conf.ConfigurationFile)
} else if ev.IsRename() {
_ = watcher.WatchFlags(conf.ConfigurationFile, fsnotify.FSN_MODIFY|fsnotify.FSN_RENAME)
}
beego.Info(ev.String())
logs.Info(ev.String())
case err := <-watcher.Error:
beego.Error("配置文件监控器错误 ->", err)
logs.Error("配置文件监控器错误 ->", err)
}
}
@@ -480,34 +481,34 @@ func RegisterAutoLoadConfig() {
err = watcher.WatchFlags(conf.ConfigurationFile, fsnotify.FSN_MODIFY|fsnotify.FSN_RENAME)
if err != nil {
beego.Error("监控配置文件失败 ->", err)
logs.Error("监控配置文件失败 ->", err)
}
}
}
//注册错误处理方法.
func RegisterError() {
beego.ErrorHandler("404", func(writer http.ResponseWriter, request *http.Request) {
web.ErrorHandler("404", func(writer http.ResponseWriter, request *http.Request) {
var buf bytes.Buffer
data := make(map[string]interface{})
data["ErrorCode"] = 404
data["ErrorMessage"] = "页面未找到或已删除"
if err := beego.ExecuteViewPathTemplate(&buf, "errors/error.tpl", beego.BConfig.WebConfig.ViewsPath, data); err == nil {
if err := adapter.ExecuteViewPathTemplate(&buf, "errors/error.tpl", web.BConfig.WebConfig.ViewsPath, data); err == nil {
_, _ = fmt.Fprint(writer, buf.String())
} else {
_, _ = fmt.Fprint(writer, data["ErrorMessage"])
}
})
beego.ErrorHandler("401", func(writer http.ResponseWriter, request *http.Request) {
web.ErrorHandler("401", func(writer http.ResponseWriter, request *http.Request) {
var buf bytes.Buffer
data := make(map[string]interface{})
data["ErrorCode"] = 401
data["ErrorMessage"] = "请与 Web 服务器的管理员联系，以确认您是否具有访问所请求资源的权限。"
if err := beego.ExecuteViewPathTemplate(&buf, "errors/error.tpl", beego.BConfig.WebConfig.ViewsPath, data); err == nil {
if err := adapter.ExecuteViewPathTemplate(&buf, "errors/error.tpl", web.BConfig.WebConfig.ViewsPath, data); err == nil {
_, _ = fmt.Fprint(writer, buf.String())
} else {
_, _ = fmt.Fprint(writer, data["ErrorMessage"])

34
commands/daemon/daemon.go
View File

@@ -6,7 +6,9 @@ import (
"path/filepath"
"github.com/astaxie/beego"
"github.com/astaxie/beego/logs"
"github.com/beego/beego/v2/adapter"
"github.com/beego/beego/v2/server/web"
"github.com/kardianos/service"
"github.com/mindoc-org/mindoc/commands"
"github.com/mindoc-org/mindoc/conf"
@@ -45,7 +47,6 @@ func (d *Daemon) Start(s service.Service) error {
func (d *Daemon) Run() {
commands.ResolveCommand(d.config.Arguments)
commands.RegisterFunction()
@@ -54,9 +55,14 @@ func (d *Daemon) Run() {
commands.RegisterError()
beego.ErrorController(&controllers.ErrorController{})
/* TODO: which ErrorController to use? currently there are
* two ErrorController avalable, one is beego/v2/server/web.ErrorController,
* and another is beego/v2/adapter.ErrorController, which is actually a calling to the first one
* here the first one is used, which is the original implementation.
*/
web.ErrorController(&controllers.ErrorController{})
f,err := filepath.Abs(os.Args[0])
f, err := filepath.Abs(os.Args[0])
if err != nil {
f = os.Args[0]
@@ -64,7 +70,7 @@ func (d *Daemon) Run() {
fmt.Printf("MinDoc version => %s\nbuild time => %s\nstart directory => %s\n%s\n", conf.VERSION, conf.BUILD_TIME, f, conf.GO_VERSION)
beego.Run()
adapter.Run()
}
func (d *Daemon) Stop(s service.Service) error {
@@ -81,15 +87,15 @@ func Install() {
s, err := service.New(d, d.config)
if err != nil {
beego.Error("Create service error => ", err)
logs.Error("Create service error => ", err)
os.Exit(1)
}
err = s.Install()
if err != nil {
beego.Error("Install service error:", err)
logs.Error("Install service error:", err)
os.Exit(1)
} else {
beego.Info("Service installed!")
logs.Info("Service installed!")
}
os.Exit(0)
@@ -100,15 +106,15 @@ func Uninstall() {
s, err := service.New(d, d.config)
if err != nil {
beego.Error("Create service error => ", err)
logs.Error("Create service error => ", err)
os.Exit(1)
}
err = s.Uninstall()
if err != nil {
beego.Error("Install service error:", err)
logs.Error("Install service error:", err)
os.Exit(1)
} else {
beego.Info("Service uninstalled!")
logs.Info("Service uninstalled!")
}
os.Exit(0)
}
@@ -118,15 +124,15 @@ func Restart() {
s, err := service.New(d, d.config)
if err != nil {
beego.Error("Create service error => ", err)
logs.Error("Create service error => ", err)
os.Exit(1)
}
err = s.Restart()
if err != nil {
beego.Error("Install service error:", err)
logs.Error("Install service error:", err)
os.Exit(1)
} else {
beego.Info("Service Restart!")
logs.Info("Service Restart!")
}
os.Exit(0)
}

19
commands/install.go
View File

@@ -7,8 +7,8 @@ import (
"flag"
"github.com/astaxie/beego"
"github.com/astaxie/beego/orm"
"github.com/astaxie/beego/logs"
"github.com/beego/beego/v2/adapter/orm"
"github.com/mindoc-org/mindoc/conf"
"github.com/mindoc-org/mindoc/models"
"github.com/mindoc-org/mindoc/utils"
@@ -40,7 +40,7 @@ func Version() {
//修改用户密码
func ModifyPassword() {
var account,password string
var account, password string
//账号和密码需要解析参数后才能获取
if len(os.Args) >= 2 && os.Args[1] == "password" {
@@ -50,7 +50,7 @@ func ModifyPassword() {
flagSet.StringVar(&password, "password", "", "用户密码.")
if err := flagSet.Parse(os.Args[2:]); err != nil {
beego.Error("解析参数失败 -> ",err)
logs.Error("解析参数失败 -> ", err)
os.Exit(1)
}
@@ -67,30 +67,29 @@ func ModifyPassword() {
fmt.Println("Password cannot be empty.")
os.Exit(1)
}
member,err := models.NewMember().FindByAccount(account)
member, err := models.NewMember().FindByAccount(account)
if err != nil {
fmt.Println("Failed to change password:",err)
fmt.Println("Failed to change password:", err)
os.Exit(1)
}
pwd,err := utils.PasswordHash(password)
pwd, err := utils.PasswordHash(password)
if err != nil {
fmt.Println("Failed to change password:",err)
fmt.Println("Failed to change password:", err)
os.Exit(1)
}
member.Password = pwd
err = member.Update("password")
if err != nil {
fmt.Println("Failed to change password:",err)
fmt.Println("Failed to change password:", err)
os.Exit(1)
}
fmt.Println("Successfully modified.")
os.Exit(0)
}
}
//初始化数据

13
commands/migrate/migrate.go
View File

@@ -11,6 +11,7 @@
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
// License for the specific language governing permissions and limitations
// under the License.
package migrate
import (
@@ -20,8 +21,8 @@ import (
"fmt"
"log"
"github.com/astaxie/beego"
"github.com/astaxie/beego/orm"
"github.com/beego/beego/v2/adapter"
"github.com/beego/beego/v2/client/orm"
"github.com/mindoc-org/mindoc/models"
)
@@ -113,17 +114,17 @@ func RunMigration() {
//导出数据库的表结构
func ExportDatabaseTable() ([]string, error) {
db_adapter := beego.AppConfig.String("db_adapter")
db_database := beego.AppConfig.String("db_database")
dbadapter := adapter.AppConfig.String("db_adapter")
dbdatabase := adapter.AppConfig.String("db_database")
tables := make([]string, 0)
o := orm.NewOrm()
switch db_adapter {
switch dbadapter {
case "mysql":
{
var lists []orm.Params
_, err := o.Raw(fmt.Sprintf("SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = '%s'", db_database)).Values(&lists)
_, err := o.Raw(fmt.Sprintf("SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = '%s'", dbdatabase)).Values(&lists)
if err != nil {
return tables, err
}

2
commands/migrate/migrate_v03.go
View File

@@ -6,7 +6,7 @@ import (
"strings"
"time"
"github.com/astaxie/beego/orm"
"github.com/beego/beego/v2/adapter/orm"
"github.com/mindoc-org/mindoc/models"
)

8
commands/update.go
View File

@@ -7,7 +7,7 @@ import (
"net/http"
"os"
"github.com/astaxie/beego"
"github.com/astaxie/beego/logs"
"github.com/mindoc-org/mindoc/conf"
)
@@ -17,14 +17,14 @@ func CheckUpdate() {
resp, err := http.Get("https://api.github.com/repos/lifei6671/mindoc/tags")
if err != nil {
beego.Error("CheckUpdate => ", err)
logs.Error("CheckUpdate => ", err)
os.Exit(1)
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
beego.Error("CheckUpdate => ", err)
logs.Error("CheckUpdate => ", err)
os.Exit(1)
}
@@ -35,7 +35,7 @@ func CheckUpdate() {
err = json.Unmarshal(body, &result)
fmt.Println("MinDoc current version => ", conf.VERSION)
if err != nil {
beego.Error("CheckUpdate => ", err)
logs.Error("CheckUpdate => ", err)
os.Exit(0)
}