Go 通过 godotenv 在使用多个数据库时查询错误的数据库

哈喽！今天心血来潮给大家带来了《Go 通过 godotenv 在使用多个数据库时查询错误的数据库》，想必大家应该对Golang都不陌生吧，那么阅读本文就都不会很困难，以下内容主要涉及到，若是你正在学习Golang，千万别错过这篇文章~希望能帮助到你！

我正在尝试从多个数据库进行查询。每个数据库使用以下函数连接：

func connectDB(dbEnv str) *sql.DB{
    // Loading environment variables from local.env file
    
    err1 := godotenv.Load(dbEnv)
    if err1 != nil {
        log.Fatalf("Some error occured. Err: %s", err1)
    }
    dialect := os.Getenv("DIALECT")
    host := os.Getenv("HOST")
    dbPort := os.Getenv("DBPORT")
    user := os.Getenv("USER")
    dbName := os.Getenv("NAME")
    password := os.Getenv("PASSWORD")

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

    // Create database object
    db, err := sql.Open(dialect,dbURI)

    if err != nil {
        log.Fatal(err)
    }

    return db

}

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

type book struct{
    OrderID string `json:"orderID"`
    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;
`
    //Get rows using the query
    rows, err := db.Query(query)
    if err != nil { //Log if error
        log.Fatal(err)
    }
    defer rows.Close()
    
    // Add each row into the "allRows" slice
    for rows.Next() {

        err := rows.Scan(&orderID, &name, &date)
        if err != nil {
            log.Fatal(err)
        }

        //Create new order struct with the received data
        row := order{
            OrderID: orderID,
            Name:  name,
        }
        allRows = append(allRows, row)
    }
    //Log if error
    err = rows.Err()
    if err != nil {
        log.Fatal(err)
    }

    return allRows
}

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

    var allRows = []book{}
    query := `
        SELECT bookID, name
        FROM books.books;
        `
    //Get rows using the query
    rows, err := db.Query(query)
    if err != nil { //Log if error
        log.Fatal(err)
    }
    defer rows.Close()
    
    // Add each row into the "allRows" slice
    for rows.Next() {

        err := rows.Scan(&bookID, &name)
        if err != nil {
            log.Fatal(err)
        }

        //Create new book struct with the received data
        row := book{
            BookID: bookID,
            Name:  name,
        }
        allRows = append(allRows, row)
    }
    //Log if error
    err = rows.Err()
    if err != nil {
        log.Fatal(err)
    }

    return allRows
}


func main() {

    ordersDB:= connectDB("ordersDB.env")
    booksDB:= connectDB("booksDB.env")

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

问题是，当我第一次使用ordersdb时，程序只识别ordersdb中的表。当我第一次使用 booksdb 时，程序只识别 booksdb 中的表。

当我尝试使用ordersdb后查询booksdb中的表时，它给出了“关系“books.books”不存在”错误。当我在使用 booksdb 后尝试查询ordersdb 中的表时，它给出“关系“orders.orders”不存在”

有没有更好的方法来连接多个数据库？

正确答案

您正在使用 github.com/joho/godotenv 从环境中加载数据库配置。总结（并删除大量细节）您正在做的事情是：

godotenv.Load("ordersDB.env")
host := os.Getenv("HOST")
// Connect to DB
godotenv.Load("booksDB.env")
host := os.Getenv("HOST")
// Connect to DB 2

但是，如 the docs 中所述“现有环境优先于稍后加载的环境”。这也得到了更明确的说明here“重要的是要注意，它不会覆盖已经存在的环境变量”。

因此，您的代码将加载到第一个 .env 文件中，填充环境变量，然后连接到数据库。然后，您将加载第二个 .env 文件，但由于环境变量已设置，因此它们不会更改，并且您将再次连接到同一数据库。

作为解决方案，您可以使用 Overload。但是，最好重新考虑环境变量的使用（也许对第二个连接使用不同的变量）。

今天关于《Go 通过 godotenv 在使用多个数据库时查询错误的数据库》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！