go語言實(shí)現(xiàn)聊天室簡單版

go語言實(shí)現(xiàn)聊天室初級版

聊天室分為服務(wù)端和客戶端。服務(wù)端主動開啟監(jiān)聽等待客戶端連接，客戶端向服務(wù)端發(fā)送連接請求，建立連接，發(fā)送消息，實(shí)現(xiàn)與服務(wù)端的通信。下面使用go語言實(shí)現(xiàn)一個簡單版的聊天室。第一部分為服務(wù)端代碼，第二部分為客戶端代碼。

一、服務(wù)端代碼

偽代碼：
1、服務(wù)端開啟監(jiān)聽
2、等待客戶端連接
3、接收客戶端連接
4、處理客戶端連接，輸出消息

//chatroom server
package main

//導(dǎo)入包
    import (
        "fmt"
        "net"
    )

//定義函數(shù)checkError，用來錯誤處理
func checkError(err error)  {
    if err != nil {
        panic(err)
    }
}

//定義一個函數(shù)，專門負(fù)責(zé)接收信息的協(xié)程
func processInfo(conn net.Conn)  {
//定義一個變量用來存儲數(shù)據(jù)
    buf := make([]byte, 1024)
    //最后需要關(guān)閉連接
    defer conn.Close()

    //無限循環(huán)讀取接收到字節(jié)數(shù)，直到讀取完后退出循環(huán)
    for {
        //讀取接收到的字節(jié)數(shù)
        numOfBytes, err := conn.Read(buf)
        if err != nil {
            break
        }

        //如果接收的字節(jié)數(shù)不為0，說明有消息發(fā)過來，并打印出消息
        if numOfBytes != 0 {
            fmt.Printf("Has received message: %s", string(buf))
        }

    }
}

//主函數(shù)入口
    func main()  {
        //開啟監(jiān)聽，監(jiān)聽協(xié)議為tcp,監(jiān)聽地址及端口為127.0.0.1:8888
        listen_socket, err := net.Listen("tcp","127.0.0.1:8888")
        checkError(err)
        //最后關(guān)閉套接字連接
        defer listen_socket.Close()
        //沒有消息進(jìn)來時默認(rèn)輸出:chat room Server is waiting...
        fmt.Println("chat room Server is waiting... ")

        //接收客戶端連接
        for {
            conn, err := listen_socket.Accept()
            checkError(err)

            //如果有客戶端連接，則打開一個協(xié)程處理
            go processInfo(conn)

        }
    }

執(zhí)行g(shù)o build server.go或者go run server.go將會輸出：chat room Server is waiting... ，等待客戶端連接

二、客戶端代碼

偽代碼
1、客戶端向服務(wù)端發(fā)送連接請求
2、客戶端向服務(wù)端發(fā)送信息

// chat room client
package main

//導(dǎo)入包
    import (
        "fmt"
        "net"
    )

func checkError(err error)  {
    if err != nil {
        panic(err)
    }
}

func main()  {
    //使用Dial函數(shù)和服務(wù)端建立連接
    conn, err := net.Dial("tcp","127.0.0.1:8888")
    checkError(err)
    //最后退出時關(guān)閉連接
    defer conn.Close()

    //向服務(wù)端發(fā)送信息
    conn.Write([]byte("Now connection.success!"))

    fmt.Println("Has sent the message!")
}

執(zhí)行g(shù)o run client.go或者go build client.go將會輸出：Has sent the message!
此時客戶端已向服務(wù)端發(fā)送信息：Now connection.success

我們再查看服務(wù)端的輸出終端，此時服務(wù)端輸出如下內(nèi)容，那就表示客戶端與服務(wù)端通信成功。
chat room Server is waiting...
Has received message: Now connection.success!

本次實(shí)驗(yàn)實(shí)現(xiàn)了簡單的聊天室，我們會發(fā)現(xiàn)客戶端僅僅只能發(fā)送一次消息，這在現(xiàn)實(shí)中是不能忍受的。所以我將在下一篇文章《go語言實(shí)現(xiàn)聊天室中級版》實(shí)現(xiàn)功能：客戶端可以多次發(fā)送消息至服務(wù)端

參考鏈接：
[1].https://studygolang.com/pkgdoc ->(go語言標(biāo)準(zhǔn)庫中文文檔)