用go實現併發聊天室

lightTrace發表於2018-04-09

還是老一套,go在網路模組的開發確實是便捷無比,使用goroutine的特性簡直是如魚得水,準備簡單的使用go做一個cs的聊天室,整體程式碼不到一千行,核心程式碼可能就一兩百行,非常通俗易懂,能幫助我們學習goroutine的特性和感受tcp。

原始碼:https://github.com/lightTrace/go-learn/tree/chatroom

效果圖
服務端:
服務端

客戶端jack:
這裡寫圖片描述

客戶端pony:
這裡寫圖片描述

一 服務端
server.go
首先主程式埠都是從main開始:

func main() {

    if len(os.Args) == 2 {
        uData = getAllUser(dataFileName) // 使用者登入、註冊資料初始化

        StartServer(os.Args[1]) // 啟動服務端,命令列的第二個引數是埠號

    } else {

        fmt.Println("輸入錯誤!")
    }

}

可以看到啟動服務端有兩步操作:getAllUser獲取使用者和StartServer啟動服務端

獲取使用者就是開啟本地檔案讀取的過程,關鍵是StartServer,StartServer首先createTCP創造一個tcp連線。
然後dataSent啟動伺服器廣播協程:用於廣播客戶端和服務端互動的所有資訊並中轉傳送給相應客戶端,通過“-”分割解析判斷;
intoManager是啟動管理模組,接收輸入的資訊來判斷是管理員註冊還是登入,登入成功後開啟管理員管理模組admimManager
connManager啟動管理協程,一旦connection通道有連線斷開就會通知

然後conn, err := l.Accept()啟動無限迴圈監聽連線,然後分別啟動recv接收資料和sentData傳送資料的協程;

最後開啟userAuth驗證協程,根據flag判斷是註冊還是登入

二 客戶端
client.go

client就更簡單了,建立連線後直接開啟資料接收和傳送協程,由使用者輸入資料判斷進行具體的操作

相關文章