利用GO语言实现多人聊天室实例教程
作者:shangguan_1234
聊天室的实现大家应该都遇到过,这篇文章主要给大家介绍了关于利用GO语言实现多人聊天室的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面来一起看看吧。
前言
运用go里面的net包中的相关方法来实现一个基于tcp的简单多人聊天室,用一个服务器来管理,主要反馈客户端是否连接成功并显示客户端输入的内容,并且发送给每一个在服务器上连接的客服端,下面话不多说了,来一起看看详细的介绍吧。
示例代码
服务器代码
// server package main import ( "fmt" "net" ) var ConnMap map[string]*net.TCPConn func checkErr(err error) int { if err != nil { if err.Error() == "EOF" { fmt.Println("用户退出") return 0 } fmt.Println("发生错误") return -1 } return 1 } func say(tcpConn *net.TCPConn) { for { data := make([]byte, 256) total, err := tcpConn.Read(data) if err != nil { fmt.Println(string(data[:total]), err) } else { fmt.Println(string(data[:total])) } flag := checkErr(err) if flag == 0 { break } for _, conn := range ConnMap { if conn.RemoteAddr().String() == tcpConn.RemoteAddr().String() { continue } conn.Write(data[:total]) } } } func main() { //var conn net.TCPConn //localAddr :=conn.LocalAddr().String() //fmt.Println(localAddr) //tcpAddr, _ := net.ResolveTCPAddr("tcp",localAddr) tcpAddr, _ := net.ResolveTCPAddr("tcp", "192.168.128.216:8080") tcpListen, _ := net.ListenTCP("tcp", tcpAddr) ConnMap = make(map[string]*net.TCPConn) for { tcpConn, _ := tcpListen.AcceptTCP() defer tcpConn.Close() ConnMap[tcpConn.RemoteAddr().String()] = tcpConn fmt.Println("连接客户端信息:", tcpConn.RemoteAddr().String()) go say(tcpConn) } }
本来打算用系统的LocalAddr()
函数来直接调用本地网络地址,但是这里调用一直报格式的错误,所以这里用本地地址直接输在里面,端口是8080//192.168.247.128:8080,也可以改成和客户端相似的代码来进行调用,但鉴于服务器唯一,所以就不做更改了
客户端代码
// client package main import ( "fmt" "net" "os" ) var ch chan int = make(chan int) var nickname string func reader(conn *net.TCPConn) { buff := make([]byte, 256) for { j, err := conn.Read(buff) if err != nil { ch <- 1 break } fmt.Printf("%s\n", buff[0:j]) } } func main() { if len(os.Args) != 2 { fmt.Fprintf(os.Stderr, "Usage:%s host:port", os.Args[0]) os.Exit(1) } service := os.Args[1] TcpAdd, _ := net.ResolveTCPAddr("tcp", service) //TcpAdd, _ := net.ResolveTCPAddr("tcp", "localhost:8080") conn, err := net.DialTCP("tcp", nil, TcpAdd) if err != nil { fmt.Println("服务没打开") os.Exit(1) } defer conn.Close() go reader(conn) fmt.Println("请输入昵称") fmt.Scanln(&nickname) fmt.Println("你的昵称是:", nickname) //var str string for { var msg string fmt.Scan(&msg) fmt.Print("<" + nickname + ">" + "说:") //for i, _ := range msg { // fmt.Printf("%c", msg[i]) //} fmt.Println(msg) b := []byte("<" + nickname + ">" + "说:" + msg) conn.Write(b) select { case <-ch: fmt.Println("server发生错误,请重新连接") os.Exit(2) default: } } }
客户端主要就是向服务器发起连接,然后发送和接收数据,原计划定义一个字符切片来存储读来的数据,但是出现了调用fmt.Println()来输出是时候显示ASCII码的错误,所以改成直接往一个空字符串里面读写,然后再来调用fmt.Println()
,显示出正确的字符串
运行服务器
go build server.go
./server
或 go run server.go
运行客户端
go build client.go
./client 服务器地址:端口(上面例子是192.168.247.128:8080)
总结
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对脚本之家的支持。