Go语言调用mysql.exe和mysqldump命令行导入导出数据库

原创
2017/08/03 09:45
阅读数 2.7K

最近准备将我平时写的ruby和python小工具推广到整个小组,以提高小组工作效率。但是效果不太理想,主要是脚本语言依赖环境,安装配置比较繁琐,组员们虽然很羡慕我的工具,但是愿意尝试很少。

Go语言最近比较火,了解了下发现它编译后生成exe,没有依赖,发给其他人就可以直接用,于是我又萌生了将部分通用工具用Go改写的念头。

第一个改写的工具很简单,就是将远程的数据库数据导出到本地并导入到本地的数据库。在ruby 中代码中主要的逻辑是以下两行代码

`mysqldump -h#{host} -u#{user} -p#{pass} #{db} --complete-insert --skip-comments --compact --add-drop-table >sqls/#{websitename}.sql`
`mysql --reconnect --default-character-set=utf8 -h #{exporthost} -u#{exportuser} -p#{exportpass} --database=#{exportdb} < #{file}`

就是这么简单的两行命令行代码在Go开发中,却碰了壁,网上搜索一番也没有直接的回答,在网上仔细研究了相关论坛回帖的只字片语和本人的不断尝试,终于解决了这个问题。

Go相关代码如下:

mysqldump:

func exportDB(info WebInfo,webname string) error{
    argv := []string{"--complete-insert","--skip-comments","--compact","--add-drop-table","-h"+info.Mysqlhost,"-u"+info.Mysqluname,"-p"+info.Mysqlupasswd,info.Mysqldb}
    cmd := exec.Command("mysqldump", argv...)
    f,err:=os.OpenFile("sqls/"+webname+".sql",os.O_CREATE|os.O_RDWR,os.ModePerm|os.ModeTemporary)
    if(err!=nil){
        fmt.Println("打开sql文件失败")
        return err
    }
    defer f.Close()
    cmd.Stdout=f
    cmd.Stderr=os.Stderr
    cmd.Start()
    cmd.Run()
    cmd.Wait()
    return nil
}

mysql:

func importToLocal(info Info) error{
    argv := []string{"--reconnect","--default-character-set=utf8","-h"+info.Mysqlhost,"-u"+info.Mysqluname,"-p"+info.Mysqlupasswd,"--database="+info.Mysqldb}
    cmd := exec.Command("mysql", argv...)
    f,err:=os.Open("sqls/"+info.Webname+".sql")
    if(err!=nil){
        fmt.Println("读取sql文件失败")
        return err
    }
    defer f.Close()
    cmd.Stdin=f
    cmd.Stdout=os.Stdout
    cmd.Stderr=os.Stderr
    cmd.Start()
    cmd.Run()
    cmd.Wait()
    return nil
}

上面两段代码,是我整个工具代码中的一部分,无法直接复制使用,要使用的话,首先要导入os/exec等package,然后将函数中部分变量替换成自己的数据库信息。

import (
	"fmt"
	"os"
	"os/exec"
)

type Info struct {
    Webname string
	Mysqlhost string
	Mysqldb string
	Mysqluname string
	Mysqlupasswd string
}

type WebInfo Info

下面说说Go中调用命令行的一些迷惑地方。

1. exec.Command 里调用命令行,必须将命令行参数拆分成数组,放入函数的参数中。这个相对而言比较容易察觉。

2. exec.Command 不支持命令行中的 < 和 > 重定向操作,如果要实现类似的功能,可以在代码里打开文件,并将文件描述符f赋值给 cmd.Stdin或者cmd.Stdout。这个问题很隐蔽,是调用命令行中很让人困惑的地方。

最后说说新学Go语言的感受,相对于php、ruby、python写起来还是很繁琐的。不过作为一个静态编译类型的语言,Go语言还是挺好入门的。编译速度很快。安装vscode插件后可以在编写代码同时验证代码语法,代码提示也很全,开发起来很方便。总体而言,开发效率挺高的。

展开阅读全文
打赏
1
0 收藏
分享
加载中
谢谢大佬分享!
09/10 14:41
回复
举报
更多评论
打赏
1 评论
0 收藏
1
分享
返回顶部
顶部