最近准备将我平时写的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插件后可以在编写代码同时验证代码语法,代码提示也很全,开发起来很方便。总体而言,开发效率挺高的。