首页 文章 网络编程 golang socket中tcp中的ioutil.ReadAll阻塞的问题
0
0
0
19

golang socket中tcp中的ioutil.ReadAll阻塞的问题

websocket ReadAll golang socket ioutil 问题

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

到此这篇关于“ golang socket中tcp中的ioutil.ReadAll阻塞的问题”的文章就介绍到这了,更多文章或继续浏览下面的相关文章,希望大家以后多多支持Go语言编程网!

相关文章

创建博客

开始创作
写作能提升自己能力,也能为他人分享知识。

在线教程

查看更多
  • Go入门指南

    Go入门指南

  • Go语言高级编程

    Go语言高级编程

  • Go Web 编程

    Go Web 编程

  • GO专家编程

    GO专家编程

  • Go语言四十二章经

    Go语言四十二章经

  • 数据结构和算法(Golang实现)

    数据结构和算法(Golang实现)

Go语言编程网

微信扫码关注订阅号


博客 资讯 教程 我的