Browse Source

test passed

topic
yedongfu 5 years ago
parent
commit
711823fc31
  1. 6
      common/conf.sample
  2. 35
      examples/data.go

6
common/conf.sample

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

35
examples/data.go

@ -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")
}
Loading…
Cancel
Save