【Golang】websocket连接存储多种方式及优缺点

gskyi

已于 2024-12-03 09:26:43 修改

阅读量359

点赞数 3

分类专栏： golang websocket 文章标签： golang websocket 开发语言

于 2024-12-03 09:24:16 首次发布

本文链接：https://blog.youkuaiyun.com/a731637163/article/details/144203838

版权

golang 同时被 2 个专栏收录

11 篇文章

订阅专栏

websocket

1 篇文章

订阅专栏

websocket连接存储可以有多种方式，以下是一些常见的替代方案及其优缺点：

1. 使用切片（Slice）

可以使用切片来存储所有在线用户的连接，而不是使用映射（map）。这种方法在需要遍历所有连接时比较简单。

var connections []*WebSocketConnection

// 添加连接
connections = append(connections, websocketConn)

// 删除连接
for i, conn := range connections {
    if conn.ID == websocketConn.ID {
        connections = append(connections[:i], connections[i+1:]...) // 删除特定连接
        break
    }
}

优点：

简单易用，适合小规模连接。

缺点：

查找和删除连接的效率较低，时间复杂度为 O(n)。

2. 使用链表（Linked List）

使用链表数据结构，可以方便地在连接中添加和删除元素。

type Node struct {
    Conn *WebSocketConnection
    Next *Node
}

var head *Node

// 添加连接
func addConnection(conn *WebSocketConnection) {
    newNode := &Node{Conn: conn, Next: head}
    head = newNode
}

// 删除连接
func removeConnection(connID string) {
    var prev *Node
    current := head
    for current != nil {
        if current.Conn.ID == connID {
            if prev != nil {
                prev.Next = current.Next
            } else {
                head = current.Next
            }
            break
        }
        prev = current
        current = current.Next
    }
}

优点：

动态大小，添加和删除操作效率较高。

缺点：

实现复杂度较高，可能会增加额外的内存开销。

3. 使用 Redis 或其他外部存储

将连接信息存储在 Redis 或其他外部存储中，可以实现跨进程或跨服务器的连接管理。

import "github.com/go-redis/redis/v8"

var rdb = redis.NewClient(&redis.Options{
    Addr: "localhost:6379",
})

// 存储连接
func storeConnection(userID string, conn *WebSocketConnection) {
    rdb.Set(ctx, userID, conn.ID, 0) // 使用 Redis 存储连接信息
}

// 删除连接
func deleteConnection(userID string) {
    rdb.Del(ctx, userID) // 使用 Redis 删除连接信息
}

优点：

支持分布式应用，连接信息持久化。

缺点：

需要额外的 Redis 服务，增加了系统复杂度。

4. 使用数据库

将连接信息存储在关系型数据库（如 MySQL、PostgreSQL）或 NoSQL 数据库中。

CREATE TABLE connections (
    id VARCHAR(255) PRIMARY KEY,
    user_id VARCHAR(255),
    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
);

优点：

数据持久化，支持复杂查询。

缺点：

性能可能不如内存存储，增加了数据库的负担。

5. 使用（map）

// WebSocketConnection 表示一个 WebSocket 连接实例
type WebSocketConnection struct {
    ID     string          // 连接的唯一标识符
    UserID string          // 用户 ID
    Conn   *websocket.Conn // WebSocket 连接对象
}

var  connections = make(map[string]*WebSocketConnection) 

// 创建新的 WebSocket 连接实例
websocketConn := newWebSocketConnection(userID, conn)

// 存储用户连接
mu.Lock() // 加锁以保护共享资源
connections[websocketConn.ID] = websocketConn // 使用 ID 存储连接
mu.Unlock()

// 确保连接在退出时关闭
defer func() {
    if err := conn.Close(); err != nil {
        log.Println("Error while closing connection:", err)
    }
    mu.Lock() // 加锁以保护共享资源
    delete(connections, websocketConn.ID) // 从连接映射中删除
    mu.Unlock()
}()