Go实现聊天小工具

╰+攻爆jí腚メ 2023-01-06 14:52 128阅读 0赞

小工具分为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:滚