服务器之家

服务器之家 > 正文

一百行Golang代码实现简单并发聊天室

时间:2020-05-27 10:22     来源/作者:Go-Bluer

项目介绍:Golang100行代码实现高并发聊天室,其中实现的功能有:上下线广播,私聊,用户改名,超时强踢,在线用户检测等

在开始项目前,我们需要理解贯穿这整个项目的两个重要变量,若能理解这两个变量的使用,那么并发聊天室项目会变得手到擒来。第一个是onlinemap全局map,第二个是Message全局channel。

取名为onlinemap的全局map类型为map[string][client],这个全局字典是用来存储当前在此聊天室的用户的,key值是string类型,为用户的ip地址+Port端口,对应的value值为一个结构体,结构体内有此用户的姓名,地址和管道(用来给每一个用户传输信息,服务于Message全局通道)

取名为Message的全局channel也贯穿在整段代码中,向其中传送数据时,Message会在另一个go程里向其他每一个在线用户的管道中发送内容,随后在另一个go程里每一个用户的管道会向对应用户转发内容。如此可以实现上下线广播,群聊的功能。而每一个用户私有的管道可以实现私聊功能。

这个图详细阐述了这段代码的工作流程。

一百行Golang代码实现简单并发聊天室

理解以上内容 下面我们再来看代码,就会很轻松,如果还是一头雾水也不要着急,小编会在下面每一行代码都加上精准通俗的注释,不多说,上代码。

?
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
package main
import (
  "net"
  "fmt"
  "strings"
  "time"
)
//定义的此结构体为全局map的value值,包括每一个用户的姓名,ip地址和私人管道
type client struct {
  name string
  addr string
  C  chan string
}
/*这个函数是将私人管道中的内容发送给用户,配合全局管道Message使用可以实现广播的功能,
单独使用可以实现私聊的功能*/
func writemsg2client(clinet client,conn net.Conn) {
  for m := range clinet.C {
    conn.Write([]byte(m + "\n"))
  }
}
//这只是一个封装好用来统一(发送信息格式)的小函数,不用在意
func makemsg(name string, addr string, s string) string {
  return "[" + addr + "]" + name + s
}
//每一个进入聊天室的用户都将启动一个handleconn的go程来处理事件
func handleconn(conn net.Conn) {
  defer conn.Close()
/*用户连接进来以后要初始化全局map,把自己的信息加入到字典里,相当于进到聊天室里之前要登
记一下个人信息,注意姓名初始为ip地址。*/
  addr := conn.RemoteAddr().String()
  fmt.Printf("用户%s进入了房间\n", addr)
  client := client{addr, addr, make(chan string)}
  //在这里启动子go程,功能上面已经提及
  go writemsg2client(client,conn)
  onlinemap[addr] = client
  //登录进来一切准备就绪后就给所有人广播上线信息啦
  Message <- makemsg(client.name, addr, "login")
  //下面这三个变量服务于下面一些小功能
  var haschat=make(chan bool)
  var ifquit=make(chan bool)
  var flag bool
  //从这单独开启一个go程来读取用户输入的信息
  go func() {
    buf:=make([]byte,4096)
    for {
      n,_:=conn.Read(buf)
      if n==0 {
        fmt.Printf("%s离开了房间\n",client.name)
        ifquit<-true
        return
      }
      //改名功能的实现
      if string(buf[:7])=="Rename|" {
        client.name=strings.Split(string(buf[:n-1]),"|")[1]
        onlinemap[addr]=client
        conn.Write([]byte("rename success\n"))
      }else if string(buf[:n-1])=="/who"{
      //查询在线用户信息的功能
        for _,s:=range onlinemap{
          conn.Write([]byte(s.name+"online\n"))
        }
      }else if string(buf[:2])=="m|"&&strings.Count(string(buf[:n]),"|")==2 {
/*私聊功能的实现,其实私聊功能就是跳过了往全局Message里传输信息,
改为直接向私人管道里传输信息*/
        flag=false
        slice:=strings.Split(string(buf[:n-1]),"|")
        for _,a:=range onlinemap{
        //遍历所有在线用户,向指定的用户管道中发送信息
          if a.name==slice[1]{
            flag=true
            a.C<-makemsg(client.name,addr,slice[2])
            conn.Write([]byte("send success"))
          }
        }
        if flag {
          conn.Write([]byte("no such man or not online"))
        }
      } else {
        Message<-makemsg(client.name,addr,string(buf[:n-1]))
      }
      haschat<-true
    }
  }()
  for {
    select {
    case <-haschat:
    //超时强踢
    case <-time.After(time.Minute*3):
      delete(onlinemap,addr)
      Message<-makemsg(client.name,addr,"out time to leave")
      close(client.C)
      return
    case <-ifquit:
    //退出处理
      delete(onlinemap,addr)
      Message<-makemsg(client.name,addr,"out time to leave")
      close(client.C)
      return
    }
  }
}
//这个函数用来将全局Message中的内容全部塞到私人管道C里,实现上下线广播和群聊的功能
func Manager() {
  for {
    msg := <-Message
    for _, s := range onlinemap {
      s.C <- msg
    }
  }
}
var Message = make(chan string)
var onlinemap map[string]client = make(map[string]client)
//主函数
func main() {
  listener, _ := net.Listen("tcp", "127.0.0.1:6666")
  defer listener.Close()
  //提前开启全局Message的go程,防止被阻塞
  go Manager()
  for {
    conn, err := listener.Accept()
    if err != nil {
      fmt.Println("accept err", err)
      continue
    }
    //每一个连接进来的用户都会被分配进入一个子go程,用来处理上面我们提到的各种功能
    go handleconn(conn)
  }
}

以上就是一个简单的高并发聊天室了,依托于go语言的强大,去掉注释只剩下不到一百行,虽然功能简单,但是涉及到channel,socket,select,map,string及go的使用,有利于此阶段在学的小伙伴们学习交流,大家有什么疑问或者想法可以在下面给我留言哦。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。

原文链接:https://blog.csdn.net/weixin_42940826/article/details/82386275

标签:

相关文章

热门资讯

歪歪漫画vip账号共享2020_yy漫画免费账号密码共享
歪歪漫画vip账号共享2020_yy漫画免费账号密码共享 2020-04-07
沙雕群名称大全2019精选 今年最火的微信群名沙雕有创意
沙雕群名称大全2019精选 今年最火的微信群名沙雕有创意 2019-07-07
玄元剑仙肉身有什么用 玄元剑仙肉身境界等级划分
玄元剑仙肉身有什么用 玄元剑仙肉身境界等级划分 2019-06-21
男生常说24816是什么意思?女生说13579是什么意思?
男生常说24816是什么意思?女生说13579是什么意思? 2019-09-17
超A是什么意思 你好a表达的是什么
超A是什么意思 你好a表达的是什么 2019-06-06
返回顶部