-
Notifications
You must be signed in to change notification settings - Fork 0
/
mysql_export_db_data.go
166 lines (155 loc) · 4.49 KB
/
mysql_export_db_data.go
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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
package mysql
import (
"context"
"database/sql"
"fmt"
"io"
"os"
"path/filepath"
"runtime"
"strings"
"time"
"github.com/admpub/errors"
"github.com/webx-top/com"
"github.com/webx-top/db"
"github.com/admpub/nging/v5/application/library/config"
"github.com/admpub/nging/v5/application/library/notice"
"github.com/nging-plugins/dbmanager/application/library/dbmanager/driver"
)
// exportDBData 导出表数据s
func (m *mySQL) exportDBData(ctx context.Context, noticer notice.Noticer,
cfg *driver.DbAuth, tables []string, dataWriter interface{}, mysqlVersion string) error {
if !com.InSlice(cfg.Charset, Charsets) {
return fmt.Errorf(`字符集charset值无效: %v`, cfg.Charset)
}
var (
w io.Writer
err error
)
switch v := dataWriter.(type) {
case io.Writer:
w = v
case string:
dir := filepath.Dir(v)
err = com.MkdirAll(dir, os.ModePerm)
if err != nil {
return fmt.Errorf(`failed to backup: %v`, err)
}
w, err = os.Create(v)
if err != nil {
return fmt.Errorf(`failed to backup: %v`, err)
}
default:
return errors.Wrapf(db.ErrUnsupported, `SQL Writer Error: %T`, v)
}
var (
selectFuncs []string
selectCols []string
wheres []string
orderFields []string
descs []string
page int = 1
limit int = -1 // 不限数量
exportStyle string
)
_, err = w.Write([]byte(`-- Nging DBManager MySQL dump ` + config.Version.Number + `, for ` + runtime.GOOS + ` (` + runtime.GOARCH + `)
--
-- Host: ` + cfg.Host + ` Database: ` + cfg.Db + `
-- ------------------------------------------------------
-- Server version ` + mysqlVersion + `
/*!40101 SET @OLD_CHARACTER_SET_CLIENT=@@CHARACTER_SET_CLIENT */;
/*!40101 SET @OLD_CHARACTER_SET_RESULTS=@@CHARACTER_SET_RESULTS */;
/*!40101 SET @OLD_COLLATION_CONNECTION=@@COLLATION_CONNECTION */;
/*!50503 SET NAMES ` + cfg.Charset + ` */;
/*!40103 SET @OLD_TIME_ZONE=@@TIME_ZONE */;
/*!40103 SET TIME_ZONE='+00:00' */;
/*!40014 SET @OLD_UNIQUE_CHECKS=@@UNIQUE_CHECKS, UNIQUE_CHECKS=0 */;
/*!40014 SET @OLD_FOREIGN_KEY_CHECKS=@@FOREIGN_KEY_CHECKS, FOREIGN_KEY_CHECKS=0 */;
/*!40101 SET @OLD_SQL_MODE=@@SQL_MODE, SQL_MODE='NO_AUTO_VALUE_ON_ZERO' */;
/*!40111 SET @OLD_SQL_NOTES=@@SQL_NOTES, SQL_NOTES=0 */;
`))
if err != nil {
return err
}
for _, table := range tables {
fields, _, err := m.tableFields(table)
if err != nil {
return err
}
var (
insert string
suffix string
hasValues bool
)
_, _, _, err = m.listData(func(cols []string, row map[string]*sql.NullString) error {
if len(insert) == 0 {
keys := make([]string, len(cols))
vals := make([]string, len(cols))
for idx, key := range cols {
key = quoteCol(key)
keys[idx] = key
vals[idx] = key + " = VALUES(" + key + ")"
}
if exportStyle == `INSERT+UPDATE` {
suffix = "\nON DUPLICATE KEY UPDATE " + strings.Join(vals, ", ")
} else {
suffix = ""
}
suffix += ";\n"
insert = "INSERT INTO " + quoteCol(table) + " (" + strings.Join(keys, `, `) + ") VALUES"
_, err = w.Write(com.Str2bytes(insert))
if err != nil {
return err
}
}
var values, sep string
for _, col := range cols {
val := row[col]
if !val.Valid {
values += sep + `NULL`
} else {
field, ok := fields[col]
var v string
if ok && reFieldTypeNumber.MatchString(field.Type) && len(val.String) > 0 && !strings.HasPrefix(field.Full_type, `[`) {
v = val.String
} else {
v = field.Format(val.String)
v = quoteVal(v)
v = com.AddRSlashes(v)
}
values += sep + unconvertField(field, v)
}
sep = `, `
}
s := "\n(" + values + ")"
if !hasValues {
s = insert + s
hasValues = true
} else {
s = "," + s
}
_, err = w.Write(com.Str2bytes(s))
return err
}, table, selectFuncs, selectCols, wheres, orderFields, descs, page, limit, 0, false)
if err == nil && hasValues {
_, err = w.Write(com.Str2bytes(suffix))
}
if err != nil {
return err
}
}
_, err = w.Write([]byte(`/*!40103 SET TIME_ZONE=@OLD_TIME_ZONE */;
/*!40101 SET SQL_MODE=@OLD_SQL_MODE */;
/*!40014 SET FOREIGN_KEY_CHECKS=@OLD_FOREIGN_KEY_CHECKS */;
/*!40014 SET UNIQUE_CHECKS=@OLD_UNIQUE_CHECKS */;
/*!40101 SET CHARACTER_SET_CLIENT=@OLD_CHARACTER_SET_CLIENT */;
/*!40101 SET CHARACTER_SET_RESULTS=@OLD_CHARACTER_SET_RESULTS */;
/*!40101 SET COLLATION_CONNECTION=@OLD_COLLATION_CONNECTION */;
/*!40111 SET SQL_NOTES=@OLD_SQL_NOTES */;
-- Dump completed on ` + time.Now().Format("2006-01-02 15:04:05") + `
`))
if c, y := w.(io.Closer); y {
c.Close()
}
return err
}