Go 实现 Websocket 服务以及代理

最新推荐文章于 2024-05-14 04:32:52 发布
最新推荐文章于 2024-05-14 04:32:52 发布
阅读量2.1k 收藏 24
点赞数 27
分类专栏： Golang 网络编程 文章标签： golang websocket 开发语言 网络协议
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/lfzs_/article/details/135964741
版权

Go 实现 Websocket服务以及代理

1. 协议说明

WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议。WebSocket 使得客户端和服务器之间的数据交换变得更加简单，允许服务端主动向客户端推送数据。Websocket 主要用在B/S架构的应用程序中，在 WebSocket API 中，浏览器和服务器只需要完成一次握手，两者之间就直接可以创建持久性的连接， 并进行双向数据传输。它的最大特点就是，服务器可以主动向客户端推送信息，客户端也可以主动向服务器发送信息，是真正的双向平等对话，属于服务器推送技术的一种。

WebSocket 协议在2008年诞生，2011 年成为国际标准。现在最新版本浏览器都已经支持了。

WebSocket 是一种应用层协议

WebSocket 的典型特点：

  1. 基于 TCP 协议的应用层协议，实现相对简单
  2. 单个 TCP 连接上进行全双工通信
  3. 兼容 HTTP 协议，默认端口也是 80 和 443

ws://host:port/path/query
wss://host:port/path/query

  1. 握手阶段采用 HTTP 协议，能通过各种 HTTP 代理服务器
  2. 数据格式比较轻量，性能开销小，通信高效
  3. 可以发送文本和二进制数据
  4. 没有浏览器的同源限制

websocket 的典型场景：

  • 即时通信

  • 协同编辑/编辑

  • 实时数据流的拉取与推送

2. WebSocket 推送和浏览器轮询

在 B/S 开发领域，若需要浏览器 B 即时得到服务器的状态更新，常使用两个方案：

  1. 浏览器端轮询

  2. 服务器端推送

浏览器轮询：浏览器端，当需要获取最新数据状态时，利用脚本程序循环向服务端发送请求。

服务器推送，服务器端，当状态改变时，将数据发送到浏览器端。
在这里插入图片描述

HTTP/2 版本也支持服务器端推送，但实现上以推送静态资源为主，不能基于业务逻辑推送特定的消息，因此当前的普及使用率 websocket 还是主流。

3. WebSocket 和 http

  • 相同点

  1. 应用层协议

  2. B/S 架构中使用

  3. 基于 TCP 协议

  4. 端口默认都是：80 和 443

  • 不同点

在这里插入图片描述

4. WebSocket 握手过程

通过 HTTP 请求响应，中的头信息，完成 websocket 握手，如图：
在这里插入图片描述

  • 在请求头中添加如下信息
# 升级为 websocket
Upgrade: websocket
Connection: Upgrade
# 一个 Base64 encode 的值，有于验证服务器端是否支持websocket
Sec-WebSocket-Key: x4JJHMbDL22zLk1GBhXDw==
# 用户协议，可以视为不同业务逻辑的频道
Sec-WebSocket-Protocol: chat
# 协议版本，13是当前通用版本，几乎不需要更改
Sec-WebSocket-Version: 13

基于以上请求头，服务器端，就知道需要将协议升级为 websocket 协议，并提供一些验证信息。

  • 服务端的响应头
HTTP/1.1 101 Switching Protocols
# 协议升级
Upgrade: websocket
# 连接状态
Connection: Upgrade
# WebSocket服务端根据Sec-WebSocket-Key生成
Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=
# WebSocket协议用户协议
Sec-WebSocket-Protocol: chat

基于以上响应头，浏览器端就知道服务器端升级成功，并通过了验证。
至此，B/S 端可以基于该连接，完成 websocket 双向通信了。

websocket 只能发送 GET 请求

5. WebSocket 状态码和消息类型

5.1 状态码

WebSocket协议状态码解析_websocket状态码-CSDN博客

5.2 消息类型

  • TextMessage 和 BinaryMessage 分别表示发送文本消息和二级制消息

  • CloseMessage 关闭帧，接收方收到这个消息就关闭连接

  • PingMessage 和 PongMessage : 是保持心跳的帧

    • 发送方 -> 接收方是 PingMessage
    • 接收方 -> 发送方是 PongMessage

由服务器发 ping 给浏览器，浏览器返回 pong 消息

6. WebSocket 服务器实现

使用 github.com/gorilla/websocket 这个库函数

func WebSocketServer() {
	addr := "localhost:8002"
	http.HandleFunc("/wshandler", WebSocketUpgrade)
	log.Println("Starting websocket server at " + addr)

	go func() {
		err := http.ListenAndServe(addr, nil)
		if err != nil {
			log.Fatal(err)
		}
	}()

	log.Println("WebSocket 服务器正在运行。按Ctrl+C退出")
	select {}
}

func WebSocketUpgrade(resp http.ResponseWriter, req *http.Request) {
	// 初始化 Upgrader
	upgrader := websocket.Upgrader{} // 使用默认的选项
	// 第三个参数是响应头,默认会初始化
	conn, err := upgrader.Upgrade(resp, req, nil)
	if err != nil {
		log.Println(err)
		return
	}
	defer conn.Close()

	// 读取客户端的发送额消息,并返回
	go ReadMessage(conn)
	select {}
}

// 读取客户端发送的消息,并返回
func ReadMessage(conn *websocket.Conn) {
	for {
		// 消息类型:文本消息和二进制消息
		messageType, msg, err := conn.ReadMessage()
		if err != nil {
			log.Println(err)
			return
		}
		fmt.Println("receive msg:", string(msg))

		err = conn.WriteMessage(messageType, msg)
		if err != nil {
			log.Println("write error:", err)
			return
		}
	}
}

使用 Apifox 测试 websocket 是否能连接并且发送消息

  • 消息发送成功，同时也接收到来服务端的消息
    在这里插入图片描述

  • 消息接收成功
    在这里插入图片描述

7. WebSocket 代理实现

package websocket

import (
	"log"
	"net/http"
	"net/http/httputil"
	"net/url"
)

var (
	// 代理服务器地址
	proxyServer     = "127.0.0.1:8082"
	// 真实websocket服务器地址
	websocketServer = "http://127.0.0.1:8002"
)

func WebSocketProxy() {
	url, err := url.Parse(websocketServer)
	if err != nil {
		log.Println(err)
	}
	proxy := httputil.NewSingleHostReverseProxy(url)
	log.Println("WebSocket 代理启动, 按CTRL+C退出")
	http.ListenAndServe(proxyServer, proxy)
}

在这里插入图片描述

8. WebSocket 服务端主动推送功能的实现

websocket 服务器每隔 3 秒会主动向服务器推送消息"Heart Beat"

func WebSocketServer() {
	addr := "localhost:8002"
	http.HandleFunc("/wshandler", WebSocketUpgrade)
	log.Println("Starting websocket server at " + addr)

	go func() {
		err := http.ListenAndServe(addr, nil)
		if err != nil {
			log.Fatal(err)
		}
	}()

	log.Println("WebSocket 服务器正在运行。按Ctrl+C退出")
	select {}
}

func WebSocketUpgrade(resp http.ResponseWriter, req *http.Request) {
	// 初始化 Upgrader
	upgrader := websocket.Upgrader{} // 使用默认的选项
	// 第三个参数是响应头,默认会初始化
	conn, err := upgrader.Upgrade(resp, req, nil)
	if err != nil {
		log.Println(err)
		return
	}
	defer conn.Close()

	// 主动向服务端推送消息
	go PushMessage(conn)

	// 读取客户端的发送额消息,并返回
	go ReadMessage(conn)
	select {}
}

// websocket 服务器主动服务器推送消息
func PushMessage(conn *websocket.Conn) {
	for {
		err := conn.WriteMessage(websocket.TextMessage, []byte("heart beat"))
		if err != nil {
			log.Println(err)
			return
		}
		time.Sleep(time.Second * 3)
	}
}

// 读取客户端发送的消息,并返回
func ReadMessage(conn *websocket.Conn) {
	for {
		// 消息类型:文本消息和二进制消息
		messageType, msg, err := conn.ReadMessage()
		if err != nil {
			log.Println(err)
			return
		}
		fmt.Println("receive msg:", string(msg))

		err = conn.WriteMessage(messageType, msg)
		if err != nil {
			log.Println("write error:", err)
			return
		}
	}
}

每隔三秒可以看到服务推送过来的消息在这里插入图片描述

Sakura,.
关注
  • 27
    点赞
  • 踩
  • 24
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
专栏目录
websocketproxy:Go 的 WebSocket 反向代理处理程序
05-31
网络套接字代理 WebsocketProxy 是一个构建在之上的 http.Handler 接口，您可以将其插入现有的 Go 网络服务器以提供 WebSocket 反向代理。 安装 go get github.com/koding/websocketproxy 例子 下面是一个简单的服务器，它代理给定的后端 URL package main import ( "flag" "net/http" "net/url" "github.com/koding/websocketproxy" ) var ( flagBackend = flag . String ( "backend" , "" , "Backend URL for proxying" ) ) func main () { u , err := url . Parse ( * flagBackend ) if e
Go语言的WebSocket框架与gorilla
禅与计算机程序设计艺术
01-21 1083
1.背景介绍 1. 背景介绍 WebSocket 是一种基于 TCP 的协议，它允许客户端和服务器之间建立持久连接，以实现实时通信。Go 语言的 WebSocket 框架是一个强大的工具，可以帮助开发者轻松地实现 WebSocket 功能。在本文中，我们将深入探讨 Go 语言的 WebSocket 框架，以及如何使用 gorilla 库来实现 WebSocket 功能。 2. 核心概念与联系...
2024年Go最新Golang Websocket框架：实时通信的新选择，2024年最新超详细
最新发布
2401_84919706的博客
05-14 831
需要这份系统化的资料的朋友，可以添加戳这里获取](https://bbs.csdn.net/topics/618658159)**// 升级HTTP请求为Websocket连接。// 读取客户端发送的消息。
Go语言实现百万级WebSocket连接
源自开发者
12-28 968
WebSocket是一种在单个TCP连接上进行全双工通信的协议。在高并发环境下，如何有效地管理和维护数百万个WebSocket连接是一个挑战，尤其是在资源有限的情况下。Go语言以其轻量级的协程和高效的网络I/O处理能力，成为实现大规模WebSocket连接的理想选择。本文将详细介绍如何在Go中构建一个能够支持百万级WebSocket连接的服务，包括系统架构设计、性能优化策略以及具体的实现步骤和代码示例。
2024年力扣热门100题之字母异位词分组【中等】(1)，2024年最新微信小程序页面跳转方法总结
2401_84910943的博客
05-11 418
输入: strs = [“eat”, “tea”, “tan”, “ate”, “nat”, “bat”]输出: [[“bat”],[“nat”,“tan”],[“ate”,“eat”,“tea”]]解法1：字符串拆分后排序，再聚合，最后再使用哈希归类。输入: strs = [“a”]解法2：求字符串权值，按权值分类。输入: strs = [“”]strs[i] 仅包含小写字母。输出: [[“a”]]输出: [[“”]]
【go语言之websocket】
qq_37674060的博客
01-08 5061
【go语言之websocket】
go-websocket
m0_61287310的博客
07-29 1321
文章目录1、websocket多人聊天室实现1.1.1. webSocket是什么erver.go文件代码hub.go文件代码data.go文件代码connection.go文件代码local.html文件代码2、代码关系图谱 1、websocket多人聊天室实现 1.1.1. webSocket是什么 WebSocket是一种在单个TCP连接上进行全双工通信的协议 WebSocket使得客户端和服务器之间的数据交换变得更加简单，允许服务端主动向客户端推送数据 在WebSocket API中，浏览器和服务
Go实现Websocket
wzbwzt的博客
08-03 1462
WebSocket WebSocket 是独立的、创建在 TCP 上的协议。 WebSocket在 HTML5 游戏和网页消息推送都使用比较多。WebSocket 是 HTML5 的重要特性，它实现了基于浏览器的远程socket，它使浏览器和服务器可以进行全双工通信，能更好的节省服务器资源和带宽并达到实时通讯的目的。 它与HTTP一样通过已建立的TCP连接来传输数据，但是它和HTTP最大不同是：WebSocket是一种双向通信协议。在建立连接后，WebSocket服务器端和客户端都能主动向对方发送或接收数
Go语言github.com/gorilla/websocket框架websocket协议通信实战
hsy12342611的专栏
07-13 2294
Go语言github.com/gorilla/websocket框架websocket协议通信实战
go语言WebSocket编程
liu289747235的专栏
01-30 864
WebSocket是一种在单个TCP连接上进行全双工通信的协议WebSocket使得客户端和服务器之间的数据交换变得更加简单，允许服务端主动向客户端推送数据在WebSocket API中，浏览器和服务器只需要完成一次握手，两者之间就直接可以创建持久性的连接，并进行双向数据传输需要安装第三方包：cmd中：go get -u -v github.com/gorilla/websocket。
go语言写的websocket代理服务器
04-19
在Go语言中，我们可以利用标准库`net/http`和第三方库如`gorilla/websocket`来实现WebSocket代理服务器。`gorilla/websocket`是一个广泛使用的Go库，提供了方便的API来创建WebSocket服务器和客户端。 首先，我们...
分布式WebSocket代理-Golang开发
05-26
dSock dSock是分布式WebSocket代理（在Go中，使用Redis）。 客户端可以进行身份​​验证和连接，并且您可以将文本/二进制消息作为API发送。 功能每个用户具有多个客户端和身份验证dSo dSock dSock是分布式WebSocket...
GoProxy:golang实现的高性能代理服务器-开源
07-20
GoProxy是一款高性能的http代理、https代理、socks5代理、ss代理、websocket代理、tcp代理、udp代理、游戏盾、游戏代理。 支持正向代理、反向代理、透明代理、internet nat 代理、https 代理负载均衡、http 代理负载...
WebSocket的那些事（5-Spring STOMP支持之连接外部消息代理）
lingbomanbu_lyl的博客
09-07 1997
Data/*** 用户唯一ID*//*** 用户昵称*//*** 用于指定用户消息推送的标识* @return*/@Override@Slf4j@Override// 如果是连接请求，记录userId下一节我们将会详细说明RabbitMQ STOMP适配器支持的各种消息目的地类型的区别以及适用场景。
websocket 反向代理
木木的博客
03-22 715
抓包 tcpdump -i eth0 port 80 and host 174.129.224.73
WebSocketClient设置代理Proxy
qq_46134549的博客
05-06 4639
工作中有时需要使用websocket通过线下机器请求线上接口，这时候就需要在配置proxy，配置代理方法很简单，首先需要找到WebSocketClient中设置代理的方法 Proxy方法里 InetSocketAddress里面可以写代理ip和端口号 然后在你自己的类中去调用该方法 TestWebSocketClient client = null; try { client.setProxy(new Proxy(Proxy.Type.HTTP, new Ine
go使用wss连接websocket服务器端,并发送数据，和接收服务器端不断传进来的消息
m0_46550201的博客
09-29 2803
连接格式为：wss://127.0.0.1：8100。使用wss连接而不是ws连接客户端。
golang代理websocket请求
FlayHigherGT的博客
10-23 4487
基础： golang websocket github地址: https://github.com/gorilla/websocket/ github库官方文档: https://godoc.org/github.com/gorilla/websocket golang websocket运行机制以及原理： https://www.jianshu.com/p/65ef71ddb910 golang websocket在线测试地址： http://www.websocket-test.com/
[Go WebSocket] 你的第一个Go WebSocket服务: echo server
Hull Qin
09-13 4574
如果你想实现WebSocket，基于压测数据，强烈建议你用Go！本文手把手教你实现一个echo服务，很适合用于新手入门！每一行代码都有详细注释。
GO语言实现websocket
07-28
在Go语言中实现WebSocket通信是相对简单的。首先，你需要导入`net/http`和`github.com/gorilla/websocket`两个包。然后，你可以按照以下步骤实现WebSocket： 1. 创建一个HTTP处理器函数，用于升级HTTP连接为WebSocket连接。 ```go func websocketHandler(w http.ResponseWriter, r *http.Request) { // 升级HTTP连接为WebSocket连接 conn, err := upgrader.Upgrade(w, r, nil) if err != nil { log.Println(err) return } // 在这里处理WebSocket连接 } ``` 2. 创建一个`Upgrader`对象，用于配置WebSocket连接的选项。 ```go var upgrader = websocket.Upgrader{ ReadBufferSize: 1024, WriteBufferSize: 1024, } ``` 3. 在WebSocket连接中实现读取和写入数据的逻辑。 ```go for { // 读取客户端发送的消息 messageType, p, err := conn.ReadMessage() if err != nil { log.Println(err) return } // 处理接收到的消息 // 发送消息给客户端 err = conn.WriteMessage(messageType, p) if err != nil { log.Println(err) return } } ``` 4. 创建一个HTTP服务器，并将WebSocket处理器注册到指定的路径上。 ```go http.HandleFunc("/websocket", websocketHandler) log.Fatal(http.ListenAndServe(":8080", nil)) ``` 这样，你就实现了一个简单的WebSocket服务器。你可以根据需要在`websocketHandler`函数中处理接收到的消息，以及在循环中发送消息给客户端。记得根据你的实际需求进行错误处理和逻辑编写。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值