我正在使用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")
}
我应如何立即停止处理程序功能(例如,以减少服务器负载)?
答案 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")
}