golang使用Websocket实例

本项目使用 github.com/gorilla/websocket 库 安装:

go 复制代码
go get github.com/gorilla/websocket

服务端代码

go 复制代码
package main

import (
	"fmt"
	"log"
	"net/http"
	"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{
		ReadBufferSize:  1024,
		WriteBufferSize: 1024,
		CheckOrigin: func(r *http.Request) bool {
			return true
		},
	} // 使用默认的选项
	// 第三个参数是响应头,默认会初始化
	conn, err := upgrader.Upgrade(resp, req, nil)
	if err != nil {
		log.Println(err)
		return
	}
	defer conn.Close()

	// 读取客户端的发送额消息,并返回
	fmt.Println("websocket 连接成功")
	// fmt.Println(conn)
	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, []byte("The server has received the message"))
		if err != nil {
			log.Println("write error:", err)
			return
		}
	}
}

func main() {
	WebSocketServer()
}

客户端

go 复制代码
package main

import (
	"encoding/json"
	"fmt"
	"log"
	"time"
	"github.com/gorilla/websocket"
)

func main() {
	wsUrl := "ws://localhost:8002/wshandler"
	// 连接WebSocket服务器
	conn, _, err := websocket.DefaultDialer.Dial(wsUrl, nil)
	if err != nil {
		log.Println(err)
		return
	}

	log.Println("连接成功")

	go func() {
		for {
			// 读取消息
			messageType, message, err := conn.ReadMessage()
			if err != nil {
				log.Fatal(err)
			}
			log.Println("消息类型:", messageType, "消息数据:", string(message))
		}
	}()

	// 发送消息
	controlMessage := map[string]interface{}{
		"type": "control",
		"data": "hello word",
	}
	jsonData, err := json.Marshal(controlMessage)
	if err != nil {
		fmt.Println("Error marshalling JSON:", err)
		return
	}

	// 间隔2秒发送一次消息
	ticket := time.NewTicker(2 * time.Second)
	defer ticket.Stop()
	for {
		<-ticket.C
		// 2. 发送消息
		err = conn.WriteMessage(websocket.TextMessage, []byte(string(jsonData)))
		if err != nil {
			log.Fatal(err)
		}
		log.Println("发送消息成功")
	}

}
相关推荐
workflower3 小时前
单元测试-例子
java·开发语言·算法·django·个人开发·结对编程
YuanlongWang3 小时前
C# 基础——装箱和拆箱
java·开发语言·c#
b78gb3 小时前
电商秒杀系统设计 Java+MySQL实现高并发库存管理与订单处理
java·开发语言·mysql
LXS_3574 小时前
Day 05 C++ 入门 之 指针
开发语言·c++·笔记·学习方法·改行学it
etsuyou6 小时前
js前端this指向规则
开发语言·前端·javascript
shizhenshide6 小时前
为什么有时候 reCAPTCHA 通过率偏低,常见原因有哪些
开发语言·php·验证码·captcha·recaptcha·ezcaptcha
mit6.8246 小时前
[Agent可视化] 配置系统 | 实现AI模型切换 | 热重载机制 | fsnotify库(go)
开发语言·人工智能·golang
友友马6 小时前
『 QT 』QT控件属性全解析 (一)
开发语言·前端·qt
小白学大数据7 小时前
实战:Python爬虫如何模拟登录与维持会话状态
开发语言·爬虫·python
一念&7 小时前
每日一个C语言知识:C 结构体
c语言·开发语言