golang websocket 数据处理和返回JSON数据示例

golang中websocket数据处理和返回json数据示例, 直接上代码:

Go 复制代码
// @author [email protected]
// golang websocket 数据处理和返回JSON数据示例,
// 这个函数返回 http.HandlerFunc 
// 将http请求升级为websocket请求 这个需要依赖第三方包 go get github.com/gorilla/websocket
func  WsHandler() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		var upgrader = websocket.Upgrader{
			ReadBufferSize:  1024,
			WriteBufferSize: 1024,
		}
		// 将普通链接转换为socket链接 
		ws, err := upgrader.Upgrade(w, r, nil)
		if err != nil {
			return
		}
		defer ws.Close()

		// 向websocket发送数据
		_ = sendStats(ws, 1*time.Second)
	}
	return nil
}

// 向websocket发送json数据 注意这里的 for range ticker.C { 的使用
func sendStats(conn *websocket.Conn, frequency time.Duration) error {
	ticker := time.NewTicker(frequency) // 定时器 在指定的时间间隔中执行
	defer ticker.Stop() // 确保退出时关闭定时器

	// 循环定时器管道,这里的for循环会 每个frequency执行一次
	for range ticker.C {
		w, err := conn.NextWriter(websocket.TextMessage)
		if err != nil {
			return err
		}
		if err := WriteValues(w); err != nil {
			return err
		}
		if err := w.Close(); err != nil {
			return err
		}
	}
}

// 同步写入锁
var mu  sync.Mutex

func WriteValues(w io.Writer) error {
	mu.Lock()
	defer mu.Unlock()
	// 定义并初始化要写到ws中的数据
	m := map[string]any{
		"a":    "123",
		"ts": time.Now().UnixMilli(),
	}
	// 其他信息写入

	// 将数据m json编码后写入到w
	if err := json.NewEncoder(w).Encode(m); err != nil {
		return fmt.Errorf("failed to write/convert values to json: %v", err)
	}
}
相关推荐
光而不耀@lgy7 分钟前
C++初登门槛
linux·开发语言·网络·c++·后端
lkbhua莱克瓦2411 分钟前
用C语言实现——一个中缀表达式的计算器。支持用户输入和动画演示过程。
c语言·开发语言·数据结构·链表·学习方法·交友·计算器
Mr__Miss15 分钟前
面试踩过的坑
java·开发语言
啊丢_17 分钟前
C++——Lambda表达式
开发语言·c++
方圆想当图灵26 分钟前
由 Mybatis 源码畅谈软件设计(七):SQL “染色” 拦截器实战
后端·mybatis·代码规范
jjw_zyfx34 分钟前
成熟的前端vue vite websocket,Django后端实现方案包含主动断开websocket连接的实现
前端·vue.js·websocket
毅航1 小时前
MyBatis 事务管理:一文掌握Mybatis事务管理核心逻辑
java·后端·mybatis
Chh07151 小时前
《R语言SCI期刊论文绘图专题计划》大纲
开发语言·r语言
Yeats_Liao1 小时前
Go 语言 TCP 端口扫描器实现与 Goroutine 池原理
开发语言·tcp/ip·golang
我的golang之路果然有问题1 小时前
速成GO访问sql,个人笔记
经验分享·笔记·后端·sql·golang·go·database