Golang聊天应用开发实战

最近在学Golang想开发一个聊天应用，请问有什么推荐的框架或库吗？需要实现基本的消息收发和用户在线状态管理功能。另外在WebSocket连接管理和性能优化方面有什么需要注意的地方？希望有实际项目经验的朋友能分享一些实战技巧。

2 回复

开发Golang聊天应用，推荐使用gorilla/websocket库。核心步骤：

  1. 建立WebSocket连接：
upgrader := websocket.Upgrader{}
conn, _ := upgrader.Upgrade(w, r, nil)
  1. 消息处理循环：
for {
    messageType, p, err := conn.ReadMessage()
    // 处理消息
    conn.WriteMessage(messageType, p)
}
  1. 客户端连接管理： 使用map存储连接，用互斥锁保证并发安全：
var clients = make(map[*websocket.Conn]bool)
var mutex sync.Mutex

  1. 广播消息： 遍历所有客户端连接发送消息

  2. 前端配合： 使用JavaScript WebSocket API建立连接和收发消息

可以扩展功能：用户认证、消息持久化、房间分组等。部署时注意WebSocket反向代理配置。

更多关于Golang聊天应用开发实战的实战系列教程也可以访问 https://www.itying.com/category-94-b0.html


Golang聊天应用开发实战

核心架构设计

1. WebSocket服务器

package main

import (
    "log"
    "net/http"
    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool { return true },
}

type Client struct {
    conn *websocket.Conn
    send chan []byte
}

type Message struct {
    Username string `json:"username"`
    Content  string `json:"content"`
    Time     string `json:"time"`
}

var clients = make(map[*Client]bool)
var broadcast = make(chan Message)

func handleConnections(w http.ResponseWriter, r *http.Request) {
    ws, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Fatal(err)
    }
    defer ws.Close()

    client := &Client{conn: ws, send: make(chan []byte, 256)}
    clients[client] = true

    for {
        var msg Message
        err := ws.ReadJSON(&msg)
        if err != nil {
            delete(clients, client)
            break
        }
        broadcast <- msg
    }
}

2. 消息广播处理

func handleMessages() {
    for {
        msg := <-broadcast
        for client := range clients {
            err := client.conn.WriteJSON(msg)
            if err != nil {
                client.conn.Close()
                delete(clients, client)
            }
        }
    }
}

func main() {
    http.HandleFunc("/ws", handleConnections)
    go handleMessages()
    
    log.Println("服务器启动在 :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

3. 前端客户端示例

<!DOCTYPE html>
<html>
<body>
    <div id="messages"></div>
    <input type="text" id="messageInput">
    <button onclick="sendMessage()">发送</button>

    <script>
        const ws = new WebSocket("ws://localhost:8080/ws");
        ws.onmessage = function(event) {
            const msg = JSON.parse(event.data);
            document.getElementById('messages').innerHTML += 
                `<div><strong>${msg.username}:</strong> ${msg.content}</div>`;
        };
        
        function sendMessage() {
            const input = document.getElementById('messageInput');
            ws.send(JSON.stringify({
                username: "用户",
                content: input.value,
                time: new Date().toISOString()
            }));
            input.value = "";
        }
    </script>
</body>
</html>

功能扩展建议

  1. 用户认证：添加JWT token验证
  2. 房间功能：支持多个聊天房间
  3. 消息持久化：集成Redis或数据库存储历史消息
  4. 文件传输：支持图片和文件分享
  5. 在线状态：显示用户在线/离线状态

部署运行

go mod init chat-app
go get github.com/gorilla/websocket
go run main.go

这个基础版本包含了实时聊天的核心功能，可以根据需求进一步扩展完善。

回到顶部