[Golang] 从零開始写Socket Server(3): 对长、短连接的处理策略(模拟心跳)
? ? 通過前兩章,我們成功是寫出了一套湊合能用的Server和Client,并在二者之間實現了通過協議交流。這么一來,一個簡易的socket通訊框架已經初具雛形了,那么我們接下來做的。就是想辦法讓這個框架更加穩定。茁壯~
? ? 作為一個可能會和非常多Client進行通訊交互的Server。首先要保證的就是整個Server執行狀態的穩定性,因此在和Client建立連接通訊的時候,確保連接的及時斷開非常重要,否則一旦和多個client建立不關閉的長連接,對于server資源的占用是非常可怕的。因此,我們須要針對可能出現的短連接和長連接,設定不同的限制策略。
? ? 針對短連接,我們能夠使用golang中的net包自帶的timeout函數,一共同擁有三個,各自是:
? ? 假設想要給server設置短連接的timeout,我們就能夠這么寫:
netListen, err := net.Listen("tcp", Port)Log("Waiting for clients")for {conn, err := netListen.Accept()if err != nil {continue}conn.SetReadDeadline(time.Now().Add(time.Duration(10) * time.Second))? ??這里的三個函數都是用于設置每次socket連接可以維持的最長時間。一旦超過設置的timeout后,便會在Server端自己主動斷開連接。當中SetReadline, SetWriteline設置的是讀取和寫入的最長持續時間,而SetDeadline則同一時候包括了 SetReadline, SetWriteline兩個函數。
? ? 通過這樣設定。每一個和Server通訊的Client連接時長最長也不會超過10s了~~
? ? 搞定短連接后,接下來就是針對長連接的處理策略了~~
? ? 作為長連接,因為我們往往非常難確定什么時候會中斷連接,因此并不能像處理短連接那樣簡單粗暴的設定一個timeout就能夠搞定,而在Golang的net包中。并沒有針對長連接的函數,因此須要我們自己設計并實現針對長連接的處理策略啦~
? ? 針對socke長連接,常見的做法是在Server和Socket之間設計通訊機制。當兩者之間沒有信息交互時。兩方便會定時發送數據包(心跳),以維持連接狀態。
? ? 這樣的方法是眼下使用相對照較多的做法。可是開銷相對也較大。特別是當Server和多個client保持長連接的時候。并發會比較高,考慮到公司的業務需求,我最后選擇了邏輯相對簡單。開銷相對較小的策略:
? ? 當Server每次收到Client發到的信息之后,便會開始心跳計時,假設在心跳計時結束之前沒有再次收到Client發來的信息。那么便會斷開跟Client的連接。而一旦在設定時間內再次收到Client發來的信息,那么Server便會重置計時器,再次又一次進行心跳計時,直到超時斷開連接為止。
以下就是實現該計時的代碼:
//長連接入口 func handleConnection(conn net.Conn,timeout int) {buffer := make([]byte, 2048)for {n, err := conn.Read(buffer)if err != nil {LogErr(conn.RemoteAddr().String(), " connection error: ", err)return}Data :=(buffer[:n])messnager := make(chan byte)//心跳計時go HeartBeating(conn,messnager,timeout)//檢測每次Client是否有數據傳來go GravelChannel(Data,messnager)Log( "receive data length:",n)Log(conn.RemoteAddr().String(), "receive data string:", string(Data))} }//心跳計時,依據GravelChannel推斷Client是否在設定時間內發來信息 func HeartBeating(conn net.Conn, readerChannel chan byte,timeout int) {select {case fk := <-readerChannel:Log(conn.RemoteAddr().String(), "receive data string:", string(fk))conn.SetDeadline(time.Now().Add(time.Duration(timeout) * time.Second))//conn.SetReadDeadline(time.Now().Add(time.Duration(5) * time.Second))breakcase <-time.After(time.Second*5):Log("It's really weird to get Nothing!!!")conn.Close()}}func GravelChannel(n []byte,mess chan byte){for _ , v := range n{mess <- v}close(mess) }func Log(v ...interface{}) {log.Println(v...) }
? ? 這樣。就能夠成功實現對于長連接的處理了~~,我們能夠這么進行測試:
func sender(conn net.Conn) {for i := 0; i <5; i++ {words:= strconv.Itoa(i)+"This is a test for long conn" conn.Write([]byte(words))time.Sleep(2*time.Second)}fmt.Println("send over")}
? ? 能夠發現。Sender函數中time.Sleep堵塞的時間設定的比Server中的timeout短的時候,Client端的信息能夠自由的發送到循環結束,而當我們設定Sender函數的堵塞時間較長時,就僅僅能發出第一次循環的信息。
我已經把SocketServer系列的代碼整合到了一起。公布到了我個人的github上:點擊鏈接, 希望大家有興趣的能夠學習star一下~
轉載于:https://www.cnblogs.com/blfbuaa/p/7073360.html
總結
以上是生活随笔為你收集整理的[Golang] 从零開始写Socket Server(3): 对长、短连接的处理策略(模拟心跳)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 在哈尔滨上雅思一对一大概多少钱?四级46
- 下一篇: lintcode-34-N皇后问题 II