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("发送消息成功")
	}

}
相关推荐
喝可乐的布偶猫34 分钟前
韩顺平之第九章综合练习-----------房屋出租管理系统
java·开发语言·ide·eclipse
江山如画,佳人北望1 小时前
C#程序入门
开发语言·windows·c#
coding随想2 小时前
JavaScript中的BOM:Window对象全解析
开发语言·javascript·ecmascript
念九_ysl2 小时前
Java 使用 OpenHTMLToPDF + Batik 将含 SVG 遮罩的 HTML 转为 PDF 的完整实践
java·开发语言·pdf
yaoxin5211233 小时前
124. Java 泛型 - 有界类型参数
java·开发语言
liulilittle3 小时前
深度剖析:OPENPPP2 libtcpip 实现原理与架构设计
开发语言·网络·c++·tcp/ip·智能路由器·tcp·通信
88号技师3 小时前
2025年6月一区-田忌赛马优化算法Tianji’s horse racing optimization-附Matlab免费代码
开发语言·算法·matlab·优化算法
勤奋的知更鸟3 小时前
Java 编程之模板方法模式
java·开发语言·模板方法模式
2501_916013743 小时前
iOS 多线程导致接口乱序?抓包还原 + 请求调度优化实战
websocket·网络协议·tcp/ip·http·网络安全·https·udp
上单带刀不带妹4 小时前
手写 Vue 中虚拟 DOM 到真实 DOM 的完整过程
开发语言·前端·javascript·vue.js·前端框架