golang Gin实现websocket

最新推荐文章于 2024-08-04 16:51:22 发布

悟世君子

最新推荐文章于 2024-08-04 16:51:22 发布

阅读量2.7k

点赞数 2

分类专栏： golang 文章标签： golang gin 开发语言 websocket

本文链接：https://blog.csdn.net/wsjzzcbq/article/details/129751015

版权

golang 专栏收录该内容

15 篇文章 0 订阅

订阅专栏

golang使用 Gin实现 websocket，这里笔者重新搭建一个项目

1、创建项目安装依赖

项目名为 go-gin-websocket

在指定文件夹下，新建项目文件夹 go-gin-websocket

进入项目文件夹，打开cmd窗口，在项目（go-gin-websocket）文件夹路径下，执行初始化命令 go mod init go-gin-websocket

go mod init go-gin-websocket

安装依赖

安装gin

go get -u github.com/gin-gonic/gin

安装websocket

go get github.com/gorilla/websocket

2、搭建项目

在项目目录下新建controller、static、templates、ws文件夹，分别用来存放controller代码、静态文件、页面、websocket代码

在templates目录下新建 index.html页面

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>
hello
<script>
    var webSocket;
    try {
        webSocket = new WebSocket('ws://127.0.0.1:8080/ws?id=abc123456789');
    }catch (e) {
        console.log(e)
    }

    webSocket.onerror = function(event) {}

    //与WebSocket建立连接
    webSocket.onopen = function(event) {
        console.log('建立连接')
        console.log(JSON.stringify(event))
        var msg = "身无彩凤双飞翼，心有灵犀一点通"
        //发送消息
        webSocket.send(msg);
    };
    //接收服务端发送的消息
    webSocket.onmessage=function (m) {
        console.log(m)
        console.log(m.data)
    }
</script>
</body>
</html>

在controller目录下新建 page_controller.go

package controller

import "github.com/gin-gonic/gin"

func Index(c *gin.Context) {
	c.HTML(200, "index.html", nil)
}

在ws目录下新建 websocket.go

package ws

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
	"log"
	"net/http"
)

func InitWebSocket(c *gin.Context) {
	var upgrader = websocket.Upgrader{
		ReadBufferSize:  1024,
		WriteBufferSize: 1024,
		CheckOrigin: func(r *http.Request) bool {
			log.Println("升级协议", r.Header["User-Agent"])
			return true
		},
	}

	conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
	if err != nil {
		log.Println(err)
		return
	}
	defer conn.Close()
	userId := c.Query("id")
	log.Println("用户id:", userId)

	for {
		mt, message, err := conn.ReadMessage()

		log.Println("获取客户端发送的消息:" + string(message))
		fmt.Println(mt)
		if err != nil {
			log.Println(err)
			break
		}

		var msg = "春风得意马蹄疾,一日看尽长安花"
		err2 := conn.WriteMessage(websocket.TextMessage, []byte(msg))
		if err2 != nil {
			log.Println("write:", err2)
			break
		}
	}

}

在项目根目录下新建 main.go

main.go内容

package main

import (
	"github.com/gin-gonic/gin"
	"go-gin-websocket/controller"
	"go-gin-websocket/ws"
)

func main() {
	e := gin.Default()
	//html页面位置
	e.LoadHTMLGlob("templates/*")
	//静态文件位置
	e.Static("/static", "./static")

	e.GET("/", controller.Index)
	e.GET("/ws", ws.InitWebSocket)

	e.Run("0.0.0.0:8080")
}

3、运行测试

浏览器请求 http://localhost:8080/

4、添加session

添加session用来存储在线用户

这里使用一个map用来保存，key是用户id，value是websocket连接，再加一个读写锁保证线程安全

package ws

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
	"log"
	"net/http"
	"sync"
)

var connects = make(map[string]*websocket.Conn)
var lock sync.RWMutex

func addSession(id string, conn *websocket.Conn) {
	lock.Lock()
	connects[id] = conn
	defer lock.Unlock()
}
func getSession(id string) *websocket.Conn {
	lock.RLock()
	defer lock.RUnlock()
	return connects[id]
}
func removeSession(id string) {
	lock.Lock()
	delete(connects, id)
	defer lock.Unlock()
}

func InitWebSocket(c *gin.Context) {
	var upgrader = websocket.Upgrader{
		ReadBufferSize:  1024,
		WriteBufferSize: 1024,
		CheckOrigin: func(r *http.Request) bool {
			log.Println("升级协议", r.Header["User-Agent"])
			return true
		},
	}

	conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
	if err != nil {
		log.Println(err)
		return
	}
	defer conn.Close()
	userId := c.Query("id")
	log.Println("用户id:", userId)

	//添加session
	addSession(userId, conn)

	for {
		mt, message, err := conn.ReadMessage()

		log.Println("获取客户端发送的消息:" + string(message))
		fmt.Println(mt)
		if err != nil {
			log.Println(err)
			break
		}

		var msg = "春风得意马蹄疾,一日看尽长安花"
		err2 := conn.WriteMessage(websocket.TextMessage, []byte(msg))
		if err2 != nil {
			log.Println("write:", err2)
			break
		}
	}

}

至此完