Go中连续请求更新用户接口出现context deadline exceeded问题排查

连续请求/admin/update-user接口到第9次时出现context deadline exceeded，且阻塞其他接口，根源在于数据库连接池耗尽：

• GetUserGroupsByUsername函数执行Query后未关闭rows，导致数据库连接无法及时释放回连接池。
• PostgreSQL默认连接池最大连接数通常为10，当连续9次请求占用连接后，第10次请求无法获取可用连接，等待超时，进而阻塞所有依赖数据库的接口。

1. 强制释放数据库连接 #

在GetUserGroupsByUsername中添加defer rows.Close()，确保无论查询成功/失败、提前返回，都能释放连接：

func (m *PostgresDBRepo) GetUserGroupsByUsername(ctx context.Context, username string, userGroups ...string) (bool, error) {
    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() // 关键：确保连接释放回池

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

        if utils.Contains(userGroups, userGroup) {
            return true, nil
        }
    }
    // 检查遍历结果集时的潜在错误
    if err = rows.Err(); err != nil {
        return false, err
    }
    return false, nil // 原代码此处返回true是逻辑错误，未匹配到用户组应返回false
}

2. 优化查询逻辑（可选） #

修改SQL语句直接判断用户是否属于目标用户组，减少数据传输量，提升效率：

var SQL_GET_USERGROUPS_BY_USERNAME = `SELECT EXISTS(
    SELECT 1 FROM user_groups ug
    JOIN user_group_mapping ugm ON ugm.user_group_id = ug.user_group_id 
    JOIN users u ON u.user_id = ugm.user_id
    WHERE u.username = $1 AND ug.user_group = ANY($2)
)`

对应Go代码调整：

func (m *PostgresDBRepo) GetUserGroupsByUsername(ctx context.Context, username string, userGroups ...string) (bool, error) {
    var exists bool
    err := m.DB.QueryRow(ctx, SQL_GET_USERGROUPS_BY_USERNAME, username, userGroups).Scan(&exists)
    if err != nil {
        log.Println("Query failed at GetUserGroupsByUsername:", err)
        return false, err
    }
    return exists, nil
}

这种方式直接返回布尔结果，无需遍历数据集，从根源避免连接泄漏风险。

3. 调整数据库连接池配置 #

根据服务并发需求，初始化数据库连接时设置合理的连接池参数：

db, err := sql.Open("postgres", "your-dsn-string")
if err != nil {
    log.Fatal(err)
}
db.SetMaxOpenConns(20) // 最大打开连接数
db.SetMaxIdleConns(10) // 最大空闲连接数
db.SetConnMaxLifetime(time.Hour) // 连接最大存活时长

内容的提问来源于stack exchange，提问作者Jessica