Go 实现 Websocket 服务以及代理

这篇具有很好参考价值的文章主要介绍了Go 实现 Websocket 服务以及代理。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

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. 服务器端推送

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

服务器推送,服务器端,当状态改变时,将数据发送到浏览器端。
go websocket,Golang 网络编程,golang,websocket,开发语言,网络协议

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

3. WebSocket 和 http

  • 相同点
  1. 应用层协议

  2. B/S 架构中使用

  3. 基于 TCP 协议

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

  • 不同点

go websocket,Golang 网络编程,golang,websocket,开发语言,网络协议

4. WebSocket 握手过程

通过 HTTP 请求响应,中的头信息,完成 websocket 握手,如图:
go websocket,Golang 网络编程,golang,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 消息类型

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

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

  • PingMessagePongMessage : 是保持心跳的帧

    • 发送方 -> 接收方是 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 是否能连接并且发送消息

  • 消息发送成功,同时也接收到来服务端的消息
    go websocket,Golang 网络编程,golang,websocket,开发语言,网络协议

  • 消息接收成功
    go websocket,Golang 网络编程,golang,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)
}

go websocket,Golang 网络编程,golang,websocket,开发语言,网络协议

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
		}
	}
}

每隔三秒可以看到服务推送过来的消息go websocket,Golang 网络编程,golang,websocket,开发语言,网络协议文章来源地址https://www.toymoban.com/news/detail-829720.html

到了这里,关于Go 实现 Websocket 服务以及代理的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • Go语言github.com/gorilla/websocket框架websocket协议通信实战

          websocket是实际开发中比较常用的应用层协议,本文利用github.com/gorilla/websocket框架进行websocket通信实战。 目录 1.下载github.com/gorilla/websocket 2.websocket服务端 3.websocket Go客户端 4.websocket 网页客户端 5.运行结果展示 go get github.com/gorilla/websocket 服务器:  Go语言客户端:  We

    2024年02月16日
    浏览(49)
  • Go语言的WebSocket框架与gorilla

    WebSocket 是一种基于 TCP 的协议,它允许客户端和服务器之间建立持久连接,以实现实时通信。Go 语言的 WebSocket 框架是一个强大的工具,可以帮助开发者轻松地实现 WebSocket 功能。在本文中,我们将深入探讨 Go 语言的 WebSocket 框架,以及如何使用 gorilla 库来实现 WebSocket 功能。

    2024年02月20日
    浏览(33)
  • Go语言网络编程(socket编程)WebSocket编程

    WebSocket是一种在单个TCP连接上进行全双工通信的协议 WebSocket使得客户端和服务器之间的数据交换变得更加简单,允许服务端主动向客户端推送数据 在WebSocket API中,浏览器和服务器只需要完成一次握手,两者之间就直接可以创建持久性的连接,并进行双向数据传输 需要安装第

    2024年02月09日
    浏览(77)
  • go入门实践四-go实现一个简单的tcp-socks5代理服务

    SOCKS是一种网络传输协议,主要用于客户端与外网服务器之间通讯的中间传递。协议在应用层和传输层之间。 本文使用先了解 socks 协议。然后实现一个socks5的tcp代理服务端。最后,进行抓包验证。 本文完整代码见仓库:laboratory/16-go-socks5 socks 协议相对http和tcp协议,还是比较

    2024年02月04日
    浏览(43)
  • 【go】gorilla/websocket如何判断客户端强制断开连接

    当客户端因为某些问题异常关闭连接时,可以判断关闭连接的异常类型 通过调用websocket.IsCloseError或websocket.IsUnexpectedCloseError即可 其中github源码如下 异常类型如下

    2024年02月16日
    浏览(60)
  • go cannot find package “github.com/gorilla/websocket“解读

    在Go开发过程中,我们经常会依赖第三方库来简化开发工作。而使用 go get 命令安装这些库时,有时候我们可能会遇到类似于以下错误的情况: 这个错误意味着Go编译器无法找到名为\\\"github.com/gorilla/websocket\\\"的包。那么我们应该如何解决这个问题呢?下面提供了几种常见的解决方

    2024年01月24日
    浏览(90)
  • Golang 包详解以及go mod

    包(package)是多个 Go 源码的集合,是一种高级的代码复用方案,Go 语言为我们提供了 很多内置包,如 fmt、strconv、strings、sort、errors、time、encoding/json、os、io 等。 Golang 中的包可以分为三种 :1、系统内置包 2、自定义包 3、第三方包 系统内置包: Golang 语言给我们提供的内置

    2024年02月14日
    浏览(51)
  • WebSocket 详解,以及用QWebSocket 实现服务端和客户端(含代码例子)

    目录 前言: 1、WebSocket 诞生背景 2、WebSocket的特点: 3、 WebSocket 简介 4、WebSocket 优点 5、QWebSocket通讯—客户端: 6、QWebSocket通讯—服务端: 前言:         要是对WebSocket 的基本知识都了解了,可以直接移步,实际如何使用这个类 (二) 用QWebSocket 实现服务端和客户端(

    2024年02月16日
    浏览(49)
  • (一)WebSocket 详解,以及用QWebSocket 实现服务端和客户端(含代码例子)

    目录 前言: 1、WebSocket 诞生背景 2、WebSocket的特点: 3、 WebSocket 简介 4、WebSocket 优点 5、QWebSocket通讯—客户端: 6、QWebSocket通讯—服务端: 前言:         要是对WebSocket 的基本知识都了解了,可以直接移步,实际如何使用这个类 (二) 用QWebSocket 实现服务端和客户端(

    2024年02月03日
    浏览(53)
  • golang Gin实现websocket

    golang使用 Gin实现 websocket,这里笔者重新搭建一个项目 项目名为 go-gin-websocket 在指定文件夹下,新建项目文件夹 go-gin-websocket 进入项目文件夹,打开cmd窗口,在项目(go-gin-websocket)文件夹路径下,执行初始化命令 go mod init go-gin-websocket 安装依赖 安装gin  安装websocket 在项

    2024年02月06日
    浏览(49)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包