go协程 tcp接收数据问题,多个发送接收到同一个里面了

  • Z时代
  • 2024-01-10
  • 分类：IT

写了一个收发文件的tcp测试,代码如下

server端

package main
import (
    //"bytes"
    "fmt"
    "io"
    "math/rand"
    "net"
    "os"
    "time"
)
func checkError(err error) {
    if err != nil {
        fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
        os.Exit(1)
    }
}
func main() {
    ln, err := net.Listen("tcp", ":8080")
    checkError(err)
    for {
        conn, err := ln.Accept()
        if err != nil {
            fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
            continue
        }
        go run(conn)
    }
}
func run(conn net.Conn) {
    var all_len int = 0
    buffer := make([]byte, 20480)
    rand.Seed(time.Now().Unix())
    filename := string(rand.Intn(100))
    writeFile, err := os.OpenFile(filename, os.O_RDWR|os.O_APPEND|os.O_CREATE, 0777)
    if err != nil {
        fmt.Println("create file error:", filename, err)
        return
    }
    defer writeFile.Close()
    for {
        lenght, err := conn.Read(buffer)
        if err != nil {
            if err == io.EOF {
                break
            }
            fmt.Fprintf(os.Stderr, "read buff error: %s", err.Error())
            return
        }
        fmt.Println("receive data lenght:", lenght)
        //_, err = writeFile.Write(buffer[:bytes.Index(buffer[:], []byte{0})])
        _, err = writeFile.Write(buffer[:lenght])
        if err != nil {
            fmt.Println("write file error", err)
            return
        }
        all_len += lenght
    }
    fmt.Println("write file done", all_len/1024/1024)
    return
}

client端

package main
import (
    //"bytes"
    "fmt"
    "io"
    "net"
    "os"
    //"time"
)
func checkError(err error) {
    if err != nil {
        fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
        os.Exit(1)
    }
}
func main() {
    ch := make(chan int)
    for i := 0; i < 3; i++ {
        go send(ch)
    }
    for w := range ch {
        fmt.Println(w)
    }
}
func send(ch chan int) {
    conn, err := net.Dial("tcp", "127.0.0.1:8080")
    checkError(err)
    fileHandle, err := os.Open("Visio2010_cn.exe")
    if err != nil {
        fmt.Println("open file ERROR", err)
        return
    }
    defer fileHandle.Close()
    io.Copy(conn, fileHandle)
    conn.Close()
    ch <- 1
}

问题:

1,我的意思是想同时发送3个文件的(发的是同一个,这个不重要),重要的是接受的时候只生成了一个文件,这个文件大小是发送的那个文件的3倍,接到的数据写到同一个文件里了,为什么会这样?

2,为什么数据包不是相等大小,有大有小,如图,是分包的原因吗?

3,以前没写过tcp的东西,代码有问题请多多指教

请输入图片描述

回答：

关键问题应该是出在这里：

rand.Seed(time.Now().Unix())
filename := string(rand.Intn(100))

因为调用 go 后三个 goroutine 可能在还没有创建文件时已经执行了，然后生成的随机数一致，导致文件是同一个。

一个解决方法是在外面生成随机数，并将文件名传递给 goroutine。还有就是在一个程序中，rand.Seed(time.Now().Unix()) 只需要初始化一次就够了。

回答：

想了一下,可能是发送的时候直接用io.copy发送数据流,没有区分,所以全都写一个文件里面去了,

有空了改下发送数据的时候分割文件,在每块数据前加文件信息,接收的时候分开处理

以上是 go协程 tcp接收数据问题,多个发送接收到同一个里面了 的全部内容， 来源链接： utcz.com/p/183686.html

回到顶部