是否存在支持单个字符串中的多个语句的 Go Mysql driver？

Question

我试图找到一个 MySql driver 可以与 Go 一起使用，它支持在一次调用中发出多个 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 中的相同功能，但似乎所有不同的 driver 都不支持它，坦率地说，这让我感到震惊。

Go-MySQL-Driver
https://github.com/go-sql-driver/mysql
这似乎是 Go 最常用的 driver。

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
这是另一个流行的 driver.

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 driver 可以像这样在一个字符串中处理多个语句？

Answer 1

https://github.com/ziutek/mymysql

能行。虽然你必须使用它的接口而不是 go 定义的接口。 go官方接口没有处理，还是多个return值。

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)
    }
}

Answer 2

我建议只拨打 2 个电话。为什么不？它使代码更容易理解并改进了错误处理。

如果转储中有一个大 SQL 文件，另一种选择是 shell 输出并一次性执行整个过程。

Answer 3

可以将 github.com/go-sql-driver/mysql 配置为使用 multiStatements=true 连接参数接受多个语句。

文档清楚地说明了为什么要小心操作。参见 https://github.com/go-sql-driver/mysql

Answer 4

为@ithkuil关于go-sql-driver包的multiStatements的回答添加一个例子以供参考。（我没有足够的代表来添加评论）。

multiStatements 的参数已添加到您的 sql.Open 调用的 dataSourceName 字符串中。例如

db, err := sql.Open("mysql", "user:pass@tcp(localhost:3306)/dbname?multiStatements=true")

建议您不要使用这样的数据库处理程序来处理用户输入，但它非常适合处理已知 sql 文件。

是否存在支持单个字符串中的多个语句的 Go Mysql driver？

Does a Go Mysql driver exist that supports multiple statements within a single string?

mysql

driver

go