Golang 用go-sql-driver 调用MySQL存储过程时的问题排查

今天在用go-sql-driver做MySQL driver调用MySQL的存储过程时遇到一个罕见的报错。本文对这个错误及其解做一个简单的总结。

1 问题描述

按照go中sql包提供的接口，应用层代码调用MySQL存储过程的代码一般如下所示：

       result, err := dbh.Exec("call some_procedure(?, ?)", param1, param2)
        if err != nil {
                // error handler
        }

       // using result...

　　但是在实际执行的时候err不为nil，err的值是MySQL返回给driver的出错信息：

Error 1312: PROCEDURE tcheck_db.update_vs_available can't return a result set in the given context

2  问题原因

出现这个问题的原因是在调用存储过程时，MySQL希望客户端的连接类型是multi-statement模式，但是go-sql-driver这个包目前没有设置这个连接模式，所以MySQL给返回了1中的错误。

实际上，在用go-sql-driver时，除了无法调用存储过程外，还存在一些其他的限制，这些限制可以在go-database-sql-surprises中看到。

go-sql-driver的讨论区也有人报出了这个问题，见这里。

3 问题解法

既然找到了病因，那就要对症下药，能想到的办法就是将go-sql-driver连接MySQL时的connection mode设置为multi-statement。方法是：

1） go get github.com/go-sql-driver/mysql 下载依赖包

2）编辑文件src/github.com/go-sql-driver/mysql/packets.go文件，在函数writeAuthPacket(cipher []byte) （大约在210行）的客户端标志位处添加下面两个标志：

 clientMultiStatements 和 clientMultiResults

添加后的代码如下：

func (mc *mysqlConn) writeAuthPacket(cipher []byte) error {
        // Adjust client flags based on server support
        clientFlags := clientProtocol41 |
                clientSecureConn |
                clientLongPassword |
                clientTransactions |
                clientLocalFiles |
                clientMultiStatements |    // 添加这行
                clientMultiResults |       //再添加这行
                mc.flags&clientLongFlag

        if mc.cfg.clientFoundRows {
                clientFlags |= clientFoundRows
        }

        ...

3，rebuild您的依赖go-sql-driver的代码。

OK。经过上面三步修改之后，执行1中的存储过程代码时，就不会报错了。