Golang WebSocket 心跳，从消息中间件看分布式系统的多种套路

先自我介绍一下，小编浙江大学毕业，去过华为、字节跳动等大厂，目前阿里P7

深知大多数程序员，想要提升技能，往往是自己摸索成长，但自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年最新Golang全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上Go语言开发知识点，真正体系化！

由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新

如果你需要这些资料，可以添加V获取：vip1024b （备注go）

正文

WebSocket是一种在客户端和服务器之间实现全双工通信的协议，它允许实时地传输数据，并且比传统的HTTP请求更加高效。在使用Golang构建WebSocket应用程序时，一个重要的考虑因素是如何实现心跳机制，以确保连接的稳定性和可靠性。本文将探讨如何使用Golang实现WebSocket心跳。

什么是心跳？

心跳是指定期发送的小型数据包，用于检测网络连接的可用性。在WebSocket中，心跳机制用于检测和维持连接的活跃状态。通过定期发送心跳数据包，服务器可以知道客户端是否在线，从而采取相应的措施，比如重新连接或关闭连接。

心跳的重要性

在WebSocket应用程序中，心跳机制具有以下重要性：

1. 连接稳定性：WebSocket连接可能因为网络问题或其他原因而断开。通过定期发送心跳包，可以及时检测到连接的断开，并采取相应的措施，如重新连接或关闭连接。
2. 资源释放：未使用的WebSocket连接可能会占用服务器资源。通过实现心跳机制，服务器可以检测到不活跃的连接，并释放资源，以提高性能和可扩展性。
3. 客户端状态：通过心跳机制，服务器可以了解客户端的在线状态，从而采取适当的操作。例如，当客户端长时间未发送心跳包时，可以将其标记为离线状态。

如何实现WebSocket心跳

在Golang中，可以使用goroutine和定时器来实现WebSocket的心跳机制。以下是实现WebSocket心跳的步骤：

第一步：创建WebSocket连接

首先，我们需要创建一个WebSocket连接。Golang提供了一个内置的websocket包，可以方便地创建和管理WebSocket连接。以下是一个简单的示例代码，用于创建一个WebSocket连接：

package main

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

var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}

func main() {
http.HandleFunc(“/ws”, func(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println(err)
return
}
})

log.Fatal(http.ListenAndServe(“:8080”, nil))
}

第二步：添加心跳处理程序

接下来，我们需要添加一个心跳处理程序，用于定期发送心跳包。我们可以使用goroutine和定时器来实现这个功能。以下是一个示例代码，用于添加心跳处理程序：

package main

import (
“log”
“net/http”
“time”

“github.com/gorilla/websocket”
)

var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}

func main() {
http.HandleFunc(“/ws”, func(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println(err)
return
}

go heartbeat(conn)

// 处理其他操作
})

log.Fatal(http.ListenAndServe(“:8080”, nil))
}

func heartbeat(conn *websocket.Conn) {
ticker := time.NewTicker(5 * time.Second)
defer ticker.Stop()

for {
select {
case <-ticker.C:
err := conn.WriteMessage(websocket.PingMessage, []byte(“heartbeat”))
if err != nil {
log.Println(err)
return
}
}
}
}

在上面的代码中，我们使用time.NewTicker函数创建了一个定时器，每隔5秒触发一次。然后，我们使用conn.WriteMessage函数发送一个WebSocket ping消息作为心跳包。

第三步：处理心跳响应

最后，我们需要处理来自客户端的心跳响应。如果客户端未在规定的时间内响应心跳包，我们可以将其标记为离线状态。以下是一个示例代码，用于处理心跳响应：

package main

import (
“log”
“net/http”
“time”

“github.com/gorilla/websocket”
)

var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}

func main() {
http.HandleFunc(“/ws”, func(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println(err)
return
}

go heartbeat(conn)

// 处理其他操作
})

log.Fatal(http.ListenAndServe(“:8080”, nil))
}

func heartbeat(conn *websocket.Conn) {
ticker := time.NewTicker(5 * time.Second)
defer ticker.Stop()

for {
select {
case <-ticker.C:
err := conn.WriteMessage(websocket.PingMessage, []byte(“heartbeat”))
if err != nil {
log.Println(err)
return
}

conn.SetReadDeadline(time.Now().Add(10 * time.Second))
_, _, err = conn.ReadMessage()
if err != nil {
log.Println(“heartbeat response error:”, err)
return
}
}
}
}

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化的资料的朋友，可以添加V获取：vip1024b （备注Go）

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！
不再深入研究，那么很难做到真正的技术提升。**

需要这份系统化的资料的朋友，可以添加V获取：vip1024b （备注Go）
[外链图片转存中…(img-3BVo4lUH-1713130341875)]

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！