如果失去连接,golang gin停止处理程序将如何立即起作用

时间:2018-11-15 15:49:08

标签: go gin gin-gonic

我正在使用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")
}

我应如何立即停止处理程序功能(例如,以减少服务器负载)?

1 个答案:

答案 0 :(得分:1)

当客户端断开连接时,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")
}