Golang TCPConn Gob Communication

时间:2017-06-22 16:40:20

标签: go tcp gob

我遇到了gob协议的问题(或者通常是网络,我的知识很弱),我不明白为什么以下代码无法正常工作。它应该是一个维护开放TCP连接并通过它发送多个gobs的简单示例。代码将发送和接收,但通常会破坏其数据。 提前谢谢。

package main

import (
    "encoding/gob"
    "fmt"
    "net"
    "strconv"
    "time"
)

type Message struct {
    Msg string
}

func main() {
    gob.Register(new(Message))

    clientAddr, err := net.ResolveTCPAddr("tcp", "localhost:12346")
    if err != nil {
        fmt.Println(err)
    }
    serverAddr, err := net.ResolveTCPAddr("tcp", "localhost:12345")
    if err != nil {
        fmt.Println(err)
    }

    serverListener, err := net.ListenTCP("tcp", serverAddr)
    if err != nil {
        fmt.Println(err)
    }
    conn, err := net.DialTCP("tcp", clientAddr, serverAddr)
    if err != nil {
        fmt.Println(err)
    }
    serverConn, err := serverListener.AcceptTCP()
    if err != nil {
        fmt.Println(err)
    }
    done := false
    go func() {
        for !done {
            recieveMessage(serverConn)
        }
    }()
    for i := 1; i < 1000; i++ {
        sent := Message{strconv.Itoa(i)}
        sendMessage(sent, conn)
    }
    time.Sleep(time.Second)
    done = true
}

func sendMessage(msg Message, conn *net.TCPConn) {
    enc := gob.NewEncoder(conn)
    err := enc.Encode(msg)
    if err != nil {
        fmt.Println(err)
    }
}

func recieveMessage(conn *net.TCPConn) {
    msg := new(Message)
    dec := gob.NewDecoder(conn) // Will read from network.
    err := dec.Decode(msg)
    if err != nil {
        fmt.Println(err)
    }
    fmt.Println("Client recieved:", msg.Msg)
}

1 个答案:

答案 0 :(得分:3)

问题是解码器可以缓冲下一条消息的数据。发生这种情况时,下一个新解码器将在消息中间启动。解决方法是使用单个编码器和解码器。

func main() {
    ...
    dec := gob.NewDecoder(conn) // Will read from network.
    enc := gob.NewEncoder(serverConn)
    go func() {
        for !done {
            recieveMessage(dec)
        }
    }()

    for i := 1; i < 1000; i++ {
        sent := Message{strconv.Itoa(i)}
        sendMessage(sent, enc)
    }
    ...
}

func sendMessage(msg Message, enc *gob.Encoder) {
    err := enc.Encode(msg)
    if err != nil {
        fmt.Println(err)
    }
}

func recieveMessage(dec *gob.Decoder) {
    msg := new(Message)
    err := dec.Decode(msg)
    if err != nil {
        fmt.Println(err)
    }
    fmt.Println("Client recieved:", msg.Msg)
}

Run it in the playground