sql - 如何使用多个驱动程序编译 Go 数据库/sql 程序？

Question

我目前正在用 Go 为 sql 数据库(Postgres 和 Mysql)编写一个测试程序。我不太了解包的“_”选项，但我正在使用它(见下文)。

我希望能够做的是编译一次为一个 RDBMS 和多个 RDBMS 使用多个 sql 驱动程序，并在运行程序时选择要使用的驱动程序和 RDBMS。我不确定这是否可能。目前我使用一个 Postgres 和一个 Mysql 驱动程序进行编译，然后在运行时选择我正在使用的驱动程序 (Postgres/Mysql)。这工作正常，但我需要记住编译了哪个驱动程序。如果能够为一个 RDBMS 使用多个驱动程序进行编译，然后在运行时选择要使用的驱动程序，那将是一件好事。我想那是不可能的。或者，最好能够在编译时选择要使用的驱动程序，并在运行时知道正在使用哪些驱动程序。如果没有这些设施之一，就可以进行测试，例如。 Postgres 并认为他们正在使用一个驱动程序，而实际上该程序是用另一个驱动程序编译的。

是否可以有一个编译器选项来选择特定的驱动程序，然后在运行时知道正在使用哪个驱动程序？另一种方法显然是编辑程序以表明这一点。

导入示例如下:

import (
    "database/sql"
    _ "github.com/go-sql-driver/mysql"
    ////_ "github.com/lib/pq"
    _ "github.com/lxn/go-pgsql"
 ........

Answer 1

I don't know much about the "_" option for packages however I am using it (see below).

在导入路径前添加 _ 将正常导入包(运行其“init() 函数)但它不会关联当前包中的名称使用导入包。

What I would like to be able to do is to compile once to use multiple sql drivers for the one RDBMS and also for multiple RDBMS's and when running the program, select which driver and RDBMS to use. I'm not sure if that's possible.

包 "github.com/go-sql-driver/mysql" 的 init() 函数执行以下操作:

func init() {
        sql.Register("mysql", &MySQLDriver{})
}

它将调用database/sql 的Register函数定义为:

func Register(name string, driver driver.Driver)

并且有条件:

If Register is called twice with the same name or if driver is nil, it panics.

注册驱动程序后，您可以使用sql.Open :

func Open(driverName, dataSourceName string) (*DB, error)

使用第一个参数指定的驱动程序打开到数据库的新连接，例如:

db, e := sql.Open("mysql", "user:pass@host:port")

顺便说一下，github.com/lxn/go-pgsql 的 init() 函数如下所示:

func init() {
        sql.Register("postgres", sqlDriver{})
}

---

我重新阅读了您的问题，我认为您还想指定在运行程序时要使用的数据库和驱动程序。

为此，您可以使用 flag package并像这样运行您的应用程序:

./my_app -driver=mysql -db="user:pass@host:port"

并将这些字符串传递给sql.Open。