最近要在一個go實現(xiàn)的K線服務器上追加一個websocket推送功能。

初步設想是在各品種每分鐘的數(shù)據(jù)完成后立即向各訂閱單位推送對應的K線數(shù)據(jù)。

一、場景抽象

1用戶連接上服務器

2用戶訂閱某品種的某周期

3用戶退訂某品種某周期

4用戶斷開服務器

訂閱：

推送：

二、程序框架

由于go語言不熟，幸好網上有人做了成熟的框架：GitHub - gorilla/websocket: A fast, well-tested and widely used WebSocket implementation for Go.這個解決了怎樣使用websocket的問題

三、主要業(yè)務邏輯

每個連接開兩個線程

一個線程進行數(shù)據(jù)寫入writePump，一個線程進行數(shù)據(jù)讀取readPump，每當有客戶端連接上服務器后，立即開啟這兩個線程。

讀線程邏輯圖：

寫線程邏輯圖：

程序啟動時開一個總控線程func (h *Hub) Run()

此線程用來處理客戶端的連接，斷開，訂閱和向訂閱信息的客戶端作相應信息的廣播。

邏輯圖如下：

處理注冊

處理注冊的邏輯非常簡單，只是將客戶端的指針放置于指針map中即可，代碼如下：

// 注冊，客戶端連接上來的處理邏輯
func (h *Hub) registerMsg(client *Client) {
	h.clients[client] = true
}

處理注銷

注銷需要分兩步走，首先把客戶端指針集合中的對應指針刪除，同時刪除對應的發(fā)送通道。然后遍歷內容訂閱集合，刪除對應的客戶端指針的元素，最后，若發(fā)現(xiàn)某訂閱內容上已經沒有任何客戶端指針，將其訂閱內容一并刪除。

程序邏輯如下：

訂閱處理

訂閱最主要的是訂閱關系，我們可以用合約.周期作為key，使用客戶端連接上來的指針集作為value,每當有用戶訂閱某內容時，將相應內容下面的指針集上加上對應此用戶的指針即可，訂閱關系如下：

推送處理

推送處理分兩種類型，一種為用戶訂閱的數(shù)據(jù)推送給到他們，一種為心跳包推送給到30秒內沒有數(shù)據(jù)推送或訂閱動作的用戶。

推送訂閱的內容：當用戶訂閱的類型數(shù)據(jù)到達時，系統(tǒng)檢查此訂閱上的用戶指針，將數(shù)據(jù)推送到對應指針的通道中，由各通道自行推送給各自的客戶端。

推送心跳包：當某個連接上來的客戶端30秒內沒有訂閱請求或是推送數(shù)據(jù)，系統(tǒng)將自動推送心跳包，以維持連接不被斷開。

到此這篇關于Go語言實現(xiàn)websocket推送程序的文章就介紹到這了,更多相關Go websocket內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！