Golang 文件上传中如何使用 context 超时？-Golang-PHP中文网

Golang 文件上传中如何使用 context 超时？

王林

发布： 2024-05-13 21:39:02

原创

1286人浏览过

go 中文件上传使用 context 超时可以防止服务器长时间等待客户端完成上传。方法包括：1）创建一个新的 context 对象，设置超时值；2）将 context 对象传递给文件操作；3）使用 ctx.err() 检查是否因超时取消操作。实际示例：1）设定上传超时；2）解析表单；3）处理文件；4）检查是否因超时取消操作。该示例确保上传在 10 秒内完成，否则返回超时错误。

Golang 文件上传中如何使用 context 超时？

Go 中文件上传时使用 context 超时

在 Go 中使用 context 包设置超时对于处理文件上传场景至关重要。它使我们能够限制上传操作的时间，防止服务器长时间等待客户端完成上传。

使用方法

立即学习“go语言免费学习笔记（深入）”；

易森网络企业版

如果您是新用户，请直接将本程序的所有文件上传在任一文件夹下，Rewrite 目录下放置了伪静态规则和筛选器，可将规则添加进IIS，即可正常使用，不用进行任何设置；(可修改图片等)默认的管理员用户名、密码和验证码都是：yeesen系统默认关闭，请上传后登陆后台点击“核心管理”里操作如下:进入“配置管理”中的&ld

查看详情

可以使用以下步骤在文件上传中使用 context 超时：

创建一个新的 context 对象，设置一个适当的超时值：

ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()

登录后复制

将 context 对象传递给处理上传的文件操作，例如 http.Request：

// 根据 ctx 处理上传的文件
if err := handler.HandleUpload(req.Context(), req); err != nil {
    // 根据错误做出响应
}

登录后复制

使用 ctx.Err() 检查操作是否因超时而取消：

// 检查是否因超时而取消
if ctx.Err() == context.DeadlineExceeded {
    // 根据超时做出响应
}

登录后复制

实战案例

以下是一个使用 context 超时的文件上传实际示例：

package main

import (
    "context"
    "net/http"
    "time"
)

// 设定文件上传超时为 10 秒
const uploadTimeout = 10 * time.Second

type handler struct{}

func (h *handler) HandleUpload(ctx context.Context, r *http.Request) error {
    // 解析上传的表单
    if err := r.ParseMultipartForm(int64(5e6)); err != nil {
        return err
    }

    // 处理上传的文件
    // ...

    // 检查是否因超时而取消
    if ctx.Err() == context.DeadlineExceeded {
        return http.ErrRequestTimeout
    }

    return nil
}

func main() {
    http.Handle("/upload", &handler{})

    http.ListenAndServe(":8080", nil)
}

登录后复制

在该示例中，我们将文件上传超时的设置为 10 秒，如果在该时间内未完成上传，则返回超时错误。

以上就是Golang 文件上传中如何使用 context 超时？的详细内容，更多请关注php中文网其它相关文章！