Go使用godotenv连接多数据库时查询指向错误库问题咨询

你遇到的跨库查表报错核心原因有2个：

1. godotenv 全局环境污染问题：godotenv.Load 方法会把读取到的配置直接写入进程全局的环境变量表，且默认逻辑是不覆盖已存在的环境变量。第一次调用connectDB加载订单库env文件后，DIALECT/HOST/DBPORT/NAME等数据库连接参数已经被设置为订单库的值，第二次调用connectDB加载书籍库env时，这些已存在的全局变量不会被覆盖，导致两次sql.Open实际连接的都是第一个初始化的数据库，自然找不到另一个库的表。
2. 代码本身存在语法/逻辑笔误：
   • getOrders方法的rows.Scan参数多传了未定义的&date变量，会直接编译失败
   • book结构体定义的字段是OrderID，但查询逻辑中Scan和赋值使用的是bookID/BookID，字段不匹配会触发panic
   • 创建*sql.DB实例后没有调用Ping校验连接有效性，sql.Open本身是懒加载机制，不会实际建立连接，配置错误时不会第一时间报错。

核心原则是不同数据库的配置读取完全隔离，不依赖全局环境变量传递配置，修复步骤如下：

• 替换godotenv.Load为不污染全局环境的配置读取方法，直接从env文件解析出对应库的连接参数，独立传给sql.Open
• 修复结构体定义、Scan参数匹配的笔误
• 每个DB实例创建后主动Ping验证连接，配置合理的连接池参数，程序退出前主动关闭连接

修正后的完整代码：

package main

import (
	"database/sql"
	"fmt"
	"log"
	"os"

	"github.com/joho/godotenv"
	_ "github.com/lib/pq" // 按实际使用的数据库驱动导入对应包
)

// connectDB 独立读取指定env文件的配置创建数据库连接，不污染全局环境变量
func connectDB(dbEnvPath string) *sql.DB {
	// 读取env文件内容为独立配置map，不写入全局环境变量
	envMap, err := godotenv.Read(dbEnvPath)
	if err != nil {
		log.Fatalf("读取数据库配置文件 %s 失败: %s", dbEnvPath, err)
	}

	// 从独立配置map取参数，和全局环境变量完全隔离
	dialect := envMap["DIALECT"]
	host := envMap["HOST"]
	dbPort := envMap["DBPORT"]
	user := envMap["USER"]
	dbName := envMap["NAME"]
	password := envMap["PASSWORD"]

	dbURI := fmt.Sprintf("port=%s host=%s user=%s password=%s dbname=%s sslmode=disable",
		dbPort, host, user, password, dbName)

	db, err := sql.Open(dialect, dbURI)
	if err != nil {
		log.Fatalf("初始化数据库实例失败: %s", err)
	}

	// 主动建立连接校验配置正确性
	if err = db.Ping(); err != nil {
		log.Fatalf("连接数据库 %s 失败: %s", dbName, err)
	}

	// 按业务场景配置连接池参数
	db.SetMaxOpenConns(10)
	db.SetMaxIdleConns(5)

	return db
}

type order struct {
	OrderID string `json:"orderID"`
	Name    string `json:"name"`
}

type book struct {
	// 修正字段名和查询返回字段匹配
	BookID string `json:"bookID"`
	Name   string `json:"name"`
}

func getOrders(db *sql.DB) []order {
	var (
		orderID string
		name    string
	)

	var allRows = []order{}
	query := `SELECT orderID, name FROM orders.orders;`

	rows, err := db.Query(query)
	if err != nil {
		log.Fatalf("查询订单列表失败: %s", err)
	}
	defer rows.Close()

	for rows.Next() {
		// 移除多余的&date参数，和查询字段数量、顺序完全匹配
		err := rows.Scan(&orderID, &name)
		if err != nil {
			log.Fatalf("读取订单行数据失败: %s", err)
		}

		row := order{
			OrderID: orderID,
			Name:    name,
		}
		allRows = append(allRows, row)
	}

	if err = rows.Err(); err != nil {
		log.Fatalf("遍历订单结果集失败: %s", err)
	}

	return allRows
}

func getBooks(db *sql.DB) []book {
	var (
		bookID string
		name   string
	)

	var allRows = []book{}
	query := `SELECT bookID, name FROM books.books;`

	rows, err := db.Query(query)
	if err != nil {
		log.Fatalf("查询书籍列表失败: %s", err)
	}
	defer rows.Close()

	for rows.Next() {
		err := rows.Scan(&bookID, &name)
		if err != nil {
			log.Fatalf("读取书籍行数据失败: %s", err)
		}

		row := book{
			BookID: bookID,
			Name:   name,
		}
		allRows = append(allRows, row)
	}

	if err = rows.Err(); err != nil {
		log.Fatalf("遍历书籍结果集失败: %s", err)
	}

	return allRows
}

func main() {
	ordersDB := connectDB("ordersDB.env")
	// 用完及时关闭连接避免泄漏
	defer ordersDB.Close()
	booksDB := connectDB("booksDB.env")
	defer booksDB.Close()

	orders := getOrders(ordersDB)
	books := getBooks(booksDB)

	// 后续业务逻辑
	log.Printf("查询到订单数: %d, 书籍数: %d", len(orders), len(books))
}

• 不要在业务函数里滥用log.Fatal，该方法会直接调用os.Exit终止程序，导致defer注册的资源关闭逻辑不执行，生产环境建议返回error给上层统一处理
• 如果两个数据库属于同一个数据库实例的不同schema，不需要创建两个独立连接，直接在查询语句里指定schema名即可，或者连接后执行SET search_path TO xxx切换schema
• 所有SQL查询的Scan参数必须和SELECT返回的字段顺序、数量、类型完全匹配，否则会触发扫描错误

内容的提问来源于stack exchange，提问作者myo tun