navicat是mysql可视化工具中最棒的,但是,在处理视图的导入导出方面,它是按照视图名称的字母顺序来处理的,若视图存在依赖,在导入过程中就会报错。前面已经用python写了一个,但在使用过程中,遇到xfffd编码,python的pymysql会直接崩溃。发现golang没有这个问题,正好用go重写,来熟悉golang。
一些关键点map & json,在处理主键与外键信息时,需要用到json数据结构来存储中间结果,因为要灵活处理,在golang中只能用map[string]interface{}来处理。
interface{} 相当于java中的object,能接受任意数据类型,方便但在使用时要做到心中有数,不然一旦数据类型不匹配,程序就会崩溃。
xfffd ,utf8中的占位符,超出范围的utf8mb4入库后,会被存储为xfffd,数据导出时,需要过滤掉。
goroutine, golang的并发支持很独特,我们的工具支持多个库同时备份,很容易使用goroutine来实现并行。
代码解析按功能模块对核心代码进行说明
main.go,并发、命令行参数使用命令行参数,接受一个参数,来指定备份的内容
package common
main.go,程序入口,处理命令行参数
if len(os.Args) > 1 { flag = common.OpFlag{ Tables: false, Datum: false, Views: false, Funcs: false, } switch os.Args[1] { //接受一个参数 case "table": flag.Tables = true //根据参数设定标识量 case "data": flag.Tables = true flag.Datum = true case "views": flag.Views = true case "funcs": flag.Funcs = true default: //参数不正确,报错退出 log.Fatal("You arg must be in : table, data, views or funcs.") } }else{ //无参数,默认导出所有 flag = common.OpFlag{ Tables: true, Datum: true, Views: true, Funcs: true, } } err := backUp.Export(flag) 根据参数进行数据库备份 Export.go备份主流程,根据configs.json生成goroutine来备份数据库,并等待完成。
var configs interface{} fr, err := os.Open("./configs.json") if err != nil { return err } decoder := json.NewDecoder(fr) //解析配置文件 err = decoder.Decode(&configs) confs := configs.(map[string]interface{}) workDir := confs["workDir"].(string) ch := make(chan string) //通道变量 for key, value := range confs { if strings.HasPrefix(key, "db_") { dbConf := value.(map[string]interface{}) dbConn := common.DbConnFields{ //数据库相应配置 DbHost: dbConf["db_host"].(string), DbPort: int(dbConf["db_port"].(float64)), DbUser: dbConf["db_user"].(string), DbPass: dbConf["db_pass"].(string), DbName: dbConf["db_name"].(string), DbCharset: dbConf["db_charset"].(string), } if dbConf["file_alias"] != nil { //生成sql备份文件的命名 dbConn.FileAlias = dbConf["file_alias"].(string) } go ExportOne(dbConn, workDir, ch, flag) //创建协程 } } for key := range confs { //阻塞主进程,待所有协程完成工作 if strings.HasPrefix(key, "db_") { fmt.Print( <- ch ) } } return nil你需要编写如下的配置文件来描述你要备份的数据库:
{ "db_name1": { "db_host": "192.168.1.8", "db_port": 3306, "db_user": "root", "db_pass": "123456", "db_name": "name1", "db_charset": "utf8mb4", "file_alias": "file name1" }, "db_name2": { "db_host": "localhost", "db_port": 3306, "db_user": "root", "db_pass": "123456", "db_name": "name2", "db_charset": "utf8mb4" }, "database_dialect": "mysql", "workDir": "/home/zhoutk/gocodes/goTools/" } ExportOne.go