go語言實現聊天室

聊天室分為服務端和客戶端。第一部分為服務端程式碼，第二部分為客戶端程式碼。

一、服務端程式碼

//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()  {
    //開啟監聽
    listen_socket, err := net.Listen("tcp","127.0.0.1:8888")
    checkError(err)
    defer listen_socket.Close()

    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... ，等待客戶端連線

二、客戶端程式碼

// chat room client
package main

import (
    "fmt"
    "net"
)

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

func main()  {
    //連線服務端的資訊
    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!