在多线程的线程里边，用一个线程处理一条连接，如何判断连接已经关闭？
试了一下，如果连接中断，读写会出现这种net.OpError，这个就可以判断是否断开
但是我也不确定有没有更好的方法？如果有的话，希望朋友们不吝赐教
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
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: 以下代码会产生死循环
代码如下:
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
package mainimport ( "fmt")func main() { c := make(chan int, 10) c <- 1 c <- 2 c <- 3 close(c) for { fmt.Println(<-c) }} |
判断短chan是否关闭
代码如下:
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
package mainimport ( "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) }} |
以上为个人经验，希望能给大家一个参考，也希望大家多多支持服务器之家。如有错误或未考虑完全的地方，望不吝赐教。
原文链接：https://blog.csdn.net/zhenshanren/article/details/103646626
