test passed

This commit is contained in:
yedongfu 2021-06-03 14:43:33 +08:00
parent 4f1939c7ac
commit 711823fc31
2 changed files with 41 additions and 0 deletions

6
common/conf.sample Normal file
View File

@ -0,0 +1,6 @@
mysql:
host: 'xxx'
user: 'xxx'
password: 'xxx'
database: 'xxx'
port: '3306'

35
examples/data.go Normal file
View File

@ -0,0 +1,35 @@
package examples
import (
"io/ioutil"
"strings"
"github.com/sirupsen/logrus"
"github.com/yedf/dtm/common"
)
func RunSqlScript(mysql map[string]string, script string) {
conf := map[string]string{}
common.MustRemarshal(mysql, &conf)
conf["database"] = ""
db, con := common.DbAlone(conf)
defer func() { con.Close() }()
content, err := ioutil.ReadFile(script)
if err != nil {
e2p(err)
}
sqls := strings.Split(string(content), ";")
for _, sql := range sqls {
s := strings.TrimSpace(sql)
if strings.Contains(strings.ToLower(s), "drop") || s == "" {
continue
}
logrus.Printf("executing: '%s'", s)
db.Must().Exec(s)
}
}
func PopulateMysql() {
common.InitApp(common.GetCurrentPath(), &Config)
RunSqlScript(Config.Mysql, common.GetCurrentPath()+"/examples.sql")
}