使用Go实现一个百行聊天服务器的示例代码
作者:烟草的香味.
前段时间, redis作者整了个c语言版本的聊天服务器,代码量拢共不过百行,于是, 心血来潮下, 我也整了个Go语言版本, 简单来说就是实现了一个聊天室的功能,文中通过代码示例给大家介绍的非常详细,需要的朋友可以参考下
前段时间, redis
作者不是整了个c
语言版本的聊天服务器嘛, 地址, 代码量拢共不过百行.
于是, 心血来潮下, 我也整了个Go
语言版本. 简单来说就是实现了一个聊天室的功能. 将所有注释空行都去掉, 刚好100行实现. 废话不多说, 先上代码:
package main import ( "fmt" "log" "net" "strings" "sync" ) type Server struct { // 服务端内容 clients map[string]*Client lock sync.Mutex } func (s *Server) delClient(client *Client) { // 客户端关闭 s.lock.Lock() defer s.lock.Unlock() delete(s.clients, client.name) } func (s *Server) addClient(client *Client) { // 客户端关闭 s.lock.Lock() defer s.lock.Unlock() s.clients[client.name] = client } func (s *Server) sendMsgToOtherClient(msg string, client *Client) { // 将消息发送给其他所有客户端 s.lock.Lock() defer s.lock.Unlock() // 将消息转发给其他客户端 for _, c := range s.clients { if c != client { c.msgCh <- "msg-> " + client.name + ": " + msg + "\n" } } } type Client struct { // 定义客户端 conn net.Conn name string // 当前客户端的名称 msgCh chan string // 发送消息的管道 server *Server } func (c *Client) receive() { // 接收消息 for msg := range c.msgCh { _, _ = c.conn.Write([]byte(msg)) } } func (c *Client) close() { c.server.delClient(c) close(c.msgCh) _ = c.conn.Close() } func (c *Client) handle() { // 开始处理连接 _, _ = c.conn.Write([]byte(fmt.Sprintf("hello %s!\n", c.name))) // 发送欢迎信息 c.server.sendMsgToOtherClient("join", c) // 通知大家, 有人加入了聊天室 defer c.close() for { buf := make([]byte, 2048) n, err := c.conn.Read(buf) // 接收客户端发送的消息 if err != nil { log.Printf("receive client data error: %s", err.Error()) return } msg := strings.TrimSpace(string(buf[:n])) if len(msg) == 0 { continue } if msg == "quit" { c.server.sendMsgToOtherClient("quit", c) // 通知大家, 有人退出了聊天室 return } c.server.sendMsgToOtherClient(msg, c) } } func main() { // 监听端口 listener, err := net.Listen("tcp", ":8080") if err != nil { panic(err) } server := &Server{ clients: make(map[string]*Client), lock: sync.Mutex{}, } nameIndex := 1 for { conn, err := listener.Accept() // 建立连接 if err != nil { log.Println(err) continue } client := &Client{ conn: conn, msgCh: make(chan string, 100), name: fmt.Sprintf("user%d", nameIndex), server: server, } nameIndex++ server.addClient(client) go client.handle() go client.receive() log.Printf("new client: %s\n", conn.RemoteAddr()) } }
可以直接telnet
作为客户端连接, 实现的功能简单来说就是一个大的聊天室, 用户发的消息会同步发给所有用户. (因代码篇幅和复杂度原因, 就不对代码做详细说明了)
看下聊天室效果:
当然了, 还是有很多极限情况没有处理, 很多异常情况没有判断. 比如:
- 客户端数量没有控制
- 缺少客户端心跳
- 缺少错误处理
- 缺少安全性检查
- 客户端发送消息的长度限制
- 等等…
但总得来说, 作为一个玩具还是可以的, 而且也达到百行的要求咯.(这也是我改了几版才改到百行的…)
以上就是使用Go实现一个百行聊天服务器的详细内容,更多关于Go实现聊天服务器的资料请关注脚本之家其它相关文章!