networking - 在一台服务器上监听 2 个端口

Question

我正在尝试修改我的 echo 服务器程序以创建一个聊天客户端。目前，当我启动客户端时，我的服务器正在监听一个端口。然后，客户端可以键入并向服务器发送消息，服务器将回显。

但是，我希望能够将 2 个客户端连接到 2 个不同的端口，并让客户端通过服务器相互发送消息。有什么办法可以做到这一点？我假设第一步是监听 2 个端口而不是一个。

这是我目前所拥有的。

服务器:

package main

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

func main() {
        fmt.Println("The server is listening on Port 3000")
        listener, _ := net.Listen("tcp", "localhost:3000")
        //listener2, _ := net.Listen("tcp", "localhost:8080")

        defer listener.Close()
        //defer listener2.Close()

        // Listen for connections
        for {
                conn, _ := listener.Accept()
                //conn2, _ := listener2.Accept()
                fmt.Println("New connection found!")

                go listenConnection(conn)
                //go listenConnection(conn2)
        }
}

//Listen for messages and reply
func listenConnection(conn net.Conn) {
        fmt.Println("Yay")
        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("Received message: ", string(data))

                // Send the message back
                _, err = conn.Write(data)
                if err != nil {
                        log.Fatalln(err)
                }
                fmt.Print("Message sent: ", string(data))
        }
}

客户:

package main

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

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

for {
        reader := bufio.NewReader(os.Stdin)
        fmt.Print("Enter text: ")
        text, _ := reader.ReadString('\n')

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

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

                data := buffer[:dataSize]
                fmt.Println("Received message: ", string(data))
                break
        }

    }
}

有没有办法在一台服务器上监听2个客户端(端口)并让它们通信？是否可以在同一端口上对两个客户端执行此操作？我尝试在服务器程序中添加另一个监听器，但我暂时将这些行注释掉，因为它们不起作用。我将不胜感激任何帮助!

Answer 1

问题中的服务器代码处理同一端口上的多个客户端。

要使用两个端口，创建两个监听器并在单独的 goroutine 中为这些监听器运行接受循环:

func main() {
    fmt.Println("The server is listening on Port 3000")
    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)  // run in the main goroutine
}

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

另外，不要忽略错误。此答案中的代码通过调用 log.Fatal 来处理错误。这可能适合也可能不适合您的应用。