還是老一套,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就更簡單了,建立串連後直接開啟資料接收和發送協程,由使用者輸入資料判斷進行具體的操作