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:滚