Go语言实现聊天小工具的示例代码

小工具分为server端和client端,废话不多说,代码如下

代码实现

server.go

package main
 
import (
        "bufio"
        "fmt"
        "net"
        "os"
        "strings"
)
 
func process(conn net.Conn) {
        defer conn.Close()
        var tmp [128]byte
        reader := bufio.NewReader(os.Stdin)
        for {
                n, err := conn.Read(tmp[:]) // 读取连接中的数据
                if err != nil {
                        fmt.Printf("read failed err:%v\n", err)
                        break
                }
                fmt.Println("Clent:", string(tmp[:n]))
                fmt.Print("Server:")
                text, _ := reader.ReadString('\n')
                msg := strings.TrimSpace(text)
                conn.Write([]byte(msg))
        }
}
 
func main() {
        // 监听tcp端口
        listen, err := net.Listen("tcp", "127.0.0.1:10086")
        if err != nil {
                fmt.Printf("start server failed err:%v\n", err)
                return
        }
        // for循环实现建立多个连接
        for {
                conn, err := listen.Accept() // 建立连接
                if err != nil {
                        fmt.Printf("accept failed err:%v\n", err)
                        return
                }
                // 每个连接建立一个goroutine处理连接
                go process(conn)
        }
}

client.go

package main
 
import (
        "bufio"
        "fmt"
        "net"
        "os"
        "strings"
)
 
func main() {
        // 与服务端建立连接
        conn, err := net.Dial("tcp", "127.0.0.1:10086")
        if err != nil {
                fmt.Printf("conn failed err:%v", err)
                return
        }
        var msg string
        var tmp [128]byte
        reader := bufio.NewReader(os.Stdin)
        for {
                fmt.Print("Client:")
                text, _ := reader.ReadString('\n')
                msg = strings.TrimSpace(text) // 识别字符串中的特别字符并去掉开头末尾空格
                if msg == "exit" {
                        break
                }
                // 往连接里写入数据
                conn.Write([]byte(msg))
                n, err := conn.Read(tmp[:]) // 读取连接中的数据
                if err != nil {
                        fmt.Printf("read failed err:%v", err)
                        return
                }
                fmt.Println("Server:", string(tmp[:n]))
        }
        conn.Close()
}

运行效果

D:\Software\godata\src\demo\server>go run main.go

Clent: 你好

Server:hello

Clent: 很高兴见到你

Server:我也是

Clent: 我是你爸爸

Server:滚

原文地址:https://blog.csdn.net/weixin_44208042/article/details/112632889