forked from weichangdong/sync-mysql-schema-data
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
executable file
·104 lines (92 loc) · 2.67 KB
/
main.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
package main
import (
"flag"
"fmt"
"log"
"os"
"runtime"
"strings"
"github.com/cdy1263/sync-mysql-schema-data_1"
)
//user:password@tcp(127.0.0.1:3306)/test
var configPath = flag.String("conf", "./config.json", "json config file path")
var sync = flag.Bool("sync", false, "sync shcema change to dest db")
var drop = flag.Bool("drop", false, "drop fields,index,foreign key")
var source = flag.String("source", "", "mysql dsn source,eg: test@(10.10.0.1:3306)/test\n\twhen it is not empty ignore [-conf] param")
var dest = flag.String("dest", "", "mysql dsn dest,eg test@(127.0.0.1:3306)/imis")
var tables = flag.String("tables", "", "table names to check,equivalent to json config [tables]\n\teg : product_base,order_*")
var tablesIGNORE = flag.String("tables_ignore", "", "table names to ignore check and ignore sync data\n\teg : product_base,order_*")
var mailTo = flag.String("mail_to", "", "overwrite config's email.to")
var syncData = flag.Bool("sync_data", true, "sync source db table data to dest db table")
var syncDataTruncate = flag.Bool("sync_data_truncate", false, "is need truncate source db table data to dest db table")
func init() {
log.SetFlags(log.Lshortfile | log.Ldate)
df := flag.Usage
flag.Usage = func() {
df()
fmt.Fprintln(os.Stderr, "")
fmt.Fprintln(os.Stderr, "mysql schema && data sync tools "+internal.Version)
fmt.Fprintln(os.Stderr, internal.AppURL+"\n")
}
}
var cfg *internal.Config
func main() {
flag.Parse()
if *source == "" {
cfg = internal.LoadConfig(*configPath)
} else {
cfg = new(internal.Config)
cfg.SourceDSN = *source
cfg.DestDSN = *dest
}
cfg.Sync = *sync
cfg.Drop = *drop
cfg.SyncData = *syncData
cfg.SyncDataTruncate = *syncDataTruncate
if *mailTo != "" && cfg.Email != nil {
cfg.Email.To = *mailTo
}
if cfg.Tables == nil {
cfg.Tables = []string{}
}
if cfg.TablesIGNORE == nil {
cfg.TablesIGNORE = []string{}
}
if *tables != "" {
_ts := strings.Split(*tables, ",")
for _, _name := range _ts {
_name = strings.TrimSpace(_name)
if _name != "" {
cfg.Tables = append(cfg.Tables, _name)
}
}
}
if *tablesIGNORE != "" {
_ts := strings.Split(*tablesIGNORE, ",")
for _, _name := range _ts {
_name = strings.TrimSpace(_name)
if _name != "" {
cfg.TablesIGNORE = append(cfg.TablesIGNORE, _name)
}
}
}
defer (func() {
if err := recover(); err != nil {
log.Println(err)
log.Println(fullStack())
cfg.SendMailFail(fmt.Sprintf("%s", err))
log.Fatalln("exit")
}
})()
cfg.Check()
if *syncData == false {
internal.CheckSchemaDiff(cfg)
} else {
internal.SyncTableData(cfg)
}
}
func fullStack() string {
var buf [2 << 11]byte
runtime.Stack(buf[:], true)
return string(buf[:])
}