go语言实现聊天室简单版

2024-11-16 技术教程

go语言实现聊天室初级版

聊天室分为服务端和客户端。服务端主动开启监听等待客户端连接，客户端向服务端发送连接请求，建立连接，发送消息，实现与服务端的通信。下面使用go语言实现一个简单版的聊天室。第一部分为服务端代码，第二部分为客户端代码。

一、服务端代码

伪代码：
1、服务端开启监听
2、等待客户端连接
3、接收客户端连接
4、处理客户端连接，输出消息

//chatroom serverpackage main//导入包 import ( "fmt" "net" )//定义函数checkError，用来错误处理func checkError(err error) { if err != nil { panic(err) }}//定义一个函数，专门负责接收信息的协程func processInfo(conn net.Conn) {//定义一个变量用来存储数据 buf := make([]byte, 1024) //最后需要关闭连接 defer conn.Close() //无限循环读取接收到字节数，直到读取完后退出循环 for { //读取接收到的字节数 numOfBytes, err := conn.Read(buf) if err != nil { break } //如果接收的字节数不为0，说明有消息发过来，并打印出消息 if numOfBytes != 0 { fmt.Printf("Has received message: %s", string(buf)) } }}//主函数入口 func main() { //开启监听，监听协议为tcp,监听地址及端口为127.0.0.1:8888 listen_socket, err := net.Listen("tcp","127.0.0.1:8888") checkError(err) //最后关闭套接字连接 defer listen_socket.Close() //没有消息进来时默认输出:chat room Server is waiting... fmt.Println("chat room Server is waiting... ") //接收客户端连接 for { conn, err := listen_socket.Accept() checkError(err) //如果有客户端连接，则打开一个协程处理 go processInfo(conn) } }

执行go build server.go或者go run server.go将会输出：chat room Server is waiting... ，等待客户端连接

二、客户端代码

伪代码
1、客户端向服务端发送连接请求
2、客户端向服务端发送信息

// chat room clientpackage main//导入包 import ( "fmt" "net" )func checkError(err error) { if err != nil { panic(err) }}func main() { //使用Dial函数和服务端建立连接 conn, err := net.Dial("tcp","127.0.0.1:8888") checkError(err) //最后退出时关闭连接 defer conn.Close() //向服务端发送信息 conn.Write([]byte("Now connection.success!")) fmt.Println("Has sent the message!")}