收听1台服务器上的2个端口

时间:2016-03-28 21:03:44

标签: networking go tcp chat

我正在尝试修改我的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
        }

    }
}

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

1 个答案:

答案 0 :(得分:4)

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

要使用两个端口,请创建两个侦听器并在单独的goroutine中为这些侦听器运行accept循环:

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来处理错误。这可能适用于您的申请,也可能不适合。