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语言标准库中文文档)