Golang实现WebSocket服务的项目实践
作者：梦想画家 时间：2024-04-28 09:15:51
我们每天接触到各类应用，如社交、在线文档、直播等，后端都需要使用WebSocket技术提供实时通信能力。本文介绍如何使用Golang实现实时后端WebSocket服务，首先使用Gin框架搭建http服务，然后使用gorilla/websocket库实现简单后端WebSocket服务，示例实现从0到1的过程，适合初学者快速入门。
Gin实现Web应用
创建Go工程，新增hello.go文件，编写下面代码：
package main
import "fmt"
func main() {
fmt.Println("Hello, World!")
}
运行可以看到输出结果。下面增加Web服务功能。安装Gin库依赖：
go get -u github.com/gin-gonic/gin
修改上面hello.go文件内容：
package main
import (
"fmt"
"github.com/gin-gonic/gin"
)
func main() {
fmt.Println("Hello, World!")
// 创建Gin引擎实例
r := gin.Default()
// 增加路由ping，简单返回pong
r.GET("/ping", func(c *gin.Context) {
c.JSON(200, gin.H{
"message": "pong",
})
})
// listen and serve on 0.0.0.0:8080
// 启动HTTP服务
r.Run()
}
运行程序，访问地址：http://localhost:8080/ping，返回结果：
{
"message": "pong"
}
实现WebSocket服务
创建server.go，修改hello.go文件的main方法为hello()，同一包下不能有两个main方法。需要安装相应的库：go get -u github.com/gorilla/websocket,增加下面内容:
package main
import (
"fmt"
"log"
"net/http"
"github.com/gin-gonic/gin"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
// 解决跨域问题
CheckOrigin: func(r *http.Request) bool {
return true
},
}
func ws(c *gin.Context) {
//更新get请求使用WebSocket协议
ws, err := upgrader.Upgrade(c.Writer, c.Request, nil)
if err != nil {
log. Print("upgrade:", err)
return
}
defer ws.Close()
// 监听 ws 消息
for {
// 从 ws 读取数据
mt, message, err := ws.ReadMessage()
if err != nil {
log.Println("read:", err)
break
}
log.Printf("recv: ％s", message)
//往 ws 写数据
err = ws.WriteMessage(mt, message)
if err != nil {
log.Println("write:", err)
break
}
}
}
func main() {
fmt.Println("Websocket Server!")
r := gin.Default()
r.GET("/ws", ws)
r.Run("localhost:8448")
}
核心逻辑为：收到消息后，往客户端写回相同内容。代码增加了注释，主要差异就是for循环中的读写消息。下面需要html文件进行交互，新建index.html文件：
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta http-equiv="X-UA-Compatible" content="IE=edge">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>Univer Server</title>
</head>
<body>
<script>
var ws = new WebSocket("ws://localhost:8448/ws");
//打开连接时触发
ws.onopen = function(evt) {
console.log("Connection open...");
ws.send("Hello WebSockets!");
};
//接收到消息时触发
ws.onmessage = function(evt) {
console.log("Received Message: " + evt.data);
};
//连接关闭时触发
ws.onclose = function(evt) {
console.log("Connection closed.");
};
</script>
Hello WebSocket.
</body>
</html>
script标签中定义了三个回调函数，连接ws后给服务端发送消息，服务端收到消息返回相同内容，最后时关闭连接时触发。
启动测试
首先启动ws，go run server.go, 启动服务端后等待客户端连接。下面启动客户端，我们可以使用vs code live server启动访问静态页面，也可以使用python启动http服务。在项目目录下打开命令行界面，输入下面命令，启用http服务：
python -m http.server 9000
通过浏览器访问 http://localhost:9000/index.html , 打开浏览器控制台可以看到信息：
Connection open...
index.html:21 Received Message: Hello WebSockets!
index.html:25 Connection closed.
首先看到第一行（连接成功打印的日志），然后是接收服务端发送的信息。最后关闭服务端程序，客户端打印连接断开日志。
来源：https://blog.csdn.net/neweastsun/article/details/130755719
0
投稿猜你喜欢
CSS实现垂直居中的5种方法
解决django后台管理界面添加中文内容乱码问题
找到个很好的例子导出excel的
Python读取xlsx数据生成图标代码实例
Python批量重命名同一文件夹下文件的方法
Python爬取视频(其实是一篇福利)过程解析
基于mysql replication的问题总结
Python实现多属性排序的方法
Python中使用支持向量机SVM实践
Python实现统计代码行的方法分析
Go 实战单队列到优先级队列实现图文示例
Python编程修改MP3文件名称的方法
Jupyter Notebook的连接密码 token查询方式
ThinkPHP发送邮件示例代码
mysql exists与not exists实例详解
Golang中字符串(string)与字节数组([]byte)一行代码互转实例
pandas进行时间数据的转换和计算时间差并提取年月日
