如何基于Go实现websocket

这篇文章主要介绍“如何基于Go实现 websocket”，在日常操作中，相信很多人在如何基于Go实现 websocket问题上存在疑惑，小编查阅了各式资料，整理出简单好用的操作方法，希望对大家解答”如何基于Go实现 websocket”的疑惑有所帮助！接下来，请跟着小编一起来学习吧！

金沙网站建设公司成都创新互联公司,金沙网站设计制作，有大型网站制作公司丰富经验。已为金沙上1000家提供企业网站建设服务。企业网站搭建\外贸网站制作要多少钱，请找那个售后服务好的金沙做网站的公司定做！

1.websocket 介绍

WebSocket协议是基于TCP的一种新的网络协议。它实现了浏览器与服务器全双工(full-duplex)通信——允许服务器主动发送信息给客户端。

《WebSocket实现实时双向通信》已经介绍java与android如何基于websocket通信。

2.实现

在golang语言中，目前有两种比较常用的实现方式：一个是golang自带的库，另一个是gorilla，功能强大。

go get github.com/gorilla/websocket

websocket实现代码

package gowebsocket

import (
	"encoding/json"
	"goweb/common/response"
	"log"
	"net/http"
	"strconv"

	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
)

// ClientManager is a websocket manager
type ClientManager struct {
	Clients    map[*Client]bool
	Broadcast  chan []byte
	Register   chan *Client
	Unregister chan *Client
}

// Client is a websocket client
type Client struct {
	ID     int
	Socket *websocket.Conn
	Send   chan []byte
}

// Message is return msg
type Message struct {
	Sender    string `json:"sender,omitempty"`
	Recipient string `json:"recipient,omitempty"`
	Content   string `json:"content,omitempty"`
}

// Manager define a ws server manager
var Manager = ClientManager{
	Broadcast:  make(chan []byte),
	Register:   make(chan *Client),
	Unregister: make(chan *Client),
	Clients:    make(map[*Client]bool),
}

//启动websocket服务
// Start is  项目运行前, 协程开启start -> go Manager.Start()
func (manager *ClientManager) Start() {
	for {
		log.Println("<---管道通信--->")

		select {
		case conn := <-Manager.Register:
			log.Printf("新用户加入:%v", conn.ID)
			Manager.Clients[conn] = true
			jsonMessage, _ := json.Marshal(&Message{Content: "Successful connection to socket service"})
			Manager.Send(jsonMessage, conn)
		case conn := <-Manager.Unregister:
			log.Printf("用户离开:%v", conn.ID)
			if _, ok := Manager.Clients[conn]; ok {
				close(conn.Send)
				delete(Manager.Clients, conn)
				jsonMessage, _ := json.Marshal(&Message{Content: "A socket has disconnected"})
				Manager.Send(jsonMessage, conn)
			}

		case message := <-Manager.Broadcast:

			jsonMessage, _ := json.Marshal(&Message{Content: string(message)})
			for conn := range Manager.Clients {
				select {
				case conn.Send <- jsonMessage:
				default:
					close(conn.Send)
					delete(Manager.Clients, conn)
				}
			}
		}
	}
}

// Send is to send ws message to ws client
func (manager *ClientManager) Send(message []byte, ignore *Client) {

	for conn := range manager.Clients {
		// if conn != ignore { //向除了自己的socket 用户发送
		conn.Send <- message
		// }
	}
}

func (c *Client) Read() {
	defer func() {
		Manager.Unregister <- c
		c.Socket.Close()
	}()

	for {

		_, message, err := c.Socket.ReadMessage()
		if err != nil {
			Manager.Unregister <- c
			c.Socket.Close()
			break
		}
		log.Printf("读取到客户端的信息:%s", string(message))
		Manager.Broadcast <- message
	}
}

func (c *Client) Write() {
	defer func() {
		c.Socket.Close()
	}()

	for {
		select {
		case message, ok := <-c.Send:
			if !ok {
				c.Socket.WriteMessage(websocket.CloseMessage, []byte{})
				return
			}
			log.Printf("发送到到客户端的信息:%s", string(message))

			c.Socket.WriteMessage(websocket.TextMessage, message)
		}
	}

}
// xxx/ws/2  2表示客户端userId 编号
//WsHandler socket 连接 中间件 作用:升级协议,用户验证,自定义信息等
func WsHandler(c *gin.Context) {

	userId, err := strconv.Atoi(c.Param("userId"))
	if err != nil {
		response.FailResult(http.StatusBadRequest, err.Error(), c)
		return
	}

	conn, err := websocket.Upgrade(c.Writer, c.Request, nil, 1024, 1024)
	if err != nil {
		http.NotFound(c.Writer, c.Request)
		return
	}
	//可以添加用户信息验证

	client := &Client{
		ID:     userId,
		Socket: conn,
		Send:   make(chan []byte),
	}
	Manager.Register <- client
	go client.Read()
	go client.Write()
}

入口引用

	go gowebsocket.Manager.Start()

	this.GET("/ws/:userId", gowebsocket.WsHandler)

测试

    
    
    
        .body{text-align: center;}
        #open{width: 120px;height: 35px;}
        #close{width: 120px;height: 35px;}
        #text{display: inline-block;margin: auto;margin-top: 10px;margin-bottom: 10px;width: 240px;}
    


打开连接
关闭连接





发送