本文中代碼可以在 github.com/alfred-zhong/wserver 獲取。

創新互聯公司主營尤溪網站建設的網絡公司,主營網站建設方案,app開發定制,尤溪h5重慶小程序開發公司搭建,尤溪網站營銷推廣歡迎尤溪等地區企業咨詢

背景

最近拿到需求要在網頁上展示報警信息。以往報警信息都是通過短信，微信和 App 推送給用戶的，現在要讓登錄用戶在網頁端也能實時接收到報警推送。

依稀記得以前工作的時候遇到過類似的需求。因為以前的瀏覽器標準比較陳舊，并且那時用 Java 較多，所以那時候解決這個問題就用了 Comet4J。具體的原理就是長輪詢，長鏈接。但現在畢竟 html5 流行開來了，IE 都被 Edge 接替了，再用以前這種技術就顯得過時。

很早以前就聽過 WebSocket 的大名，但因為那時很多用戶的瀏覽器還不支持，所以對這個技術也就是淺嘗輒止，沒有太深入研究過。現在趁著項目需要，就來稍微深入了解一下。

websocket 簡介

以往瀏覽器要獲取服務端數據，都是通過發送 HTTP 請求，然后等待服務端回應的。也就是說瀏覽器端一直是整個請求的發起者，只有它主動，才能獲取到數據。而要讓瀏覽器一側能夠獲取到服務端的實時數據，就需要不停地向服務端發起請求。雖然大多數情況下并沒有獲取到實際數據，但這大大增加了網絡壓力，對于服務端來說壓力也直線上升。

后來我們學會了使用長連接 + 長輪詢的方式。換句話說，也就是延長 HTTP 請求的存在時間，盡量保持 HTTP 連接。雖然這在一定程度上降低了不少壓力，但仍然需要不停地進行輪詢，也做不到真正的實時性。（借用一張圖）

隨著 HTML5 的到來，WebSocket 在 2011 年被定為標準（詳情請參見 RFC 6455）。

借用 《Go Web 編程》的話。WebSocket 采用了一些特殊的報頭，使得瀏覽器和服務器只需要做一個握手的動作，就可以在瀏覽器和服務器之間建立一條連接通道。且此連接會保持在活動狀態，你可以使用 JavaScript 來向連接寫入或從中接收數據，就像在使用一個常規的 TCP Socket 一樣。它解決了 Web 實時化的問題。

由于 WebSocket 是全雙工通信，所以當建立了 WebSocket 連接之后，接下來的通信就類似于傳統的 TCP 通信了。客戶端和服務端可以相互發送數據，不再有實時性的問題。

開發包的選擇

在 Go 官方的 SDK 中，并不包含對 WebSocket 的支持，所以必須使用第三方庫。

要使用 Golang 開發 WebSocket，選擇基本就在 x/net/websocket 和 gorilla/websocket 之間。《Go Web 編程》一書中的例子使用了 x/net/websocket 作為開發包，而且貌似它也更加官方且正式。而實際根據我在網上查詢得到的反饋看來，并非如此。x/net/websocket 貌似 Bug 較多，且較為不穩定，問題解決也并不及時。相比之下，gorilla/websocket 則更加優秀。

還有對于 Gorilla web toolkit 組織的貢獻，必須予以感謝。🙏。其下不僅有 WebSocket 的實現，也有一些其他工具。歡迎大家使用并且能夠給予反饋或貢獻。

推送服務實現
基本原理

項目初步設計如下：

server 啟動以后會注冊兩個 Handler。

websocketHandler 用于提供瀏覽器端發送 Upgrade 請求并升級為 WebSocket 連接。
pushHandler 用于提供外部推送端發送推送數據的請求。

瀏覽器首先連接 websocketHandler （默認地址為 ws://ip:port/ws）升級請求為 WebSocket 連接，當連接建立之后需要發送注冊信息進行注冊。這里注冊信息中包含一個 token 信息。server 會對提供的 token 進行驗證并獲取到相應的 userId（通常來說，一個 userId 可能同時關聯許多 token），并保存維護好 token, userId 和 conn（連接）之間的關系。

推送端發送推送數據的請求到 pushHandler（默認地址為 ws://ip:port/push），請求中包含了 userId 字段和 message 字段。server 會根據 userId 獲取到所有此時連接到該 server 的 conn，然后將 message 一一進行推送。

由于推送服務的實時性，推送的數據并沒有也不需要進行緩存。

代碼詳解

我在此處會稍微講述一下代碼的基本構成，也順便說說 Go 語言中一些常用的寫法和模式（本人也是從其他語言轉向 Go 語言，畢竟 Go 語言也相當年輕。所以有建議的話，敬請提出。）。由于 Go 語言的發明人和一些主要維護者大都來自于 C/C++ 語言，所以 Go 語言的代碼也更偏向于 C/C++ 系。

首先先看一下 Server 的結構：

// Server defines parameters for running websocket server.
type Server struct {
 // Address for server to listen on
 Addr string

 // Path for websocket request, default "/ws".
 WSPath string

 // Path for push message, default "/push".
 PushPath string

 // Upgrader is for upgrade connection to websocket connection using
 // "github.com/gorilla/websocket".
 //
 // If Upgrader is nil, default upgrader will be used. Default upgrader is
 // set ReadBufferSize and WriteBufferSize to 1024, and CheckOrigin always
 // returns true.
 Upgrader *websocket.Upgrader

 // Check token if it's valid and return userID. If token is valid, userID
 // must be returned and ok should be true. Otherwise ok should be false.
 AuthToken func(token string) (userID string, ok bool)

 // Authorize push request. Message will be sent if it returns true,
 // otherwise the request will be discarded. Default nil and push request
 // will always be accepted.
 PushAuth func(r *http.Request) bool

 wh *websocketHandler
 ph *pushHandler
}

另外有需要云服務器可以了解下創新互聯scvps.cn，海內外云服務器15元起步，三天無理由+7*72小時售后在線，公司持有idc許可證，提供“云服務器、裸金屬服務器、高防服務器、香港服務器、美國服務器、虛擬主機、免備案服務器”等云主機租用服務以及企業上云的綜合解決方案，具有“安全穩定、簡單易用、服務可用性高、性價比高”等特點與優勢，專為企業上云打造定制，能夠滿足用戶豐富、多元化的應用場景需求。

當前文章：利用Go語言編寫一個簡單的WebSocket推送服務-創新互聯
轉載來源：http://vcdvsql.cn/article36/phhpg.html

成都網站建設公司_創新互聯，為您提供網站設計公司、靜態網站、做網站、網站策劃、虛擬主機、云服務器

廣告

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