我正在尝试找到一个可以与Go一起使用的MySql驱动程序,它支持在一次调用中发出多个SQL语句。例如,我可能希望使用以下SQL创建数据库:
DROP SCHEMA IF EXISTS foo;
CREATE SCHEMA IF NOT EXISTS foo;
在PHP等语言中,您可以将两个SQL语句放在一个字符串中并一次执行,如下所示:
$db = new PDO(...);
$db->query("DROP SCHEMA IF EXISTS foo; CREATE SCHEMA IF NOT EXISTS foo;");
我需要这个的原因是因为我有SQL转储(来自mysqldump
)我想以编程方式应用于各种数据库。
我在Go中寻找相同的功能,但似乎所有不同的驱动程序都不支持它,坦率地说,这让我感到震惊。
GO-MySQL的驱动程序 https://github.com/go-sql-driver/mysql 这似乎是Go最常用的驱动程序。
package main
import "database/sql"
import "log"
import _ "github.com/go-sql-driver/mysql"
func main() {
db, err := sql.Open("mysql", "user:password@(127.0.0.1:3306)/")
if err != nil {
log.Println(err)
}
sql := "DROP SCHEMA IF EXISTS foo; CREATE SCHEMA IF NOT EXISTS foo;"
_, err = db.Exec(sql)
if err != nil {
log.Println(err)
}
db.Close()
}
输出:
2015/02/16 18:58:08 Error 1064: You have an error in your SQL syntax; check the manual that corresponds to your MySQL server version for the right syntax to use near 'CREATE SCHEMA IF NOT EXISTS foo' at line 1
MyMySQL https://github.com/ziutek/mymysql 这是另一个受欢迎的驱动器
package main
import "database/sql"
import "log"
import _ "github.com/ziutek/mymysql/godrv"
func main() {
db, err := sql.Open("mymysql", "database/user/password")
if err != nil {
log.Println(err)
}
sql := "DROP SCHEMA IF EXISTS foo; CREATE SCHEMA IF NOT EXISTS foo;"
_, err = db.Exec(sql)
if err != nil {
log.Println(err)
}
sql = "USE DATABASE foo;"
_, err = db.Exec(sql) // <-- error
if err != nil {
log.Println(err)
}
db.Close()
}
输出:
2015/02/16 18:58:08 packet sequence error
有没有人知道任何与Go兼容的MySql驱动程序可以处理一个字符串中的多个语句?
https://github.com/ziutek/mymysql
可以做到。虽然你必须使用它的界面与go定义的界面。 go官方界面不处理它,或多个返回值。
package main
import (
"flag"
"fmt"
"github.com/ziutek/mymysql/autorc"
"github.com/ziutek/mymysql/mysql"
_ "github.com/ziutek/mymysql/thrsafe"
)
type ScanFun func(int, []mysql.Row, mysql.Result) error
func RunSQL(hostport, user, pass, db, cmd string, scan ScanFun) error {
conn := autorc.New("tcp", "", hostport, user, pass, db)
err := conn.Reconnect()
if err != nil {
return err
}
res, err := conn.Raw.Start(cmd)
if err != nil {
return err
}
rows, err := res.GetRows()
if err != nil {
return err
}
RScount := 0
scanErr := error(nil)
for {
if scanErr == nil {
func() {
defer func() {
if x := recover(); x != nil {
scanErr = fmt.Errorf("%v", x)
}
}()
scanErr = scan(RScount, rows, res)
}()
}
if res.MoreResults() {
res, err = res.NextResult()
if err != nil {
return err
}
rows, err = res.GetRows()
if err != nil {
return err
}
} else {
break
}
RScount++
}
return scanErr
}
func main() {
host := flag.String("host", "localhost:3306", "define the host where the db is")
user := flag.String("user", "root", "define the user to connect as")
pass := flag.String("pass", "", "define the pass to use")
db := flag.String("db", "information_schema", "what db to default to")
sql := flag.String("sql", "select count(*) from columns; select * from columns limit 1;", "Query to run")
flag.Parse()
scan := func(rcount int, rows []mysql.Row, res mysql.Result) error {
if res.StatusOnly() {
return nil
}
for idx, row := range rows {
fmt.Print(rcount, "-", idx, ") ")
for i, _ := range row {
fmt.Print(row.Str(i))
fmt.Print(" ")
}
fmt.Println("")
}
return nil
}
fmt.Println("Host - ", *host)
fmt.Println("Db - ", *db)
fmt.Println("User - ", *user)
if err := RunSQL(*host, *user, *pass, *db, *sql, scan); err != nil {
fmt.Println(err)
}
}
可以将github.com/go-sql-driver/mysql
配置为使用multiStatements=true
连接参数接受多个语句。
文档清楚地说明了为什么要小心这样做。见https://github.com/go-sql-driver/mysql
添加@ithkuil关于go-sql-driver包的multiStatements的答案的示例以供参考。 (我没有足够的代表作为评论添加)。
multiStatements的参数被添加到sql.Open调用的dataSourceName字符串中。例如
db, err := sql.Open("mysql", "user:pass@tcp(localhost:3306)/dbname?multiStatements=true")
建议您不要使用这样的数据库处理程序来处理用户输入,但它非常适合处理已知的sql文件。
我建议只打2个电话。为什么不?它使代码更易于理解并改进了错误处理。
另一个选项,如果你有一个来自转储的大型SQL文件,那就是一次性地执行整个操作。
直截了当地说,不要在一次调用中发出多个语句。这是一个安全问题。如果一个黑客可以进入一点点,他可以接管你的机器。 (阅读“SQL注入”。)
存储的例程为您提供了一种安全的方式来组合语句(有一些限制)。