从一个端口向另一个端口发送消息

时间:2016-04-18 01:05:33

标签: networking go tcp chat port

我有一个服务器和2个客户端,每个客户端都使用自己的端口。当客户端向服务器发送消息时,服务器将其回送给发送者。然后,服务器也应该将消息发送给另一个客户端。这就是我遇到问题的地方。

我相信我可能在writeMessage()中设置连接错误,但我不完全确定。目前,我的功能只是尝试在端口8080上写入客户端2以进行调试。

服务器:

package main

import (
    "fmt"
    "log"
    "net"
)

var message string = ""

func main() {
    fmt.Println("The server is listening on Port 3000 and 8080")
    listener, err := net.Listen("tcp", "localhost:3000")
    if err != nil {
        log.Fatal(err) 
    }

    listener2, err := net.Listen("tcp", "localhost:8080")
    if err != nil {
        log.Fatal(err)
    }

    go acceptLoop(listener)
    acceptLoop(listener2)


}

func acceptLoop(l net.Listener) {
    defer l.Close()
    for {
            c, err := l.Accept()
            if err != nil {
                log.Fatal(err)
            }
            fmt.Println("New connection found!")
            listenConnection(c)
    }
}

func listenConnection(conn net.Conn) {
        for {
                buffer := make([]byte, 1400)
                dataSize, err := conn.Read(buffer)
                if err != nil {
                    fmt.Println("Connection has closed")
                    return
                }

                //This is the message you received
                data := buffer[:dataSize]
                fmt.Print(string(data))
                conn.Write(data)
                message = string(data)
                writeMessage()
        }
}

func writeMessage(){
    conn2, e := net.Dial("tcp", "localhost:8080")
    if e != nil {
        log.Fatalln(e)
    }
    defer conn2.Close()
    conn2.Write([]byte(message))
}

客户1:

package main

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

func main() {
        conn, err := net.Dial("tcp", "localhost:3000")
        if err != nil {
                log.Fatalln(err)
        }

        go listenConnection(conn, err)
        writeMessage(conn, err)

}

func listenConnection(conn net.Conn, err error){
    for {
                buffer := make([]byte, 1400)
                dataSize, err := conn.Read(buffer)
                if err != nil {
                        fmt.Println("connection closed")
                        return
                }

                data := buffer[:dataSize]
                fmt.Print("Server: ", string(data))
        }
}

func writeMessage(conn net.Conn, err error){
    for {
        reader := bufio.NewReader(os.Stdin)
        //fmt.Print("Enter message: ")
        text, _ := reader.ReadString('\n')
        text = "3000: " + text

        _, err = conn.Write([]byte(text))
        if err != nil {
                log.Fatalln(err)
        }
    }
}

(我的第二个客户端与我的第一个客户端相同,只是端口3000被8080替换。)

为了复制问题,用户必须运行服务器代码,然后运行两个客户端。然后,客户端可以向服务器发送消息。

对此的任何帮助将不胜感激。我根本无法解决这个问题。希望一些更有经验的Go用户能够提供帮助!

1 个答案:

答案 0 :(得分:0)

你的逻辑似乎有缺陷,其中acceptloop调用listenconnection,它在无限循环内创建一个无限循环。我认为您需要使用用例方案重建系统。

如果我的理解是正确的,您希望服务器等待客户端连接。首先,client1进来,发送一些数据。服务器接受此连接,它只需要为每个客户端接受一次。此时,如果client2在线,则将数据发送到client2。所有这些都发生在无限循环中。

查看此代码 https://play.golang.org/p/_5AREG3jnG