使用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作为客户端连接, 实现的功能简单来说就是一个大的聊天室, 用户发的消息会同步发给所有用户. (因代码篇幅和复杂度原因, 就不对代码做详细说明了)

看下聊天室效果:

当然了, 还是有很多极限情况没有处理, 很多异常情况没有判断. 比如:

• 客户端数量没有控制
• 缺少客户端心跳
• 缺少错误处理
• 缺少安全性检查
• 客户端发送消息的长度限制
• 等等…

但总得来说, 作为一个玩具还是可以的, 而且也达到百行的要求咯.(这也是我改了几版才改到百行的..)