golang开发 gorilla websocket的使用示例详解

更新时间：2024年05月31日 09:15:37 作者：飞翔码农

这篇文章主要介绍了golang开发 gorilla websocket的使用示例详解,介绍了websocket的简单使用,我们使用的版本是1.3.0,具体操作方法跟随小编一起学习吧

很多APP都需要主动向用户推送消息，这就需要用到长连接的服务，即我们通常提到的websocket，同样也是使用socket服务，通信协议是基本类似的，在go中用的最多的、也是最简单的socket服务就是gorilla/websocket，它有21.1K的star，足以说明它的受欢迎程度，它的github地址是 https://github.com/gorilla/websocket，我们的长连接服务也是通过gorilla/websocket改造出来的。

websocket的简单使用

我们使用的版本是1.3.0，首先下载websocket
go get github.com/gorilla/websocket@v1.3.0

把websocket/examples/echo下面的 client.go server.go 现在下来拷贝到项目里面。
https://github.com/gorilla/websocket/blob/v1.3.0/examples/echo/server.go

https://github.com/gorilla/websocket/blob/v1.3.0/examples/echo/client.go

在一个终端执行 websocket 服务
go run server.go

recv: 2024-04-16 15:09:45.805438 +0800 CST m=+1.007536889
recv: 2024-04-16 15:09:46.805425 +0800 CST m=+2.007517605
recv: 2024-04-16 15:09:47.806274 +0800 CST m=+3.008359325
recv: 2024-04-16 15:09:48.80495 +0800 CST m=+4.007028866
recv: 2024-04-16 15:09:49.805743 +0800 CST m=+5.007816108
recv: 2024-04-16 15:09:50.806087 +0800 CST m=+6.008153310
recv: 2024-04-16 15:09:51.805348 +0800 CST m=+7.007407266

再打开一个终端执行 go run client.go，充当客户端

connecting to ws://localhost:8080/echo
recv: 2024-04-16 15:09:45.805438 +0800 CST m=+1.007536889
recv: 2024-04-16 15:09:46.805425 +0800 CST m=+2.007517605
recv: 2024-04-16 15:09:47.806274 +0800 CST m=+3.008359325
recv: 2024-04-16 15:09:48.80495 +0800 CST m=+4.007028866
recv: 2024-04-16 15:09:49.805743 +0800 CST m=+5.007816108
recv: 2024-04-16 15:09:50.806087 +0800 CST m=+6.008153310
recv: 2024-04-16 15:09:51.805348 +0800 CST m=+7.007407266

我们看看这个简单的例子。
client.go

go func() {
		defer close(done)
		for {
			_, message, err := c.ReadMessage()
			if err != nil {
				log.Println("read:", err)
				return
			}
			log.Printf("recv: %s", message)
		}
	}()
	ticker := time.NewTicker(time.Second)
	defer ticker.Stop()
	for {
		select {
		case <-done:
			return
		case t := <-ticker.C:
			err := c.WriteMessage(websocket.TextMessage, []byte(t.String()))
			if err != nil {
				log.Println("write:", err)
				return
			}
		case <-interrupt:
			log.Println("interrupt")
			// Cleanly close the connection by sending a close message and then
			// waiting (with timeout) for the server to close the connection.
			err := c.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
			if err != nil {
				log.Println("write close:", err)
				return
			}
			select {
			case <-done:
			case <-time.After(time.Second):
			}
			return
		}
	}

go func() 的 c.ReadMessage 不停的从socket里面读取数据并且打印。
for 循环不停的从1秒的定时器里面读取时间，写到socket

server.go

for {
		mt, message, err := c.ReadMessage()
		if err != nil {
			log.Println("read:", err)
			break
		}
		log.Printf("recv: %s", message)
		err = c.WriteMessage(mt, message)
		if err != nil {
			log.Println("write:", err)
			break
		}
	}

for c.ReadMessage 循环不停的从 socket读取数据，把数据打印之后，又写给客户端。
这是大致的websocket 客户端与服务端通信的简单例子。

改成自定义的协议

我们使用二进制数据流，自定义的协议是这样的，
先是无符号的 uint 占4个字节，表示行为逻辑，比如111-获取信息， 110- 加好友等等
后面是具体的数据，跟HTTP请求的GET或者POST参数类似,具体数据格式定义好就行，比如可以使用JSON数据、可以使用RPC定义好的数据格式

修改完之后 client.go 代码是这样的

for {
		select {
		case <-done:
			return
		case <-ticker.C:
			buf := bytes.NewBuffer([]byte{})
			binary.Write(buf, binary.BigEndian, uint32(110))
			binary.Write(buf, binary.BigEndian, []byte("我们都好"))
			err := c.WriteMessage(websocket.BinaryMessage, buf.Bytes())
			if err != nil {
				log.Println("write:", err)
				return
			}
		case <-interrupt:
			log.Println("interrupt")
			// Cleanly close the connection by sending a close message and then
			// waiting (with timeout) for the server to close the connection.
			err := c.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
			if err != nil {
				log.Println("write close:", err)
				return
			}
			select {
			case <-done:
			case <-time.After(time.Second):
			}
			return
		}
	}

主要是这个

buf := bytes.NewBuffer([]byte{})
binary.Write(buf, binary.BigEndian, uint32(110))
binary.Write(buf, binary.BigEndian, []byte("我们都好"))

创建一个bytes Buffer
写入一个uint32，占4个字节，表示行为逻辑
写入行为逻辑具体的数据
以上采用大端通信

修改之后的 server.go

for {
		mt, message, err := c.ReadMessage()
		if err != nil {
			log.Println("read:", err)
			break
		}
		log.Println("messagetype:%d", mt)
		log.Println("messagelen:%d", len(message))
		log.Println("messagetype:%d", binary.BigEndian.Uint32(message[0:4]))
		log.Println("recv: %s", string(message[4:]))
		err = c.WriteMessage(mt, message)
		if err != nil {
			log.Println("write:", err)
			break
		}
	}

mt websocket 类型
TextMessage = 1 文本类型传传输
BinaryMessage = 2 字节类型传输

从socket里面获取到message之后
先取前4个字节，表示逻辑行为
再取剩余的字节，表示行为逻辑需要的参数

我们重新执行一下，server.go 和 client.go

go run client.go
connecting to ws://localhost:8080/echo
recv: n我们都好
recv: n我们都好
go run server.go
messagetype:%d 2
messagelen:%d 16
messagetype:%d 110
recv: %s 我们都好

messagetype=2表示二进制通信
message长度是16，前四个字节是无符号整形，四个汉字占12个字节，一共16个字节，无符号uint是110表示加好友。

这就是一个简单自定义协议的socket服务。

如果有需要登录鉴权的话，在server.go中可以从r *http.Request 获取header或者cookie，进行登录鉴权验证。

生产的部署

因为是 websocket 协议，如果有代理服务器的话，需要在各代理服务器上的Nginx做协议升级，将HTTP 升级为 websocket ，比如下面的代理服务器。

map $http_upgrade $connection_upgrade {
default upgrade;
''      close;
}
server {
    listen 80;
    location /serv/ {
        proxy_set_header Upgrade $http_upgrade;
        proxy_set_header Connection $connection_upgrade;
        proxy_connect_timeout 3;
        proxy_redirect     off;
        proxy_read_timeout 3600;
        proxy_send_timeout 3600;
        proxy_pass http://push_web/serv/;
    }
}

到此这篇关于golang开发 gorilla websocket的使用示例详解的文章就介绍到这了,更多相关golang开发 gorilla websocket内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

Go语言实现的可读性更高的并发神库详解
这篇文章主要为大家介绍了Go语言实现的可读性更高的并发神库详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2023-01-01
golang利用redis和gin实现保存登录状态校验登录功能
这篇文章主要介绍了golang利用redis和gin实现保存登录状态校验登录功能,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友参考下吧
2024-01-01
Go实现替换(覆盖)文件某一行内容的示例代码
本文主要介绍了Go实现替换(覆盖)文件某一行内容的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2022-07-07
基于Go+OpenCV实现人脸识别功能的详细示例
OpenCV是一个强大的计算机视觉库，提供了丰富的图像处理和计算机视觉算法，本文将向你介绍在Mac上安装OpenCV的步骤，并演示如何使用Go的OpenCV绑定库进行人脸识别,需要的朋友可以参考下
2023-07-07
Golang中基础的命令行模块urfave/cli的用法说明
这篇文章主要介绍了Golang中基础的命令行模块urfave/cli的用法说明，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-12-12
详解Golang中Context的三个常见应用场景
Golang context主要用于定义超时取消，取消后续操作，在不同操作中传递值。本文通过简单易懂的示例进行说明，感兴趣的可以了解一下
2022-12-12
Golang接口使用教程详解
在 Go 语言中接口包含两种含义:它既是方法的集合, 同时还是一种类型并且在Go 语言中是隐式实现的。本文通过示例详细介绍了Golang接口的使用，需要的可以参考一下
2022-09-09
Go条件控制语句详解(if-else、switch和select)
条件语句用于检查一个条件是否为真,并根据条件的真假来决定是否执行相应的代码,下面这篇文章主要给大家介绍了关于Go条件控制语句(if-else、switch和select)的相关资料,需要的朋友可以参考下
2024-03-03
一文详解Golang中的基础语法
这篇文章主要为大家详细介绍了Golang中基础语法的相关知识，文中的示例代码讲解详细，对我们学习Golang有一定的帮助，感兴趣的可以了解一下
2023-03-03
一文教你如何在Golang中用好泛型
golang的泛型已经出来了一年多了，从提案被接受开始我就在关注泛型了，好用是好用，但问题也很多，所以本文就来教大家如何在Golang中用好泛型吧
2023-07-07