golang socket tcp ioutil.ReadAll 阻塞问题

最近有写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结束符或者加入其他逻辑判断区分。


 

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值