【go】go协程 tcp接收数据问题,多个发送接收到同一个里面了
写了一个收发文件的tcp测试,代码如下
server端
package mainimport (
//"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 mainimport (
//"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】go协程 tcp接收数据问题,多个发送接收到同一个里面了 的全部内容, 来源链接: utcz.com/a/110528.html