You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
2.2 KiB
101 lines
2.2 KiB
package utils
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
_ "github.com/go-sql-driver/mysql"
|
|
"github.com/go-xorm/core"
|
|
"github.com/go-xorm/xorm"
|
|
)
|
|
|
|
// Mysqldump mysql导出数据对象
|
|
type Mysqldump struct {
|
|
conn *xorm.Engine
|
|
cfg *Config
|
|
isClose bool
|
|
}
|
|
|
|
// New 创建一个Mysqldump对象
|
|
func New(cfg *Config) (*Mysqldump, error) {
|
|
if cfg == nil {
|
|
return nil, errors.New("配置信息不能为nil")
|
|
}
|
|
// 处理配置信息
|
|
if cfg.OutPath == "" {
|
|
return nil, errors.New("导出sql输出路径不能是空")
|
|
}
|
|
if cfg.ExportDataStep == 0 {
|
|
cfg.ExportDataStep = 1000
|
|
}
|
|
|
|
// 创建导出对象
|
|
mysqldump := &Mysqldump{
|
|
cfg: cfg,
|
|
isClose: false,
|
|
}
|
|
// 连接mysql
|
|
err := mysqldump.OpenMysql()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return mysqldump, nil
|
|
}
|
|
|
|
// Close 不使用导出功能时，关闭连接资源
|
|
func (md *Mysqldump) Close() error {
|
|
md.isClose = true
|
|
return md.conn.Close()
|
|
}
|
|
|
|
// OpenMysql 连接mysql
|
|
func (md *Mysqldump) OpenMysql() error {
|
|
// 拼接连接数据库字符串
|
|
connStr := fmt.Sprintf("%s:%s@(%s:%d)/%s?charset=utf8&parseTime=True&loc=UTC",
|
|
md.cfg.DbCfg.User,
|
|
md.cfg.DbCfg.Passwd,
|
|
md.cfg.DbCfg.Address,
|
|
md.cfg.DbCfg.Port,
|
|
md.cfg.DbCfg.DbName)
|
|
|
|
// 连接数据库
|
|
engine, err := xorm.NewEngine("mysql", connStr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 是否开启debug模式
|
|
if md.cfg.Debug {
|
|
engine.Logger().SetLevel(core.LOG_DEBUG) // 调试信息
|
|
engine.ShowSQL(true) // 显示sql
|
|
}
|
|
engine.SetMaxIdleConns(2) // 空闲连接池数量
|
|
engine.SetMaxOpenConns(8) // 最大连接数
|
|
engine.SetMapper(core.GonicMapper{}) // 命名规则
|
|
|
|
// 设置数据库时区
|
|
engine.DatabaseTZ = time.UTC
|
|
engine.TZLocation = time.UTC
|
|
|
|
md.conn = engine
|
|
|
|
log.Println("连接数据库成功")
|
|
return nil
|
|
}
|
|
|
|
// GetRootDir 获取程序跟目录,返回值尾部包含'/'
|
|
func (md *Mysqldump) GetRootDir() string {
|
|
// 文件不存在获取执行路径
|
|
file, err := filepath.Abs(filepath.Dir(os.Args[0]))
|
|
if err != nil {
|
|
file = fmt.Sprintf(".%s", string(os.PathSeparator))
|
|
} else {
|
|
file = fmt.Sprintf("%s%s", file, string(os.PathSeparator))
|
|
}
|
|
return file
|
|
}
|
|
|