Golang WebSocket 创建单独会话_golang 实现聊天会话，经典好文

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

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

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

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

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

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

正文

}

defer conn.Close()

// 处理 WebSocket 会话
for {
// 读取客户端发送的消息
messageType, message, err := conn.ReadMessage()
if err != nil {
log.Println(“Read message failed:”, err)
break
}

// 处理客户端发送的消息
log.Printf(“Received message: %s”, message)

// 回复客户端消息
err = conn.WriteMessage(messageType, message)
if err != nil {
log.Println(“Write message failed:”, err)
break
}
}
}

在上面的示例中，我们首先创建了一个 upgrader 对象，它用于将 HTTP 连接升级为 WebSocket 连接。然后，我们定义了一个 handleWebSocket 函数，用于处理 WebSocket 会话。在该函数中，我们首先将 HTTP 连接升级为 WebSocket 连接，然后进入一个无限循环，不断读取客户端发送的消息，并给客户端回复相同的消息。

最后，我们使用 http.HandleFunc 函数将 /ws 路径映射到 handleWebSocket 函数，然后调用 http.ListenAndServe 函数来启动 WebSocket 服务器。

创建 WebSocket 客户端

接下来，我们需要创建一个 WebSocket 客户端，用于向服务器发送 WebSocket 请求，并处理服务器推送的消息。以下是一个简单的 WebSocket 客户端示例：

package main

import (
“log”
“os”
“os/signal”
“time”

“github.com/gorilla/websocket”
)

func main() {
// 创建一个 dialer 对象，用于建立 WebSocket 连接
dialer := websocket.Dialer{
HandshakeTimeout: 10 * time.Second,
}

// 建立 WebSocket 连接
conn, _, err := dialer.Dial(“ws://localhost:8080/ws”, nil)
if err != nil {
log.Fatal(“Dial failed:”, err)
}
defer conn.Close()

// 启动一个 goroutine 来接收服务器推送的消息
go func() {
for {
_, message, err := conn.ReadMessage()
if err != nil {
log.Println(“Read message failed:”, err)
break
}

log.Printf(“Received message: %s”, message)
}
}()

// 向服务器发送消息
message := []byte(“Hello, WebSocket!”)
err = conn.WriteMessage(websocket.TextMessage, message)
if err != nil {
log.Println(“Write message failed:”, err)
return
}

// 等待用户按下 Ctrl+C 终止程序
interrupt := make(chan os.Signal, 1)
signal.Notify(interrupt, os.Interrupt)
<-interrupt
}

在上面的示例中，我们首先创建了一个 dialer 对象，它用于建立 WebSocket 连接。然后，我们使用 dialer.Dial 函数建立 WebSocket 连接，并指定服务器的地址为 ws://localhost:8080/ws。然后，我们使用 conn.WriteMessage 函数向服务器发送消息，并使用一个 goroutine 来接收服务器推送的消息。

最后，我们使用 signal.Notify 函数来注册一个信号监听器，当用户按下 Ctrl+C 时，程序会接收到一个中断信号，然后程序退出。

创建单独会话

在实际应用中，我们可能需要为每个客户端创建一个单独的会话，以便管理和跟踪客户端的状态。在 Golang 中，可以通过为每个 WebSocket 连接创建一个独立的 goroutine 来实现这一点。以下是一个示例：

package main

import (
“log”
“net/http”

“github.com/gorilla/websocket”
)

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

func main() {
http.HandleFunc(“/ws”, handleWebSocket)
log.Fatal(http.ListenAndServe(“:8080”, nil))
}

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

defer conn.Close()

// 创建一个独立的会话
session := NewSession()

// 处理会话
session.Handle(conn)
}

type Session struct {
conn *websocket.Conn
}

func NewSession() *Session {
return &Session{}
}

func (s *Session) Handle(conn *websocket.Conn) {
s.conn = conn

go s.readLoop()
go s.writeLoop()
}

func (s *Session) readLoop() {
for {
messageType, message, err := s.conn.ReadMessage()
if err != nil {
log.Println(“Read message failed:”, err)
break
}

log.Printf(“Received message: %s”, message)
}
}

func (s *Session) writeLoop() {
for {
select {
// 从消息队列中获取消息并发送给客户端
case message := <-s.messageQueue:
err := s.conn.WriteMessage(websocket.TextMessage, message)
if err != nil {

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

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

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

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

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