golang中io.Copy()问题

cfanbo
  • 658
package main

import (
    "net"
    "log"
    "fmt"
    "os"
    "io"
)

func main() {
    l, err := net.Listen("tcp", "127.0.0.1:8888")
    defer func() {
        if err := recover(); err != nil {
            log.Fatal(err)
        }
    }()
    if err != nil {
        log.Fatal(err)
    }

    // pxy
    dstConn, err := net.Dial("tcp", "192.168.199.201:9999")
    if err != nil {
        log.Fatal("连接后端失败:", err)
    }
    defer dstConn.Close()
    defer l.Close()
    for {
        conn, err := l.Accept()
        if err != nil || err == io.EOF {
            fmt.Println("accept error:", err)
            os.Exit(1)
        }
        fmt.Printf("message %s; %s\n", conn.RemoteAddr(), conn.LocalAddr())

        go handleConn(conn, dstConn)
    }

}

func handleConn(conn net.Conn, dstConn net.Conn) {
    ip := conn.RemoteAddr().String()
    defer func() {
        fmt.Println("disconnect:", ip)
        conn.Close()
    }()

    fmt.Println("ip:", ip)


    data := make([]byte, 128)
    for {
        _, err := conn.Read(data)
        if err != nil || err == io.EOF {
            log.Println("连接已断开!")
            break
        }

        // fmt.Printf("收到%d个字符, %s\n", len, data)
        // conn.Write(data)

        // 转发
        go io.Copy(conn, dstConn)
        io.Copy(dstConn, conn)    
    }

}

发现客户端第一次发送的消息总是不转发,以后的消息才转发,问题在哪里?

回复
阅读 2.6k
2 个回答
✓ 已被采纳
_, err := conn.Read(data)

这里读出来的并没有转发

这不是先读取了一段数据才ioCopy 的吗?

 _, err := conn.Read(data)
        if err != nil || err == io.EOF {
            log.Println("连接已断开!")
            break
        }
撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
宣传栏