在多线程的线程里边，用一个线程处理一条连接，如何判断连接已经关闭？
试了一下，如果连接中断，读写会出现这种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 main
import (
"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 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)
}
}
|
以上为个人经验，希望能给大家一个参考，也希望大家多多支持服务器之家。如有错误或未考虑完全的地方，望不吝赐教。
原文链接：https://blog.csdn.net/zhenshanren/article/details/103646626