利用GO语言实现多人聊天室实例教程
(编辑：jimmy 日期: 2024/11/24 浏览：3 次 )
前言
运用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)
总结
以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，如果有疑问大家可以留言交流，谢谢大家对的支持。
下一篇：如何编写Go语言中间件的实例教程