Golang一百行代碼實現簡單的併發聊天室

> 項目介紹：Golang100行代碼實現高併發聊天室，其中實現的功能有：上下線廣播，私聊，用戶改名，超時強踢，在線用戶檢測等

在開始項目前，我們需要理解貫穿這整個項目的兩個重要變量，若能理解這兩個變量的使用，那麼併發聊天室項目會變得手到擒來。第一個是onlinemap全局map，第二個是Message全局channel。

取名爲onlinemap的全局map類型爲map[string][client],這個全局字典是用來存儲當前在此聊天室的用戶的，key值是string類型，爲用戶的ip地址+Port端口，對應的value值爲一個結構體，結構體內有此用戶的姓名，地址和管道（用來給每一個用戶傳輸信息，服務於Message全局通道）

取名爲Message的全局channel也貫穿在整段代碼中，向其中傳送數據時，Message會在另一個go程裏向其他每一個在線用戶的管道中發送內容，隨後在另一個go程裏每一個用戶的管道會向對應用戶轉發內容。如此可以實現上下線廣播，羣聊的功能。而每一個用戶私有的管道可以實現私聊功能。

這個圖詳細闡述了這段代碼的工作流程。

理解以上內容 下面我們再來看代碼，就會很輕鬆，如果還是一頭霧水也不要着急，小編會在下面每一行代碼都加上精準通俗的註釋，不多說，上代碼。

package main
import (
    "net"
    "fmt"
    "strings"
    "time"
)
//定義的此結構體爲全局map的value值，包括每一個用戶的姓名，ip地址和私人管道
type client struct {
    name string
    addr string
    C    chan string
}
/*這個函數是將私人管道中的內容發送給用戶，配合全局管道Message使用可以實現廣播的功能，
單獨使用可以實現私聊的功能*/
func writemsg2client(clinet client,conn net.Conn) {
    for m := range clinet.C {
        conn.Write([]byte(m + "\n"))
    }
}
//這只是一個封裝好用來統一(發送信息格式)的小函數，不用在意
func makemsg(name string, addr string, s string) string {
    return "[" + addr + "]" + name + s
}
//每一個進入聊天室的用戶都將啓動一個handleconn的go程來處理事件
func handleconn(conn net.Conn) {
    defer conn.Close()
/*用戶連接進來以後要初始化全局map，把自己的信息加入到字典裏，相當於進到聊天室裏之前要登
記一下個人信息，注意姓名初始爲ip地址。*/
    addr := conn.RemoteAddr().String()
    fmt.Printf("用戶%s進入了房間\n", addr)
    client := client{addr, addr, make(chan string)}
    //在這裏啓動子go程，功能上面已經提及
    go writemsg2client(client,conn)
    onlinemap[addr] = client
    //登錄進來一切準備就緒後就給所有人廣播上線信息啦
    Message <- makemsg(client.name, addr, "login")
    //下面這三個變量服務於下面一些小功能
    var haschat=make(chan bool)
    var ifquit=make(chan bool)
    var flag bool
    //從這單獨開啓一個go程來讀取用戶輸入的信息
    go func() {
        buf:=make([]byte,4096)
        for  {
            n,_:=conn.Read(buf)
            if n==0 {
                fmt.Printf("%s離開了房間\n",client.name)
                ifquit<-true
                return
            }
            //改名功能的實現
            if string(buf[:7])=="Rename|" {
                client.name=strings.Split(string(buf[:n-1]),"|")[1]
                onlinemap[addr]=client
                conn.Write([]byte("rename success\n"))
            }else if string(buf[:n-1])=="/who"{
            //查詢在線用戶信息的功能
                for _,s:=range onlinemap{
                    conn.Write([]byte(s.name+"online\n"))
                }
            }else if string(buf[:2])=="m|"&&strings.Count(string(buf[:n]),"|")==2 {
/*私聊功能的實現，其實私聊功能就是跳過了往全局Message裏傳輸信息，
改爲直接向私人管道里傳輸信息*/
                flag=false
                slice:=strings.Split(string(buf[:n-1]),"|")
                for _,a:=range onlinemap{
                //遍歷所有在線用戶，向指定的用戶管道中發送信息
                    if a.name==slice[1]{
                        flag=true
                        a.C<-makemsg(client.name,addr,slice[2])
                        conn.Write([]byte("send success"))
                    }
                }
                if flag {
                    conn.Write([]byte("no such man or not online"))
                }
            } else {
                Message<-makemsg(client.name,addr,string(buf[:n-1]))
            }
            haschat<-true
        }
    }()
    for  {
        select {
        case <-haschat:
        //超時強踢
        case <-time.After(time.Minute*3):
            delete(onlinemap,addr)
            Message<-makemsg(client.name,addr,"out time to leave")
            close(client.C)
            return
        case <-ifquit:
        //退出處理
            delete(onlinemap,addr)
            Message<-makemsg(client.name,addr,"out time to leave")
            close(client.C)
            return
        }
    }
}
//這個函數用來將全局Message中的內容全部塞到私人管道C裏，實現上下線廣播和羣聊的功能
func Manager() {
    for {
        msg := <-Message
        for _, s := range onlinemap {
            s.C <- msg
        }
    }
}
var Message = make(chan string)
var onlinemap map[string]client = make(map[string]client)
//主函數
func main() {
    listener, _ := net.Listen("tcp", "127.0.0.1:6666")
    defer listener.Close()
    //提前開啓全局Message的go程，防止被阻塞
    go Manager()
    for {
        conn, err := listener.Accept()
        if err != nil {
            fmt.Println("accept err", err)
            continue
        }
        //每一個連接進來的用戶都會被分配進入一個子go程，用來處理上面我們提到的各種功能
        go handleconn(conn)
    }
}

以上就是一個簡單的高併發聊天室了，依託於go語言的強大，去掉註釋只剩下不到一百行，雖然功能簡單，但是涉及到channel,socket,select,map,string及go的使用，有利於此階段在學的小夥伴們學習交流，大家有什麼疑問或者想法可以在下面給我留言哦。