package serf
|
|
import (
|
"errors"
|
"fmt"
|
"strings"
|
|
"github.com/jinzhu/gorm"
|
)
|
|
type DumpSql struct {
|
Sql string `json:"sql"`
|
}
|
|
type TableDesc struct {
|
Cid int `json:"cid"`
|
Name string `json:"name"`
|
Type string `json:"type"`
|
Notnull bool `json:"notnull"`
|
DFltValue interface{} `json:"dflt_value"`
|
Pk int `json:"pk"`
|
}
|
|
var syncSqlChan = make(chan string, 10)
|
|
func DumpTables(db *gorm.DB, tableNames []string) ([]string, error) {
|
db.LogMode(false)
|
defer db.LogMode(true)
|
|
if tableNames != nil {
|
var arr []string
|
var dumpSql []DumpSql
|
|
for _, table := range tableNames {
|
fmt.Println("dump current tableName:", table)
|
|
dumpSql = make([]DumpSql, 0)
|
var tDescArr []TableDesc
|
|
tSql := fmt.Sprintf(`PRAGMA table_info("%s")`, table)
|
err := db.Raw(tSql).Scan(&tDescArr).Error
|
if err != nil {
|
return nil, errors.New("tableDesc err")
|
}
|
|
if tDescArr == nil || len(tDescArr) == 0 {
|
continue
|
}
|
|
var columnNames []string
|
var columnValues []string
|
for _, col := range tDescArr {
|
columnNames = append(columnNames, col.Name)
|
columnValues = append(columnValues, fmt.Sprintf(`'||quote("%s")||'`, col.Name))
|
}
|
|
tSql = fmt.Sprintf(`SELECT 'INSERT INTO "%s" (%s) VALUES(%s)' as sql FROM "%s";`,
|
table,
|
strings.Join(columnNames, ","),
|
strings.Join(columnValues, ","),
|
table)
|
//fmt.Println("tSql:", tSql)
|
|
err = db.Raw(tSql).Scan(&dumpSql).Error
|
if err != nil {
|
return nil, errors.New("dump err")
|
}
|
|
if len(dumpSql) > 0 {
|
for _, d := range dumpSql {
|
arr = append(arr, d.Sql)
|
}
|
}
|
}
|
|
return arr, nil
|
}
|
|
return nil, errors.New("tableNames is nil")
|
}
|
