go語言實現聊天室簡單版

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語言標準庫中文文檔)