TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-06 01:50:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

实现数据库版本升级命令

Browse Source
This commit is contained in:
刘祥超
2020-10-13 20:10:42 +08:00
parent 0e3330e1c6
commit 9ab6ac055a
2 changed files with 32 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
internal/setup/sql_executor.go
View File

@@ -5,11 +5,14 @@ import (
"github.com/TeaOSLab/EdgeAPI/internal/errors"
"github.com/TeaOSLab/EdgeAPI/internal/setup/sqls"
_ "github.com/go-sql-driver/mysql"
"github.com/go-yaml/yaml"
"github.com/iwind/TeaGo/Tea"
"github.com/iwind/TeaGo/dbs"
"github.com/iwind/TeaGo/lists"
"github.com/iwind/TeaGo/rands"
"github.com/iwind/TeaGo/types"
stringutil "github.com/iwind/TeaGo/utils/string"
"io/ioutil"
"sort"
"strings"
"time"
@@ -25,6 +28,20 @@ func NewSQLExecutor(dbConfig *dbs.DBConfig) *SQLExecutor {
}
}
func NewSQLExecutorFromCmd() (*SQLExecutor, error) {
// 执行SQL
config := &dbs.Config{}
configData, err := ioutil.ReadFile(Tea.ConfigFile("db.yaml"))
if err != nil {
return nil, err
}
err = yaml.Unmarshal(configData, config)
if err != nil {
return nil, err
}
return NewSQLExecutor(config.DBs[Tea.Env]), nil
}
func (this *SQLExecutor) Run() error {
db, err := dbs.NewInstanceFromConfig(this.dbConfig)
if err != nil {