如何在 Go 中实现动态加载数据库驱动的多数据源方案-Golang-PHP中文网

database/sql驱动必须在main包init阶段通过空白导入完成注册，运行时无法动态加载；多数据源需提前导入所有可能驱动，再通过配置解析和sql.open分别构建独立连接池。

如何在 go 中实现动态加载数据库驱动的多数据源方案

database/sql 的驱动注册必须在 main 包 init 阶段完成

Go 的 database/sql 不自带驱动，所有驱动都靠 sql.Register() 注册后才能用。而注册动作发生在驱动包的 init() 函数里——这意味着你不能在运行时（比如根据 flag 或配置）再去 import 某个驱动包并期望它“即时生效”。_ "github.com/go-sql-driver/mysql" 这类空白导入必须出现在编译期可见的代码路径中，否则 sql.Open("mysql", ...) 会 panic：unknown driver "mysql"。

常见错误现象：sql: unknown driver "postgres" (forgotten import?) —— 不是配置写错了，而是对应驱动没被编译进去。

所有可能用到的驱动（如 mysql、postgres、sqlite3、dm）都得提前 blank-import 到 main.go 或某个会被编译进最终二进制的包里
不要试图用 plugin 或 go:embed 动态加载驱动；Go 官方不支持运行时注入驱动实现
如果项目只部署 MySQL，但代码里 import 了 PostgreSQL 驱动，不会增加运行时开销，只是二进制体积略大

用 viper + struct tag 解析多数据源配置并构造多个 *sql.DB 实例

驱动注册完只是第一步，真正实现“动态选择”靠的是配置解析和连接池构建。推荐用 viper 加载 YAML/JSON/TOML，把不同数据库的 driver 名（如 "mysql"）、dsn、连接参数分组管理。

示例配置片段（config.yaml）：

database:
- name: primary
  driver: mysql
  dsn: root:pass@tcp(127.0.0.1:3306)/app?parseTime=true&loc=Local
  max_open_conns: 50
- name: analytics
  driver: postgres
  dsn: host=localhost port=5432 user=app dbname=report sslmode=disable
  max_open_conns: 20

关键点：

driver 字段值必须和 sql.Register() 时传入的 name 严格一致（大小写敏感）
每个数据源需独立调用 sql.Open()，返回各自的 *sql.DB；别共用一个实例
务必设置 SetMaxOpenConns 和 SetMaxIdleConns，否则默认 0（不限制）容易打爆数据库
建议为每个 *sql.DB 实例加前缀日志或封装成带名称的 struct，避免后续混淆

GORM v2 多数据源需显式传入不同驱动的 Dialector

如果你用 GORM，v2 的 gorm.Open() 不再接受字符串 driver name，而是要求传具体驱动的 Dialector 实例。这意味着你不能仅靠配置里的 driver: "mysql" 就自动选型——得手动匹配。

使用Go语言搭建家庭相册系统-相关课件

下载

典型做法是写一个映射函数：

func NewDialector(cfg DatabaseConfig) gorm.Dialector {
    switch cfg.Driver {
    case "mysql":
        return mysql.Open(cfg.DSN)
    case "postgres":
        return postgres.Open(cfg.DSN)
    case "sqlite":
        return sqlite.Open(cfg.DSN)
    default:
        panic("unsupported driver: " + cfg.Driver)
    }
}

注意：