最近有写socket相关的东西,刚好有人问我这个问题,我就记录一下,发现这个问题还是有很多人不了解的。
在socket-tcp的代码中大致这两种接受数据的方法,con.Read以及ioutil.ReadAll,这两种方法的区别是什么,以及他们的使用方法。
区别
func (c *conn) Read(b []byte) (int, error)
在socket中,Read方法直接读取一段指定长度的字节,当然这个方法也是阻塞的,如果对面没有发送数据,他会一直等到超时,
ioutil.ReadAll
他一直读取直到读取到err或eof,就是出错或者到文件结束标志,一般用来一次性获取所有的流数据。
问题发生
先看这段代码read部分,双方建立了长连接后,如果conn.Read(request)如果换成ioutil.ReadAll方法来读取,每次读取到这个ReadAll的时候,居然阻塞了???这是为什么。一直等到超时断开连接,才能收到这个数据。
解决过程
使用read方法输出每次的字节流内容,发现并没有EOF方法,只有当连接断开,他才读取到,或者说读取到长度为0,这就明白了,这里在连接中,ioutil.ReadAll并没有接受到结束标志,也就是说这个for中客户端可能会一直在发数据,所以ioutil.ReadAll会一直读取,直到他断开连接或者手动写入结束标识。
如果要使用ioutil.ReadAll,那就不能这么使用长连接,要一次性使用的话连接后发送完数据就断开即可,类似于http请求一次,看实际需求。
服务端代码如下
package main
import (
"encoding/json"
"fmt"
"net"
"os"
"time"
)
func main() {
tcpAddr, err := net.ResolveTCPAddr("tcp4", ":1200")
checkError(err)
listener, err := net.ListenTCP("tcp", tcpAddr)
checkError(err)
for {
conn, err := listener.Accept()
if err != nil {
continue
}
go handleClient(conn)
}
}
func handleClient(conn net.Conn) {
conn.SetReadDeadline(time.Now().Add(2 * time.Minute)) // set 2 minutes timeout
log.Println("adress:", conn.LocalAddr().String())
defer conn.Close() // close connection before exit
for {
conn.Write([]byte("123"))
request := make([]byte, 128)
readLen, err := conn.Read(request)
if readLen == 0 {
break // connection already closed by client
}
log.Println(string(request))
if err != nil {
break
}
}
}
func checkError(err error) {
if err != nil {
fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
os.Exit(1)
}
}
客户端代码如下
package main
import (
"fmt"
"net"
"os"
"time"
)
func main() {
tcpAddr, err := net.ResolveTCPAddr("tcp4", ":1200")
checkError(err)
conn, err := net.DialTCP("tcp", nil, tcpAddr)
conn.Write([]byte("fi:1234"))
checkError(err)
for {
result := make([]byte, 256)
_, err = conn.Read(result)
checkError(err)
fmt.Println("get mes:", string(result))
time.Sleep(time.Second * 1)
conn.Write([]byte("this is fi" + time.Now().Format("2006-01-02 15:04:05")))
}
}
func checkError(err error) {
if err != nil {
fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
os.Exit(1)
}
}
总结
注意:理论上con.Read和ioutil.ReadAll都是可以读取,但是ioutil.ReadAll要收到error或者EOF才会返回,也就是说如果发送端,发送数据后,调用 Close 关闭连接,不等待服务端的返回数据,服务端可 以用 ioutil.ReadAll 来读取数据,这时可以判断出 EOF,读取结束。但如果客户端发送数据 后,没有关闭,而是等待服务端的数据返回,用 ReadAll 是不行的,你们长连接交互的话,就用read。一定要用ioutil.ReadAll的话,试着手动写入EOF结束符或者加入其他逻辑判断区分。
有疑问加站长微信联系(非本文作者)