go語言實現聊天室初級版
聊天室分爲服務端和客戶端。服務端主動開啓監聽等待客戶端連接,客戶端向服務端發送連接請求,建立連接,發送消息,實現與服務端的通信。下面使用go語言實現一個簡單版的聊天室。第一部分爲服務端代碼,第二部分爲客戶端代碼。
一、服務端代碼
僞代碼:
1、服務端開啓監聽
2、等待客戶端連接
3、接收客戶端連接
4、處理客戶端連接,輸出消息
//chatroom server
package main
//導入包
import (
"fmt"
"net"
)
//定義函數checkError,用來錯誤處理
func checkError(err error) {
if err != nil {
panic(err)
}
}
//定義一個函數,專門負責接收信息的協程
func processInfo(conn net.Conn) {
//定義一個變量用來存儲數據
buf := make([]byte, 1024)
//最後需要關閉連接
defer conn.Close()
//無限循環讀取接收到字節數,直到讀取完後退出循環
for {
//讀取接收到的字節數
numOfBytes, err := conn.Read(buf)
if err != nil {
break
}
//如果接收的字節數不爲0,說明有消息發過來,並打印出消息
if numOfBytes != 0 {
fmt.Printf("Has received message: %s", string(buf))
}
}
}
//主函數入口
func main() {
//開啓監聽,監聽協議爲tcp,監聽地址及端口爲127.0.0.1:8888
listen_socket, err := net.Listen("tcp","127.0.0.1:8888")
checkError(err)
//最後關閉套接字連接
defer listen_socket.Close()
//沒有消息進來時默認輸出:chat room Server is waiting...
fmt.Println("chat room Server is waiting... ")
//接收客戶端連接
for {
conn, err := listen_socket.Accept()
checkError(err)
//如果有客戶端連接,則打開一個協程處理
go processInfo(conn)
}
}
執行go build server.go或者go run server.go將會輸出:chat room Server is waiting... ,等待客戶端連接
二、客戶端代碼
僞代碼
1、客戶端向服務端發送連接請求
2、客戶端向服務端發送信息
// chat room client
package main
//導入包
import (
"fmt"
"net"
)
func checkError(err error) {
if err != nil {
panic(err)
}
}
func main() {
//使用Dial函數和服務端建立連接
conn, err := net.Dial("tcp","127.0.0.1:8888")
checkError(err)
//最後退出時關閉連接
defer conn.Close()
//向服務端發送信息
conn.Write([]byte("Now connection.success!"))
fmt.Println("Has sent the message!")
}
執行go run client.go或者go build client.go將會輸出:Has sent the message!
此時客戶端已向服務端發送信息:Now connection.success
我們再查看服務端的輸出終端,此時服務端輸出如下內容,那就表示客戶端與服務端通信成功。
chat room Server is waiting...
Has received message: Now connection.success!
本次實驗實現了簡單的聊天室,我們會發現客戶端僅僅只能發送一次消息,這在現實中是不能忍受的。所以我將在下一篇文章《go語言實現聊天室中級版》實現功能:客戶端可以多次發送消息至服務端
參考鏈接:
[1].https://studygolang.com/pkgdoc ->(go語言標準庫中文文檔)