请求上下文期限已过

对于一个Golang开发者来说，牢固扎实的基础是十分重要的，golang学习网就来带大家一点点的掌握基础知识点。今天本篇文章带大家了解《请求上下文期限已过》，主要介绍了，希望对大家的知识积累有所帮助，快点收藏起来吧，否则需要时就找不到了！

我正在处理 go 中的请求上下文。我正在使用此端点来更新用户帐户。当我连续向同一端点发送多个传入请求 (postman) 时，它返回错误 context 最后期限超出 。我不确定为什么只有当我连续向同一端点发送相同的请求时才会发生这种情况。 context deadline超过错误在第9个请求左右出现，并将在1秒后出现。之后，它也会阻止对其他端点的所有请求。

routes.go [每个请求都会有一个请求上下文 - requestmiddleware]

func routes(app repository.databaserepo) http.handler {

    r := chi.newrouter()

    // creates request context here with requestid
    r.use(appmiddleware.requestmiddleware)

    updateuserh := admin.new(app)
    r.route("/admin", func(r chi.router) {
        r.patch("/update-user", utils.makehttphandler(updateuserh.updateuser))
    })

    return r
}

requestmiddleware.go [使用 requestid 创建上下文]

type requestidkey string
const requestid requestidkey = "request_id"

func requestmiddleware(next http.handler) http.handler {
    return http.handlerfunc(func(w http.responsewriter, req *http.request) {
        requestiduuid := uuid.new().string()
        ctx := context.withvalue(req.context(), requestid, requestiduuid)
        next.servehttp(w, req.withcontext(ctx))
    })
}

updateuser.go [使用请求上下文并设置超时]

func (app application) updateuser(w http.responsewriter, req *http.request) error {

    ....

    ctx := req.context()
    ctx, cancel := context.withtimeout(ctx, 5*time.second)
    defer cancel()

    utilsapp := utils.application{db: app.db}
    err := utils.injectug(utilsapp, ctx, w.header().get("username"), "admin")
    if err != nil {
        return err
    }

    return utils.writejson(w, http.statusok, utils.apisuccess{success: "[admin] successfully updated '" + updateuser.username + "' user!", status: http.statusok})

}

更新 utils.injectug 函数正在调用此数据库查询。此查询导致整个上下文在触发多个传入请求时超出截止时间。当我注释掉 utils.injectug 时，它工作得很好。我需要此数据库查询来检查用户是否属于某个用户组才能访问特定资源。

type application struct {
    db repository.databaserepo
}

func injectug(app application, ctx context.context, username string, usergroups ...string) error {
    return app.checkusergroup(ctx, username, usergroups...)
}

// determines if a user has been assigned that usergroup
func (app application) checkusergroup(ctx context.context, username string, usergroups ...string) error {

    isauthorizeduser, err := app.db.getusergroupsbyusername(ctx, username, usergroups...)
    if err != nil {
        return apierror{err: "internal server error", status: http.statusinternalservererror}
    }
    if isauthorizeduser {
        return nil
    }

    return apierror{err: "access denied: user does not have permission to access this resource", status: http.statusforbidden}
}

var SQL_GET_USERGROUPS_BY_USERNAME = `SELECT ug.user_group FROM user_groups ug
    LEFT JOIN user_group_mapping ugm 
    ON ugm.user_group_id = ug.user_group_id 
    WHERE ugm.user_id = (SELECT user_id FROM users WHERE username = $1);`

func (m *PostgresDBRepo) GetUserGroupsByUsername(ctx context.Context, username string, userGroups ...string) (bool, error) {
    // IT FAILS ON THIS QUERY: context deadline exceeded
    rows, err := m.DB.Query(ctx, SQL_GET_USERGROUPS_BY_USERNAME, username)
    if err != nil {
        log.Println("Query failed at GetUserGroupsByUsername:", err)
        return false, err
    }

    var userGroup string
    if rows != nil {
        for rows.Next() {
            if err = rows.Scan(&userGroup); err != nil {
                return false, err
            }

            if utils.Contains(userGroups, userGroup) {
                return true, nil
            }
        }
    }
    return true, nil
}

正确答案

[奇怪的是我找不到重复的问题]

应显式关闭 rows。

修复方法是在错误检查后添加 defer rows.close()：

rows, err := m.db.query(ctx, sql_get_usergroups_by_username, username)
  if err != nil {
    log.println("query failed at getusergroupsbyusername:", err)
    return false, err
  }
+ defer rows.close()

参考文献：

• Querying for multiple rows

  一直循环行也会隐式关闭它，但最好使用 defer 来确保行无论如何都关闭。

• https://pkg.go.dev/github.com/jackc/pgx/v5#Conn.Query

  必须先关闭返回的行，然后才能再次使用连接。

这是一个重现该问题的简单演示：

package main

import (
    "context"
    "log"
    "time"

    "github.com/jackc/pgx/v5/pgxpool"
    "golang.org/x/exp/slices"
)

func getusergroupsbyusername(ctx context.context, db *pgxpool.pool, usergroups ...string) (bool, error) {
    rows, err := db.query(ctx, "values ('g1'),('g2')")
    if err != nil {
        log.println("query failed at getusergroupsbyusername:", err)
        return false, err
    }

    // uncomment the next line to address the issue.
    // defer rows.close()

    var usergroup string
    for rows.next() {
        if err = rows.scan(&usergroup); err != nil {
            return false, err
        }

        if slices.contains(usergroups, usergroup) {
            return true, nil
        }
    }
    log.println("error:", rows.err())
    return true, nil
}

func main() {
    connstring := "postgres://username:password@localhost:5432/dbname"
    pool, err := pgxpool.new(context.background(), connstring)
    if err != nil {
        log.fatalf("unable to connect to database: %v\n", err)
    }
    defer pool.close()

    ctx, cancel := context.withtimeout(context.background(), time.second)
    defer cancel()

    for i := 0; i < 10; i++ {
        log.println(getusergroupsbyusername(ctx, pool, "g1"))
    }
}

输出：

2023/05/08 09:10:33 true 
2023/05/08 09:10:33 true 
2023/05/08 09:10:33 true 
2023/05/08 09:10:33 true 
2023/05/08 09:10:33 true 
2023/05/08 09:10:33 true 
2023/05/08 09:10:33 true 
2023/05/08 09:10:33 true 
2023/05/08 09:10:34 Query failed at GetUserGroupsByUsername: context deadline exceeded
2023/05/08 09:10:34 false context deadline exceeded
2023/05/08 09:10:34 Query failed at GetUserGroupsByUsername: context deadline exceeded
2023/05/08 09:10:34 false context deadline exceeded
^Csignal: interrupt

今天关于《请求上下文期限已过》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！