如果连接丢失，golang gin 如何立即停止处理程序

小伙伴们有没有觉得学习Golang很有意思？有意思就对了！今天就给大家带来《如果连接丢失，golang gin 如何立即停止处理程序》，以下内容将会涉及到，若是在学习中对其中部分知识点有疑问，或许看了本文就能帮到你！

我正在使用 gin-gonic/gin 来编写我的服务器。

看起来即使连接丢失，处理函数仍在运行。例如，如果我访问http://127.0.0.1:8080/ping并突然关闭浏览器，屏幕将继续打印所有数字。

package main

import (
    "github.com/gin-gonic/gin"
    "log"
    "time"
)

func main() {
    r := gin.Default()
    r.GET("/ping", func(c *gin.Context) {
        for i := 1; i < 15; i++ {
            time.Sleep(time.Second * 1)
            log.Println(i)
        }
        c.JSON(200, gin.H{
            "message": "pong",
        })
    })
    r.Run("127.0.0.1:8080")
}

我应该如何立即停止处理函数（例如减少服务器负载）？

解决方案

当客户端断开连接时，request context 会被取消，因此只需检查 c.Done() 是否准备好接收即可：

package main

import (
    "log"
    "time"

    "github.com/gin-gonic/gin"
)

func main() {
    r := gin.Default()
    r.GET("/ping", func(c *gin.Context) {
        t := time.NewTicker(1 * time.Second)
        defer t.Stop()

        for i := 1; i < 15; i++ {
            select {
            case <-t.C:
                log.Println(i)
            case <-c.Request.Context().Done():
                // client gave up
                return
            }
        }

        c.JSON(200, gin.H{
            "message": "pong",
        })
    })
    r.Run("127.0.0.1:8080")
}

本篇关于《如果连接丢失，golang gin 如何立即停止处理程序》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！