如何基于Go實現(xiàn)websocket

這篇文章主要介紹“如何基于Go實現(xiàn) websocket”，在日常操作中，相信很多人在如何基于Go實現(xiàn) websocket問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”如何基于Go實現(xiàn) websocket”的疑惑有所幫助！接下來，請跟著小編一起來學習吧！

金沙網(wǎng)站建設公司成都創(chuàng)新互聯(lián)公司,金沙網(wǎng)站設計制作，有大型網(wǎng)站制作公司豐富經(jīng)驗。已為金沙上1000家提供企業(yè)網(wǎng)站建設服務。企業(yè)網(wǎng)站搭建\外貿(mào)網(wǎng)站制作要多少錢，請找那個售后服務好的金沙做網(wǎng)站的公司定做！

1.websocket 介紹

WebSocket協(xié)議是基于TCP的一種新的網(wǎng)絡協(xié)議。它實現(xiàn)了瀏覽器與服務器全雙工(full-duplex)通信——允許服務器主動發(fā)送信息給客戶端。

《WebSocket實現(xiàn)實時雙向通信》已經(jīng)介紹java與android如何基于websocket通信。

2.實現(xiàn)

在golang語言中，目前有兩種比較常用的實現(xiàn)方式：一個是golang自帶的庫，另一個是gorilla，功能強大。

go get github.com/gorilla/websocket

websocket實現(xiàn)代碼

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  項目運行前, 協(xié)程開啟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 用戶發(fā)送
		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("發(fā)送到到客戶端的信息:%s", string(message))

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

}
// xxx/ws/2  2表示客戶端userId 編號
//WsHandler socket 連接 中間件 作用:升級協(xié)議,用戶驗證,自定義信息等
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)

測試

<!DOCTYPE html>
<html>
<head>
    <meta charset="utf-8">
    <title></title>
    <style type="text/css">
        .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;}
    </style>
</head>
<body class="body">
<button id="open">打開連接</button>
<button id="close">關閉連接</button>
<br/>
<input type="text" name="text" id="text" value="" />
<br/>
<button id="send">發(fā)送</button>
<div id="msg">

</div>
</body>
<script>
    var openbtn = document.getElementById("open")
    var closebtn = document.getElementById("close")
    var text = document.getElementById("text")
    var send = document.getElementById("send")
    var msg = document.getElementById("msg")
    var websocket
    openbtn.onclick = function(){
        websocket = new WebSocket("ws://localhost:8080/api/v1/ws/11")
        websocket.onopen=function(){
            console.log("connected");
        }
        websocket.onmessage = function(e){
            console.log(e);
            msg.innerHTML += '<br/>接收：'+e.data;
        }
        websocket.onclose=function(e){
            console.log("closed",e);
        }
    }
    closebtn.onclick=function(){
        websocket.close(1000,"close")
    }
    send.onclick=function(){
        var msg = text.value
        websocket.send(msg)
    }
</script>
</html>

如何基于Go實現(xiàn) websocket

到此，關于“如何基于Go實現(xiàn) websocket”的學習就結束了，希望能夠解決大家的疑惑。理論與實踐的搭配能更好的幫助大家學習，快去試試吧！若想繼續(xù)學習更多相關知識，請繼續(xù)關注創(chuàng)新互聯(lián)網(wǎng)站，小編會繼續(xù)努力為大家?guī)砀鄬嵱玫奈恼拢?/p>

網(wǎng)站標題：如何基于Go實現(xiàn)websocket
URL網(wǎng)址：http://muchs.cn/article0/jpijoo.html

成都網(wǎng)站建設公司_創(chuàng)新互聯(lián)，為您提供用戶體驗、建站公司、營銷型網(wǎng)站建設、網(wǎng)站設計公司、網(wǎng)站策劃、ChatGPT

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉載內(nèi)容為主，如果涉及侵權請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉載，或轉載時需注明來源： 創(chuàng)新互聯(lián)

成都做網(wǎng)站