golang判斷net.Conn 是否已關閉的操作
在多線程的線程裡邊,用一個線程處理一條連接,如何判斷連接已經關閉?
試瞭一下,如果連接中斷,讀寫會出現這種net.OpError,這個就可以判斷是否斷開
但是我也不確定有沒有更好的方法?如果有的話,希望朋友們不吝賜教
func dealConn(conn net.Conn){ //defer conn.Close() //defer conn.Flush() //長連接裡邊的讀寫操作必須放到循環裡面這樣才能進行多次的讀寫 // 如果連接已經斷開,就把這個線程中斷掉,怎麼判斷這個連接已經斷開? thread_c:=0;//如果連續100秒中讀取不到內容,就終止循環 for{ defer func() { if r := recover(); r != nil { buf:=make([]byte,666) buf=buf[:runtime.Stack(buf,false)] log.Printf("運行時錯誤:%v.Runtime error caught: %s",r, buf) } }() // 註意continue這裡也要等待,不然造成內存耗盡,處理器耗盡 time.Sleep(50*time.Millisecond) //#log.Println(len,string(text)) thread_c++ if thread_c>20*100{ log.Println(conn.RemoteAddr(),"超過100秒未讀取到內容,本連接將關閉") conn.Close(); c--; break; } frame,op_err:=readAllShut(conn) if op_err!=nil{ log.Println(conn.RemoteAddr(),"出現讀寫錯誤,連接不可用,將會被關閉") conn.Close(); c--; break;//這種已經關閉的連接,要終止循環,退出這條線程 } if(len(frame)==0){ // //time.Sleep(50*time.Millisecond) continue } thread_c=0; log.Printf("-----------------收到tcp請求:報文的長度是%v,詳細內容如下:%s,轉換成16進制是:%x", len(frame),frame,frame) //TODO //這裡寫自己的業務代碼 } } func readAllShut(conn net.Conn) ([]byte,error){ //這個手動方法可以避免粘包的問題 //bufio.NewWriter re:=bytes.NewBuffer(nil) const N=666 for{ var text [N]byte lens,err:=conn.Read(text[0:]) re.Write(text[:lens]) if lens==0 || err!=nil{ //log.Println(err) //在這個死循環裡面,不要有任何的輸出 // if errors.As(err,*net.OpError) // if _,ok:=err.( *net.OpError) ;ok{ return nil,err } break } //conn //log.Println(lens,text) if lens<N{ break } } rb:= re.Bytes() //log.Println(rb,"len",len(rb)) return rb,nil /*data,err:=ioutil.ReadAll(conn) if err!=nil{ log.Printf("讀取出現錯誤%T:%v",err,err) } return data;*/ }
補充:Go — 判斷chan channel是否關閉的方法
如果不判斷chan是否關閉
Notice: 以下代碼會產生死循環
代碼如下:
package main import ( "fmt" ) func main() { c := make(chan int, 10) c <- 1 c <- 2 c <- 3 close(c) for { fmt.Println(<-c) } }
判斷短chan是否關閉
代碼如下:
package main import ( "fmt" ) func main() { c := make(chan int, 10) c <- 1 c <- 2 c <- 3 close(c) for { i, isClose := <-c if !isClose { fmt.Println("channel closed!") break } fmt.Println(i) } }
以上為個人經驗,希望能給大傢一個參考,也希望大傢多多支持WalkonNet。如有錯誤或未考慮完全的地方,望不吝賜教。
推薦閱讀:
- Golang 並發以及通道的使用方式
- 示例剖析golang中的CSP並發模型
- Go語言中的通道channel詳情
- Go語言如何輕松編寫高效可靠的並發程序
- Go緩沖channel和非緩沖channel的區別說明