forked from hidu/mysql-schema-sync
/
db.go
88 lines (80 loc) · 1.8 KB
/
db.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
package internal
import (
"database/sql"
"fmt"
"log"
_ "github.com/go-sql-driver/mysql" // mysql driver
)
// MyDb db struct
type MyDb struct {
Db *sql.DB
dbType string
}
// NewMyDb parse dsn
func NewMyDb(dsn string, dbType string) *MyDb {
db, err := sql.Open("mysql", dsn)
if err != nil {
panic(fmt.Sprintf("connected to db [%s] failed,err=%s", dsn, err))
}
return &MyDb{
Db: db,
dbType: dbType,
}
}
// GetTableNames table names
func (db *MyDb) GetTableNames() []string {
rs, err := db.Query("show table status")
if err != nil {
panic("show tables failed:" + err.Error())
}
defer rs.Close()
var tables []string
columns, _ := rs.Columns()
for rs.Next() {
var values = make([]any, len(columns))
valuePtrs := make([]any, len(columns))
for i := range columns {
valuePtrs[i] = &values[i]
}
if err := rs.Scan(valuePtrs...); err != nil {
panic("show tables failed when scan," + err.Error())
}
var valObj = make(map[string]any)
for i, col := range columns {
var v any
val := values[i]
b, ok := val.([]byte)
if ok {
v = string(b)
} else {
v = val
}
valObj[col] = v
}
if valObj["Engine"] != nil {
tables = append(tables, valObj["Name"].(string))
}
}
return tables
}
// GetTableSchema table schema
func (db *MyDb) GetTableSchema(name string) (schema string) {
rs, err := db.Query(fmt.Sprintf("show create table `%s`", name))
if err != nil {
log.Println(err)
return
}
defer rs.Close()
for rs.Next() {
var vname string
if err := rs.Scan(&vname, &schema); err != nil {
panic(fmt.Sprintf("get table %s 's schema failed, %s", name, err))
}
}
return
}
// Query execute sql query
func (db *MyDb) Query(query string, args ...any) (*sql.Rows, error) {
log.Println("[SQL]", "["+db.dbType+"]", query, args)
return db.Db.Query(query, args...)
}