如何在 Go 中为行插入重用单个 Postgres DB 连接？

一分耕耘，一分收获！既然打开了这篇文章《如何在 Go 中为行插入重用单个 Postgres DB 连接？》，就坚持看下去吧！文中内容包含golang等等知识点...希望你能在阅读本文后，能真真实实学到知识或者帮你解决心中的疑惑，也欢迎大佬或者新人朋友们多留言评论，多给建议！谢谢！

问题内容

我正在尝试使用 Go 为从 rabbitmq 接收到的每条新消息插入一行数据到 Postgres 表中，使用到在下面代码的 init 函数中打开的 DB 的单个连接。

代码不是只打开一个连接，而是打开 497 并最大化导致行插入停止......

我已经尝试使用这些问题中的信息[在 Go 应用程序中打开和关闭数据库连接](https://stackoverflow.com/questions/40587008/how-do-i-handle- opening-closing-db-connection-in-a-go- app/40587071#40587071)并[在一个函数](https://stackoverflow.com/questions/42581697/referring- to-an-open-database-connection-inside-a-function- golang)中打开数据库连接，该函数说我应该打开一个连接并使用全局数据库来允许主函数将 sql 语句传递给打开的连接初始化函数。

我以为我已经这样做了，但是正在为每个新行打开一个新连接，因此一旦达到 postgres 连接限制，代码就会停止工作......

我是 Go 新手，编程经验有限，过去两天我一直在尝试理解/解决这个问题，我真的可以通过一些帮助来理解我在哪里出错了......

var db *sql.DB

func init() {
    var err error
    db, err = sql.Open ( "postgres", "postgres://postgres:postgres@SERVER/PORT/DB")
    if err != nil {
        log.Fatal("Invalid DB config:", err)
    }
    if err = db.Ping(); err != nil {
        log.Fatal("DB unreachable:", err)
    }
}

func main() {

// RABBITMQ CONNECTION CODE IS HERE

// EACH MESSAGE RECEIVED IS SPLIT TO LEGEND, STATUS, TIMESTAMP VARIABLES

// VARIABLES ARE PASSED TO sqlSatement    

        sqlStatement := `
        INSERT INTO heartbeat ("Legend", "Status", "TimeStamp")
        VALUES ($1, $2, $3)
`
        // sqlStatement IS THEN PASSED TO db.QueryRow

        db.QueryRow(sqlStatement, Legend, Status, TimeStamp)
    }
}()


<p>完整代码如下所示：</p>
<pre class="brush:go;toolbar:false">package main

import (
    "database/sql"
    "log"
    _ "github.com/lib/pq"

    "github.com/streadway/amqp"
    "strings"
)
var db *sql.DB

func failOnError(err error, msg string) {
    if err != nil {
        log.Fatalf("%s: %s", msg, err)
    }
}

func init() {
    var err error
    db, err = sql.Open ( "postgres", "postgres://postgres:postgres@192.168.1.69:5432/test?sslmode=disable")
    if err != nil {
        log.Fatal("Invalid DB config:", err)
    }
    if err = db.Ping(); err != nil {
        log.Fatal("DB unreachable:", err)
    }
}

func main() {
    conn, err := amqp.Dial("amqp://Admin:Admin@192.168.1.69:50003/")
    failOnError(err, "Failed to connect to RabbitMQ")
    defer conn.Close()

    ch, err := conn.Channel()
    failOnError(err, "Failed to open a channel")
    defer ch.Close()

    q, err := ch.QueueDeclare(
        "HEARTBEAT", // name
        false,       // durable
        false,       // delete when unused
        false,       // exclusive
        false,       // no-wait
        nil,         // arguments
    )
    failOnError(err, "Failed to declare a queue")

    msgs, err := ch.Consume(
        q.Name, // queue
        "",     // consumer
        false,  // auto-ack
        false,  // exclusive
        false,  // no-local
        false,  // no-wait
        nil,    // args
    )
    failOnError(err, "Failed to register a consumer")

    forever := make(chan bool)

    go func() {

        for d := range msgs {
            myString := string(d.Body[:])
            result := strings.Split(myString, ",")
            Legend := result[0]
            Status := result[1]
            TimeStamp := result[2]

            sqlStatement := `
    INSERT INTO heartbeat ("Legend", "Status", "TimeStamp")
    VALUES ($1, $2, $3)
    `
            //
            db.QueryRow(sqlStatement, Legend, Status, TimeStamp)
        }
    }()

    
<h2 class="daan">
    正确答案
</h2>
<p>首先，<code>*sql.DB</code>不是一个连接，而是一个连接池，它将打开尽可能多的连接，只要它需要和 postgres
服务器允许的数量。只有当池中没有空闲连接可供使用时，它才会打开新连接。</p>
<hr><p>所以问题是DB打开的连接没有被释放，为什么？因为您在使用<code>QueryRow</code>时没有调用<code>Scan</code>返回<code>*Row</code>值。</p>
<p>在底层<code>*Row</code>拥有一个<code>*Rows</code>实例，该实例可以访问自己的连接，并且该连接在<code>Scan</code>被调用时会自动释放。如果<code>Scan</code>未调用，则不会释放连接，这会导致<code>DB</code>池在下一次调用
时打开新连接<code>QueryRow</code>。因此，由于您没有释放任何连接，因此<code>DB</code>会不断打开新连接，直到达到 postgres
设置指定的限制，然后下一次调用<code>QueryRow</code>挂起，因为它等待连接变为空闲。</p>
<p>因此，<code>Exec</code>如果您不关心输出，则需要使用，或者需要调用<code>Scan</code>返回的<code>*Row</code>.</p><p>好了，本文到此结束，带大家了解了《如何在 Go 中为行插入重用单个 Postgres DB 连接？》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！</p>