利用GO语言实现多人聊天室实例教程

时间:2022-02-22 04:30:04

前言

运用go里面的net包中的相关方法来实现一个基于tcp的简单多人聊天室,用一个服务器来管理,主要反馈客户端是否连接成功并显示客户端输入的内容,并且发送给每一个在服务器上连接的客服端,下面话不多说了,来一起看看详细的介绍吧。

示例代码

服务器代码

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
// 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,也可以改成和客户端相似的代码来进行调用,但鉴于服务器唯一,所以就不做更改了

客户端代码

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
// 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语言实现多人聊天室实例教程

总结

以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对服务器之家的支持。

原文链接:https://blog.csdn.net/shangguan_1234/article/details/79640856