https://github.com/gin-gonic/gin使用 HTTP 服務(wù)器。是否可以處理客戶端套接字關(guān)閉？在處理程序中 router.GET("/v1/get/*uri", func (c *gin.Context) { var url = c.Param("uri") if url[0:1] == "/" { url = url[1:] } req, err := http.NewRequest("GET", url, nil)我向 url 發(fā)出 HTTP 客戶端請求。如果我從我的 wget/curl 客戶端關(guān)閉與 gin 服務(wù)器的連接，gin HTTP 客戶端請求仍會被處理。我想中斷客戶端套接字?jǐn)嚅_事件上的處理程序執(zhí)行，以免浪費(fèi)資源。
1 回答
寶慕林4294392
TA貢獻(xiàn)2021條經(jīng)驗(yàn) 獲得超8個贊
如果客戶端斷開連接，則 gin.Context.Done() 返回。http.Transport.CancelRequest() 中斷對 url 的 http 請求。所以 http 請求是在 goroutine 中發(fā)出的，我們等待 select 中的第一個事件：
http請求完成
客戶端與 gin 服務(wù)器斷開連接
router.GET("/v1/get/*uri", func (c *gin.Context) { var url = c.Param("uri") if url[0:1] == "/" { url = url [1:]}
tr := &http.Transport{ TLSClientConfig : (&tls.Config{ InsecureSkipVerify: true}),
DisableKeepAlives : false,
DisableCompression : false,
MaxIdleConns : 0,
MaxConnsPerHost : 0,
IdleConnTimeout : 0,
ResponseHeaderTimeout : 0,
ExpectContinueTimeout : 0,
ForceAttemptHTTP2 : true}
client := http.Client{ Timeout: 20 * time.Second, Transport : tr}
req, err := http.NewRequest("GET", url, nil)
if err != nil {
fail(1, "NewRequest")
}
for k, v := range c.Request.Header {
req.Header.Set(k, strings.Join(v, "\n"))
}
is_url_done := make(chan HTTPRequestResult, 1)
go func() {
resp, err := client.Do(req)
is_url_done <- HTTPRequestResult{res, err}
}()
for {
select {
case <-c.Done():
tr.CancelRequest(req)
return
case r:= <-is_url_done:
if r.err != nil {
c.String(http.StatusInternalServerError, "Error: %v\n", r.err)
} else {
var headers = map[string]string{}
for key, values := range(r.res.Header) {
headers[key] = strings.Join(values, ", ")
}
c.DataFromReader(r.res.StatusCode, r.res.ContentLength, "test", r.res.Body, headers)
}
return
}
}
- 1 回答
- 0 關(guān)注
- 275 瀏覽
添加回答
舉報(bào)
0/150
提交
取消