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)
	}
}
相关推荐
跟着珅聪学java1 小时前
spring boot +Elment UI 上传文件教程
java·spring boot·后端·ui·elementui·vue
我命由我123451 小时前
Spring Boot 自定义日志打印(日志级别、logback-spring.xml 文件、自定义日志打印解读)
java·开发语言·jvm·spring boot·spring·java-ee·logback
徐小黑ACG2 小时前
GO语言 使用protobuf
开发语言·后端·golang·protobuf
豆芽脚脚2 小时前
合并相同 patient_id 的 JSON 数据为数组
postgresql·json
0白露3 小时前
Apifox Helper 与 Swagger3 区别
开发语言
Tanecious.4 小时前
机器视觉--python基础语法
开发语言·python
叠叠乐5 小时前
rust Send Sync 以及对象安全和对象不安全
开发语言·安全·rust
战族狼魂5 小时前
CSGO 皮肤交易平台后端 (Spring Boot) 代码结构与示例
java·spring boot·后端
Tttian6226 小时前
Python办公自动化(3)对Excel的操作
开发语言·python·excel
杉之6 小时前
常见前端GET请求以及对应的Spring后端接收接口写法
java·前端·后端·spring·vue