在Go中檢測丟失連接的正確方法是什麼？

Question

我正在使用從移動客戶端讀取上傳的Go HTTP服務器實現。但是，由於長時間保持活動狀態，我遇到了問題，如果移動客戶端脫機（經常發生），服務器將掛起相當長時間的請求緩衝區。

檢測掉線連接並關閉輸入緩衝區的正確方法是什麼？

Answer 1

因爲我想在寫停止（快捷地只斷開連接，這樣我就可以記錄接收到的數據，到目前爲止，並允許客戶端恢復），ReadTimeout不適合我。

我在this gist找到了答案。您需要在連接本身上設置讀取。

包nettimeout

import (
    "net" 
    "time" 
) 

// Listener wraps a net.Listener, and gives a place to store the timeout 
// parameters. On Accept, it will wrap the net.Conn with our own Conn for us. 
type Listener struct { 
    net.Listener 
    ReadTimeout time.Duration 
    WriteTimeout time.Duration 
} 

func (l *Listener) Accept() (net.Conn, error) { 
    c, err := l.Listener.Accept() 
    if err != nil { 
     return nil, err 
    } 
    tc := &Conn{ 
     Conn:   c, 
     ReadTimeout: l.ReadTimeout, 
     WriteTimeout: l.WriteTimeout, 
    } 
    return tc, nil 
} 

// Conn wraps a net.Conn, and sets a deadline for every read 
// and write operation. 
type Conn struct { 
    net.Conn 
    ReadTimeout time.Duration 
    WriteTimeout time.Duration 
} 

func (c *Conn) Read(b []byte) (int, error) { 
    err := c.Conn.SetReadDeadline(time.Now().Add(c.ReadTimeout)) 
    if err != nil { 
     return 0, err 
    } 
    return c.Conn.Read(b) 
} 

func (c *Conn) Write(b []byte) (int, error) { 
    err := c.Conn.SetWriteDeadline(time.Now().Add(c.WriteTimeout)) 
    if err != nil { 
     return 0, err 
    } 
    return c.Conn.Write(b) 
} 

func NewListener(addr string, readTimeout, writeTimeout time.Duration) (net.Listener, error) { 
    l, err := net.Listen("tcp", addr) 
    if err != nil { 
     return nil, err 
    } 

    tl := &Listener{ 
     Listener:  l, 
     ReadTimeout: readTimeout, 
     WriteTimeout: writeTimeout, 
    } 
    return tl, nil 
} 

Answer 2

在服務器上設置一個合理的超時，例如：

srv := &http.Server{ 
    Addr: ":443", 
    ReadTimeout: time.Minute * 2, 
    WriteTimeout: time.Minute * 2, 
} 
log.Fatal(srv.ListenAndServeTLS(certFile, keyFile))