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

Question

更新 1:似乎使用绑定(bind)到 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 之间正确设置和传递 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)
    }
}

Answer 1

TLDR:使用 r.Context() 在生产中工作正常，使用浏览器进行测试是个问题。

HTTP 请求获得自己的上下文，当请求完成时该上下文被取消。这是一个特性，而不是一个错误。当请求被客户端中断或超时时，开发人员应该使用它并优雅地关闭执行。例如，取消的请求可能意味着客户端永远不会看到响应(交易结果)，开发人员可以决定回滚该交易。

在生产中，对于正常设计/构建的 API，请求取消不会经常发生。通常，流程由服务器控制，服务器在取消请求之前返回结果。多个Client请求不会互相影响，因为他们得到独立的go-routine和context。同样，我们正在谈论正常设计/构建应用程序的快乐路径。您的示例应用看起来不错，应该可以正常工作。

问题是我们如何测试应用程序。我们使用浏览器并刷新单个浏览器 session ，而不是创建多个独立的请求。我没有检查到底发生了什么，但假设浏览器终止了现有请求，以便在您单击 ctrl-R 时运行新请求。服务器看到该请求终止并将其作为上下文取消传达给您的代码。

尝试使用 curl 或其他创建独立请求的脚本/实用程序来测试您的代码。我相信在这种情况下您不会看到取消。