在HTTP处理程序中执行延迟 X 秒

为了在 HTTP 处理程序中执行延迟，可以使用 context.Context 来设置超时。通过使用通道，可以实现超时功能，并在给定的时间内停止处理函数。这可以确保在指定的时间后返回结果，即使数据库调用被阻塞。同时，为了确保处理程序始终在指定时间后立即返回，可以将数据库代码推送到另一个 goroutine 中，并在稍后返回结果。

我想运行函数 insertrecords 30 秒，并测试在给定时间内可以插入多少条记录。

如何在 x 秒后停止处理 insertrecords，然后从处理程序返回结果？

func benchmarkHandler(w http.ResponseWriter, r *http.Request) {

    counter := InsertRecords()

    w.WriteHeader(200)
    io.WriteString(w, fmt.Sprintf("counter is %d", counter))    
}

func InsertRecords() int {
  counter := 0
  // db code goes here
  return counter
}

解决方案

取消和超时通常通过 context.Context 完成。

虽然这个简单的示例可以单独使用通道来完成，但使用此处的上下文使其更加灵活，并且还可以考虑客户端断开连接。

func benchmarkhandler(w http.responsewriter, r *http.request) {
    ctx, cancel := context.withtimeout(r.context(), 30*time.second)
    defer cancel()

    counter := insertrecords(ctx)

    w.writeheader(200)
    io.writestring(w, fmt.sprintf("counter is %d", counter))
}

func insertrecords(ctx context.context) int {
    counter := 0
    done := ctx.done()
    for {
        select {
        case <-done:
            return counter
        default:
        }

        // db code goes here
        counter++
    }
    return counter
}

这将运行至少 30 秒，返回完整的数据库迭代次数。如果你想确保处理程序总是在 30 秒后立即返回，即使 db 调用被阻塞，那么你需要将 db 代码推送到另一个 goroutine 中并让它稍后返回。最短的示例是使用与上面类似的模式，但同步对计数器变量的访问，因为它可以在返回时由数据库循环写入。

func InsertRecords(ctx context.Context) int {
    counter := int64(0)
    done := ctx.Done()

    go func() {
        for {
            select {
            case <-done:
                return
            default:
            }

            // db code goes here
            atomic.AddInt64(&counter, 1)
        }
    }()

    <-done
    return int(atomic.LoadInt64(&counter))
}

请参阅 @joshuakolden 的回答，了解带有生产者和超时的示例，它也可以与现有的请求上下文相结合。

好了，本文到此结束，带大家了解了《在HTTP处理程序中执行延迟 X 秒》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！