exec.go
1.63 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
package lxmysql
import (
"github.com/astaxie/beego/orm"
"io/ioutil"
"os"
"strings"
)
type ExecSqlConfig struct {
Sql string //表名
TableName string //表名
DBIndex string //指定数据库
Orm orm.Ormer //实例化 ORM
CreaeSql string //创建表的SQL语句
CreatePath string //创建表的SQL文件路径
}
func ExecSql(config ExecSqlConfig) (err error) {
var (
tcount int
)
if config.Orm == nil {
config.Orm = orm.NewOrm()
}
if config.DBIndex == "" {
config.DBIndex = "default"
}
RSTART:
config.Orm.Using(config.DBIndex)
_, err = config.Orm.Raw(config.Sql).Exec()
if err != nil {
if b := strings.Contains(err.Error(), "doesn't exist"); b && (config.CreaeSql != "" || config.CreatePath != "") && tcount == 0 {
if config.CreaeSql == "" {
config.CreaeSql, _ = readFilesql(config.CreatePath, config.TableName)
}
//表不存在 去创建
err = CreateTable(config.CreaeSql, config.Orm, config.DBIndex)
if err != nil {
return
}
tcount++
//重新执行一次
goto RSTART
}
return
}
return
}
func CreateTable(csql string, o orm.Ormer, dbindex string) (err error) {
o.Using(dbindex)
_, err = o.Raw(csql).Exec()
if err != nil {
return
}
return
}
func readFilesql(pth, tbname string) (sql string, err error) {
filePtr, err := os.Open(pth)
if err != nil {
return
}
defer filePtr.Close()
by, err := ioutil.ReadAll(filePtr)
if err != nil {
return
}
sql = strings.Replace(string(by), "{tablename}", tbname, -1)
return
}
func ReadFilesql(pth, tbname string) (sql string, err error) {
pth += ".sql"
return readFilesql(pth, tbname)
}