当前位置：首页 > 文章列表 > Golang > Go问答 > 在 Go 中，在 http 处理程序中使用 pgx 上下文的正确方法是什么？

在 Go 中，在 http 处理程序中使用 pgx 上下文的正确方法是什么？

来源：stackoverflow 2024-04-10 08:27:37 0浏览收藏

大家好，我们又见面了啊~本文《在 Go 中，在 http 处理程序中使用 pgx 上下文的正确方法是什么？》的内容中将会涉及到等等。如果你正在学习Golang相关知识，欢迎关注我，以后会给大家带来更多Golang相关文章，希望我们能一起进步！下面就开始本文的正式内容~

问题内容

更新 1：似乎使用与 http 请求绑定的上下文可能会导致“上下文已取消”错误。但是，使用 context.background() 作为父级似乎工作正常。

    // this works, no 'context canceled' errors
    ctx, cancel := context.withtimeout(context.background(), 100*time.second)

    // however, this creates 'context canceled' errors under mild load
    // ctx, cancel := context.withtimeout(r.context(), 100*time.second)

    defer cancel()
    app.insert(ctx, record)

（更新了下面的代码示例以生成用于重现的独立示例）

在 go 中，我有一个 http 处理程序，如下代码所示。在对此端点的第一个 http 请求中，我收到 context cancelled 错误。然而，数据实际上是插入到数据库中的。在对该端点的后续请求中，不会给出此类错误，并且数据也成功插入到数据库中。

问题：我是否在 http 处理程序和 pgx queryrow 方法之间正确设置并传递了 context ？（如果没有的话还有更好的方法吗？）

如果将此代码复制到 main.go 并运行 go run main.go，转到 localhost:4444/create 并按住 ctrl-r 以产生轻微负载，您应该会看到产生一些上下文取消错误。

package main

import (
    "context"
    "fmt"
    "log"
    "math/rand"
    "net/http"
    "time"

    "github.com/jackc/pgx/v4/pgxpool"
)

type application struct {
    DB *pgxpool.Pool
}

type Task struct {
    ID     string
    Name   string
    Status string
}

//HTTP GET /create
func (app *application) create(w http.ResponseWriter, r *http.Request) {
    fmt.Println(r.URL.Path, time.Now())
    task := &Task{Name: fmt.Sprintf("Task #%d", rand.Int()%1000), Status: "pending"}
    // -------- problem code here ----
    // This line works and does not generate any 'context canceled' errors
    //ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
    // However, this linegenerates 'context canceled' errors under mild load
    ctx, cancel := context.WithTimeout(r.Context(), 100*time.Second)
    // -------- end -------
    defer cancel()
    err := app.insertTask(ctx, task)
    if err != nil {
        fmt.Println("insert error:", err)
        return
    }
    fmt.Fprintf(w, "%+v", task)
}
func (app *application) insertTask(ctx context.Context, t *Task) error {
    stmt := `INSERT INTO task (name, status) VALUES ($1, $2) RETURNING ID`
    row := app.DB.QueryRow(ctx, stmt, t.Name, t.Status)
    err := row.Scan(&t.ID)
    if err != nil {
        return err
    }
    return nil
}

func main() {
    rand.Seed(time.Now().UnixNano())
    db, err := pgxpool.Connect(context.Background(), "postgres://test:test123@localhost:5432/test")
    if err != nil {
        log.Fatal(err)
    }
    log.Println("db conn pool created")
    stmt := `CREATE TABLE IF NOT EXISTS public.task (
        id uuid NOT NULL DEFAULT gen_random_uuid(),
        name text NULL,
        status text NULL,
        PRIMARY KEY (id)
     ); `
    _, err = db.Exec(context.Background(), stmt)
    if err != nil {
        log.Fatal(err)
    }
    log.Println("task table created")
    defer db.Close()
    app := &application{
        DB: db,
    }
    mux := http.NewServeMux()
    mux.HandleFunc("/create", app.create)

    log.Println("http server up at localhost:4444")
    err = http.ListenAndServe(":4444", mux)
    if err != nil {
        log.Fatal(err)
    }
}