go语言实现聊天服务器

看了两天 go 语言，是时候练练手了。
go 的 routine（例程） 和 chan（通道） 简直是神器，实现多线程（在 go 里准确的来说是 多例程）简直不要太轻松。

于是动手码了一个傻瓜版的黑框聊天器。

server 端：
监听 TCP 连接；支持自定义客户端命令；支持消息分发；理论上支持广播；...

package main

import (
    "fmt"
    "net"
    "io"
    "strconv"
    "time"
    "strings"
)

const (
    NORMAL_MESSAGE = iota
    LIST_MESSAGE
)

var clientSenders = make(map[string] chan string)

func send (addr string, conn *net.Conn){
    senderChan := clientSenders[addr]
    for s := range senderChan{
        (*conn).Write([]byte(s))
    }
}

func sendUsersInfo(addr string){
    senderChan := clientSenders[addr]
    if nil != senderChan{
        ls := strconv.Itoa(LIST_MESSAGE)
        cs := strconv.Itoa(NORMAL_MESSAGE) + "已登录客户端列表：\n"
        i := 1
        for k := range clientSenders{
            a := ""
            if k == addr {
                a = "（我）"
            }
            cs = cs + strconv.Itoa(i) + "）"  + k + a + "\n"
            ls += k + "\n"
            i ++
        }
        cs += "发送消息，可使用 1<-这是给1号客户端的消息\n（请使用英文以获取最佳体验）\n"

        senderChan <- cs
        time.Sleep(time.Millisecond * 300)
        senderChan <- ls

        // 发送格式化的列表

        fmt.Println("已发送“登录用户信息”", addr)
    } else{
        fmt.Println("客户端接受通道不存在", addr)
    }
}

func serve (conn *net.Conn){
    connect := *conn

    addr := connect.RemoteAddr().String()

    fmt.Println(addr, "接入服务")

    senderChan := make(chan string, 3)
    clientSenders[addr] = senderChan

    // 启动发送
    go send(addr, conn)

    // 发送当前用户信息
    go sendUsersInfo(addr)

    buff := make([]byte, 10240)
    for {
        n, err := connect.Read(buff)
        if err != nil {
            if err == io.EOF {
                fmt.Println("客户端断开链接，", addr)
                delete(clientSenders, addr)
                return
            } else{
                fmt.Println(err)
            }
        }

        msg := string(buff[:n])

        // 刷新客户端列表
        if msg == "ls\n" {
            go sendUsersInfo(addr)
            continue
        }

        // 提取数据
        msgs := strings.Split(msg, "<-")
        if len(msg) < 2{
            senderChan <- string("数据格式不正确，请联系开发者")
            continue
        }

        aimAddr := msgs[0]
        aimSender := clientSenders[aimAddr]
        if aimSender == nil {
            senderChan <- string("客户端已下线，使用 ls 命令获取最新的客户端列表")
            continue
        }

        aimSender <- strconv.Itoa(NORMAL_MESSAGE) + "[from:" + addr + "]:" + strings.Join(msgs[1:], "<-")
    }
}

func main(){
    addr := ":8080"
    listener, err := net.Listen("tcp", addr)
    if err != nil{
        fmt.Println(err)
        return
    }

    // 启动消息调度器

    defer listener.Close()

    // 启动连接监听
    for {
        conn, err := listener.Accept()
        if err != nil {
            fmt.Println(err)
            continue
        }

        go serve(&conn)
    }
}

客户端：
支持断线重连；支持给特定其他客户端发信息

package main

import (
    "net"
    "fmt"
    "io"
    "os"
    "bufio"
    "sync"
    "time"
    "strings"
    "strconv"
)


var conn *net.Conn
var addrs []string

const (
    NORMAL_MESSAGE = iota
    LIST_MESSAGE
)

func read(conn2 *net.Conn){
    defer func() {
        fmt.Println("尝试重连")
        go connectServer()
    }()

    connect := *conn2
    buff := make([]byte, 20140)
    for {
        n, err := connect.Read(buff)
        if err != nil {
            if err == io.EOF{
                fmt.Println("结束")
                (*conn2).Close()
                conn = nil
                return
            } else{
                fmt.Println(err)
            }
        }

        msg := string(buff[:n])
        t, err := strconv.Atoi(string(msg[0]))
        msg = msg[1:]

        switch t {
        case NORMAL_MESSAGE:
            fmt.Print(msg)
            break
        case LIST_MESSAGE:
            // 解析客户端列表数据
            addrs = strings.Split(msg, "\n")
            fmt.Println("已接收客户端列表。\n")
            break
        default:
            fmt.Print(msg)
            break
        }
    }
}

func connectServer(){
    addr := "192.168.99.236:8080"
    fmt.Println("等待服务器开启中")
    conn2, err := net.Dial("tcp", addr)
    if err != nil {
        fmt.Print(err)
        fmt.Println("连接失败，10s后尝试")
        time.Sleep(10 * time.Second)
        go connectServer()
        return
    }

    fmt.Println("已连接")

    conn = &conn2
    go read(&conn2)
}

func send (){
    inputReader := bufio.NewReader(os.Stdout)
    for {
        input, err := inputReader.ReadString('\n')
        if err != nil {
            if err == io.EOF{
                return
            } else{
                fmt.Println(err)
            }
        }

        if input == "ls\n" {
            (*conn).Write([]byte(input))
            continue
        }

        msgs := strings.Split(input, "<-")
        if len(msgs) < 2 {
            fmt.Println("发送的姿势不正确，应该像这样 1<-给1号发送消息\n")
            continue
        }

        index, err := strconv.Atoi(msgs[0])
        if err != nil {
            fmt.Println("发送的姿势不正确，应该像这样 1<-给1号发送消息\n")
            continue
        }

        if len(addrs) <= index {
            fmt.Println("不存在第" + strconv.Itoa(index) + "个客户端\n")
            continue
        }

        addr := addrs[index-1]

        input = addr + "<-" + strings.Join(msgs[1:], "<-")

        if nil != conn {
            (*conn).Write([]byte(input))
        }
    }
}

func main (){
    var wg sync.WaitGroup
    wg.Add(2)
    go connectServer()
    go send()
    wg.Wait()

    defer func() {
        if nil != conn {
            (*conn).Close()
        }
    }()
}

有疑问加站长微信联系（非本文作者）