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