简易聊天室的开发---GO语言

需求

• 服务器接收客户端的信息
• 接收完以后将客户端的信息发送到所有的客户端上
• 客户端使用/quit退出聊天

项目结构

E:\project\go\mychat>tree /f
卷 文档 的文件夹 PATH 列表
卷序列号为 000B-2502
E:.
├─bin
│      client.exe
│      server.exe
│
├─pkg
└─src
    ├─client  --客户端
    │      client.go
    │
    └─server  --服务端
            server.go

go环境变量：

C:\Users\zhanglf>go env
set GOARCH=amd64
set GOBIN=
set GOEXE=.exe
set GOHOSTARCH=amd64
set GOHOSTOS=windows
set GOOS=windows
set GOPATH=E:\project\go\mychat
set GORACE=
set GOROOT=C:\Go
set GOTOOLDIR=C:\Go\pkg\tool\windows_amd64
set CC=gcc
set GOGCCFLAGS=-m64 -mthreads -fmessage-length=0
set CXX=g++
set CGO_ENABLED=1

实现

//server.go

package main

import (

    "fmt"

    "log"

    "net"

    "os"

    "strconv"

)

//

//错误检查

//

func checkError(err error, info string) (res bool) {

    if err != nil {

        fmt.Fprintln(os.Stderr, info+" : "+err.Error())

        return false

    }

    return true

}

//

//服务器端接收数据线程

//参数：

//      数据连接 conn

//      通讯通道 messages

//

func HandleClient(conn net.Conn, messages chan string) {

    buf := make([]byte, 1024)

    for {

        length, err := conn.Read(buf)

        if checkError(err, "Connection") == false {

            conn.Close()

            break

        }

        receiveStr := string(buf[0:length])

        messages <- receiveStr

    }

}

//

//服务器发送数据的线程

//

//参数

//      连接字典 conns

//      数据通道 messages

//

func Broadcast(conns *map[string]net.Conn, messages chan string) {

    for {

        msg := <-messages

        Log(msg)

        for key, value := range *conns {

            Log("broadcast to", key)

            _, err := value.Write([]byte(msg))

            if err != nil {

                Log(err.Error())

                delete(*conns, key)

            }

        }

    }

}

//

//启动服务器

//参数

//  端口 port

//

func StartServer(port string) {

    service := ":" + port

    tcpAddr, err := net.ResolveTCPAddr("tcp4", service)

    if res := checkError(err, "ResolveTCPAddr"); !res {

        os.Exit(1)

    }

    listener, err := net.ListenTCP("tcp", tcpAddr)

    if res := checkError(err, "ListenTCP"); !res {

        os.Exit(1)

    }

    conns := make(map[string]net.Conn)

    messages := make(chan string, 10)

    //启动服务器广播线程

    go Broadcast(&conns, messages)

    for {

        Log("Listening ...")

        conn, err := listener.Accept()

        if res := checkError(err, "Accept"); !res {

            continue

        }

        Log(conn.RemoteAddr().String(), "connected in")

        conns[conn.RemoteAddr().String()] = conn

        //启动一个新线程

        go HandleClient(conn, messages)

    }

}

//

//日志打印函数

//参数说明：

//    输出参数 v

//

//

func Log(v ...interface{}) {

    log.Println(v...)

}

//

//主程序

//参数说明：

//

//

//

func main() {

    if len(os.Args) != 2 {

        fmt.Fprintln(os.Stderr, "Usage: server PortNumber")

        fmt.Fprintln(os.Stdout, "e.g. server 9090")

        os.Exit(1)

    }

    _, err := strconv.Atoi(os.Args[1])

    if err != nil {

        fmt.Fprintln(os.Stderr, "请输入数字格式的端口号")

        os.Exit(1)

    }

    StartServer(os.Args[1])

}

//client.go

package main

import (

    "bufio"

    "fmt"

    "net"

    "os"

    "strconv"

)

//

//错误检查

//

func checkError(err error, info string) (res bool) {

    if err != nil {

        fmt.Println(info + "  " + err.Error())

        return false

    }

    return true

}

//

//客户端发送线程

//参数

//      发送连接 conn

//

func chatSend(conn net.Conn) {

    reader := bufio.NewReader(os.Stdin)

    username := conn.LocalAddr().String()

    for {

        input, _, _ := reader.ReadLine()

        if string(input) == "/quit" {

            fmt.Println("ByeBye...")

            conn.Close()

            os.Exit(0)

        }

        if len(input) == 0 {

            continue

        }

        _, err := conn.Write([]byte(username + " Say> " + string(input)))

        if err != nil {

            fmt.Println(err.Error())

            conn.Close()

            break

        }

    }

}

//

//客户端启动函数

//参数

//      远程ip地址和端口 tcpaddr

//

func StartClient(ip, port string) {

    server := ip + ":" + port

    tcpAddr, err := net.ResolveTCPAddr("tcp4", server)

    if res := checkError(err, "ResolveTCPAddr"); !res {

        os.Exit(1)

    }

    conn, err := net.DialTCP("tcp", nil, tcpAddr)

    if res := checkError(err, "DialTCP"); !res {

        fmt.Println("could not connect to server:", server)

        os.Exit(1)

    }

    //启动客户端发送线程

    go chatSend(conn)

    //开始客户端轮训

    buf := make([]byte, 1024)

    for {

        length, err := conn.Read(buf)

        if checkError(err, "Connection") == false {

            conn.Close()

            fmt.Println("Server is dead ...ByeBye")

            os.Exit(0)

        }

        fmt.Println(string(buf[0:length]))

    }

}

//

//主程序

//

//参数说明：

//

//

func main() {

    if len(os.Args) != 3 {

        fmt.Fprintln(os.Stderr, "Usage: client ip port")

        fmt.Fprintln(os.Stdout, "e.g. client 127.0.0.1 9090")

        os.Exit(1)

    }

    ip := net.ParseIP(os.Args[1])

    if ip == nil {

        fmt.Fprintln(os.Stderr, "Please input a valid ip address")

        os.Exit(1)

    }

    _, err := strconv.Atoi(os.Args[2])

    if err != nil {

        fmt.Fprintln(os.Stderr, "请输入数字格式的端口号")

        os.Exit(1)

    }

    StartClient(os.Args[1], os.Args[2])

}

运行

到此为止，一个简单的实时聊天系统实现了，目前只是实现了群发的功能。