go語言實(shí)現(xiàn)聊天服務(wù)器的示例代碼
看了兩天 go 語言,是時(shí)候練練手了。
go 的 routine(例程) 和 chan(通道) 簡直是神器,實(shí)現(xiàn)多線程(在 go 里準(zhǔn)確的來說是 多例程)簡直不要太輕松。
于是動(dòng)手碼了一個(gè)傻瓜版的黑框聊天器。
server 端:
監(jiān)聽 TCP 連接;支持自定義客戶端命令;支持消息分發(fā);理論上支持廣播;...
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 += "發(fā)送消息,可使用 1<-這是給1號(hào)客戶端的消息\n(請(qǐng)使用英文以獲取最佳體驗(yàn))\n" senderChan <- cs time.Sleep(time.Millisecond * 300) senderChan <- ls // 發(fā)送格式化的列表 fmt.Println("已發(fā)送“登錄用戶信息”", addr) } else{ fmt.Println("客戶端接受通道不存在", addr) } } func serve (conn *net.Conn){ connect := *conn addr := connect.RemoteAddr().String() fmt.Println(addr, "接入服務(wù)") senderChan := make(chan string, 3) clientSenders[addr] = senderChan // 啟動(dòng)發(fā)送 go send(addr, conn) // 發(fā)送當(dāng)前用戶信息 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 } // 提取數(shù)據(jù) msgs := strings.Split(msg, "<-") if len(msg) < 2{ senderChan <- string("數(shù)據(jù)格式不正確,請(qǐng)聯(lián)系開發(fā)者") 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 } // 啟動(dòng)消息調(diào)度器 defer listener.Close() // 啟動(dòng)連接監(jiān)聽 for { conn, err := listener.Accept() if err != nil { fmt.Println(err) continue } go serve(&conn) } }
客戶端:
支持?jǐn)嗑€重連;支持給特定其他客戶端發(fā)信息
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("結(jié)束") (*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: // 解析客戶端列表數(shù)據(jù) addrs = strings.Split(msg, "\n") fmt.Println("已接收客戶端列表。\n") break default: fmt.Print(msg) break } } } func connectServer(){ addr := "192.168.99.236:8080" fmt.Println("等待服務(wù)器開啟中") 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("發(fā)送的姿勢(shì)不正確,應(yīng)該像這樣 1<-給1號(hào)發(fā)送消息\n") continue } index, err := strconv.Atoi(msgs[0]) if err != nil { fmt.Println("發(fā)送的姿勢(shì)不正確,應(yīng)該像這樣 1<-給1號(hào)發(fā)送消息\n") continue } if len(addrs) <= index { fmt.Println("不存在第" + strconv.Itoa(index) + "個(gè)客戶端\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() } }() }
以上就是本文的全部內(nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
go實(shí)現(xiàn)grpc四種數(shù)據(jù)流模式
這篇文章主要為大家介紹了go實(shí)現(xiàn)grpc四種數(shù)據(jù)流模式,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步早日升職加薪2022-04-04golang 檢查網(wǎng)絡(luò)狀態(tài)是否正常的方法
今天小編就為大家分享一篇golang 檢查網(wǎng)絡(luò)狀態(tài)是否正常的方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-07-07Golang微服務(wù)框架Kratos實(shí)現(xiàn)分布式任務(wù)隊(duì)列Asynq的方法詳解
任務(wù)隊(duì)列(Task Queue) 一般用于跨線程或跨計(jì)算機(jī)分配工作的一種機(jī)制,在Golang語言里面,我們有像Asynq和Machinery這樣的類似于Celery的分布式任務(wù)隊(duì)列,本文就給大家詳細(xì)介紹一下Golang微服務(wù)框架Kratos實(shí)現(xiàn)分布式任務(wù)隊(duì)列Asynq的方法,需要的朋友可以參考下2023-09-09實(shí)時(shí)通信的服務(wù)器推送機(jī)制 EventSource(SSE) 簡介附go實(shí)現(xiàn)示例代碼
EventSource是一種非常有用的 API,適用于許多實(shí)時(shí)應(yīng)用場景,它提供了一種簡單而可靠的方式來建立服務(wù)器推送連接,并實(shí)現(xiàn)實(shí)時(shí)更新和通知,這篇文章主要介紹了實(shí)時(shí)通信的服務(wù)器推送機(jī)制 EventSource(SSE)簡介附go實(shí)現(xiàn)示例,需要的朋友可以參考下2024-03-03從并發(fā)到并行解析Go語言中的sync.WaitGroup
Go?語言提供了許多工具和機(jī)制來實(shí)現(xiàn)并發(fā)編程,其中之一就是?sync.WaitGroup。本文就來深入討論?sync.WaitGroup,探索其工作原理和在實(shí)際應(yīng)用中的使用方法吧2023-05-05Go利用反射reflect實(shí)現(xiàn)獲取接口變量信息
反射是通過實(shí)體對(duì)象獲取反射對(duì)象(Value、Type),然后可以操作相應(yīng)的方法。本文將利用Go語言中的反射reflect實(shí)現(xiàn)獲取接口變量信息,需要的可以參考一下2022-05-05