使用 GoFrame 实现 WebSocket 通信和心跳机制：实践指南-Golang-PHP中文网

使用 GoFrame 实现 WebSocket 通信和心跳机制：实践指南

DDD

发布： 2024-12-19 14:32:09

原创

273 人浏览过

Implementing WebSocket Communication and Heartbeat Mechanism with GoFrame: A Hands-on Guide

在现代 Web 开发中，实时通信变得越来越重要。 WebSocket 是实现客户端和服务器之间双向通信的首选技术。本指南将引导您使用 GoFrame 实现 WebSocket 通信和强大的心跳机制。

你将学到什么

使用 GoFrame 设置 WebSocket 服务器
实现客户端WebSocket通信
处理并发 WebSocket 连接
构建可靠的心跳机制
生产就绪的 WebSocket 应用程序的最佳实践

先决条件

Go 编程基础知识
GoFrame 框架已安装
了解 WebSocket 协议基础知识

设置 WebSocket 服务器

让我们从创建一个基本的 WebSocket 服务器开始：

package main

import (
    "github.com/gogf/gf/v2/frame/g"
    "github.com/gogf/gf/v2/net/ghttp"
    "github.com/gogf/gf/v2/os/gctx"
)

func main() {
    ctx := gctx.New()
    s := g.Server()
    s.BindHandler("/ws", func(r *ghttp.Request) {
       ws, err := r.WebSocket()
       if err != nil {
          g.Log().Error(ctx, err)
          return
       }
       defer ws.Close()

       for {
          msgType, msg, err := ws.ReadMessage()
          if err != nil {
             return
          }
          if err = ws.WriteMessage(msgType, msg); err != nil {
             return
          }
       }
    })
    s.SetPort(8399)
    s.Run()
}

登录后复制

这将创建一个简单的回显服务器，该服务器侦听端口 8399 并回显它收到的任何消息。

客户端实施

这是一个基本的 HTML/JavaScript 客户端实现：

<!DOCTYPE html>
<html>
<head>
    <title>WebSocket Client</title>
</head>
<body>
    <script>
        const socket = new WebSocket('ws://localhost:8399/ws');

        socket.onopen = function(e) {
            console.log('Connection established');
            socket.send('Hello, server!');
        };

        socket.onmessage = function(event) {
            console.log('Message received:', event.data);
        };

        socket.onclose = function(event) {
            console.log('Connection closed');
        };
    </script>
</body>
</html>

登录后复制

处理并发连接

在生产环境中，您需要有效地处理多个连接。以下是如何实现连接池：

import "github.com/gogf/gf/v2/os/gmlock"

var (
    connPool = make(map[string]*ghttp.WebSocket)
    mu       = gmlock.New()
)

func addConn(id string, ws *ghttp.WebSocket) {
    mu.Lock()
    connPool[id] = ws
    mu.Unlock()
}

func removeConn(id string) {
    mu.Lock()
    delete(connPool, id)
    mu.Unlock()
}

func broadcastMessage(ctx context.Context, id string, message []byte) {
    mu.RLock(id)
    defer mu.RUnlock(id)

    for _, ws := range connPool {
       go func(ws *ghttp.WebSocket) {
          if err := ws.WriteMessage(websocket.TextMessage, message); err != nil {
             g.Log().Error(ctx, err)
          }
       }(ws)
    }
}

登录后复制

实现心跳机制

这是一个生产就绪的心跳实现：

package main

import (
    "github.com/gogf/gf/v2/frame/g"
    "github.com/gogf/gf/v2/net/ghttp"
    "github.com/gogf/gf/v2/os/gctx"
)

func main() {
    ctx := gctx.New()
    s := g.Server()
    s.BindHandler("/ws", func(r *ghttp.Request) {
       ws, err := r.WebSocket()
       if err != nil {
          g.Log().Error(ctx, err)
          return
       }
       defer ws.Close()

       for {
          msgType, msg, err := ws.ReadMessage()
          if err != nil {
             return
          }
          if err = ws.WriteMessage(msgType, msg); err != nil {
             return
          }
       }
    })
    s.SetPort(8399)
    s.Run()
}

登录后复制

客户端心跳处理

<!DOCTYPE html>
<html>
<head>
    <title>WebSocket Client</title>
</head>
<body>
    <script>
        const socket = new WebSocket('ws://localhost:8399/ws');

        socket.onopen = function(e) {
            console.log('Connection established');
            socket.send('Hello, server!');
        };

        socket.onmessage = function(event) {
            console.log('Message received:', event.data);
        };

        socket.onclose = function(event) {
            console.log('Connection closed');
        };
    </script>
</body>
</html>

登录后复制