golang判断net.Conn 是否已关闭的操作
程序员文章站
2022-07-06 13:36:38
在多线程的线程里边,用一个线程处理一条连接,如何判断连接已经关闭?试了一下,如果连接中断,读写会出现这种net.operror,这个就可以判断是否断开但是我也不确定有没有更好的方法?如果有的话,希望朋...
在多线程的线程里边,用一个线程处理一条连接,如何判断连接已经关闭?
试了一下,如果连接中断,读写会出现这种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) } }
以上为个人经验,希望能给大家一个参考,也希望大家多多支持。如有错误或未考虑完全的地方,望不吝赐教。