go语言实现聊天室

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!