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

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

一、場(chǎng)景抽象

1用戶連接上服務(wù)器

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

3用戶退訂某品種某周期

4用戶斷開服務(wù)器

訂閱：

推送：

二、程序框架

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

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

每個(gè)連接開兩個(gè)線程

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

讀線程邏輯圖：

寫線程邏輯圖：

程序啟動(dòng)時(shí)開一個(gè)總控線程func (h *Hub) Run()

此線程用來(lái)處理客戶端的連接，斷開，訂閱和向訂閱信息的客戶端作相應(yīng)信息的廣播。

邏輯圖如下：

處理注冊(cè)

處理注冊(cè)的邏輯非常簡(jiǎn)單，只是將客戶端的指針?lè)胖糜谥羔榤ap中即可，代碼如下：

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

處理注銷

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

程序邏輯如下：

訂閱處理

訂閱最主要的是訂閱關(guān)系，我們可以用合約.周期作為key，使用客戶端連接上來(lái)的指針集作為value,每當(dāng)有用戶訂閱某內(nèi)容時(shí)，將相應(yīng)內(nèi)容下面的指針集上加上對(duì)應(yīng)此用戶的指針即可，訂閱關(guān)系如下：

推送處理

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

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

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

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

最新評(píng)論