go语言实现聊天室
聊天室分为服务端和客户端。第一部分为服务端代码,第二部分为客户端代码。
一、服务端代码
//chatroom server
package 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() {
//开启监听
listen_socket, err := net.Listen("tcp","127.0.0.1:8888")
checkError(err)
defer listen_socket.Close()
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... ,等待客户端连接
二、客户端代码
// chat room client
package main
import (
"fmt"
"net"
)
func checkError(err error) {
if err != nil {
panic(err)
}
}
func main() {
//连接服务端的信息
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!")
}
执行go run client.go或者go build client.go将会输出:Has sent the message!
此时客户端已向服务端发送信息:Now connection.success
我们再查看服务端的输出终端,此时服务端输出如下内容,那就表示客户端与服务端通信成功。
chat room Server is waiting...
Has received message: Now connection.success!
有疑问加站长微信联系(非本文作者)