go語言實現(xiàn)聊天服務器的示例代碼

更新時間：2018年08月10日 08:25:28 作者：Y_xx

這篇文章主要介紹了go語言實現(xiàn)聊天服務器的示例代碼，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧

看了兩天 go 語言，是時候練練手了。

go 的 routine（例程）和 chan（通道）簡直是神器，實現(xiàn)多線程（在 go 里準確的來說是多例程）簡直不要太輕松。

于是動手碼了一個傻瓜版的黑框聊天器。

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號客戶端的消息\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, "接入服務")

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

  // 啟動發(fā)送
  go send(addr, conn)

  // 發(fā)送當前用戶信息
  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ù)格式不正確，請聯(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
  }

  // 啟動消息調度器

  defer listener.Close()

  // 啟動連接監(jiān)聽
  for {
    conn, err := listener.Accept()
    if err != nil {
      fmt.Println(err)
      continue
    }

    go serve(&conn)
  }
}

客戶端：

支持斷線重連；支持給特定其他客戶端發(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("結束")
        (*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("等待服務器開啟中")
  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ā)送的姿勢不正確，應該像這樣 1<-給1號發(fā)送消息\n")
      continue
    }

    index, err := strconv.Atoi(msgs[0])
    if err != nil {
      fmt.Println("發(fā)送的姿勢不正確，應該像這樣 1<-給1號發(fā)送消息\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()
    }
  }()
}

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章:

快速掌握Go 語言 HTTP 標準庫的實現(xiàn)方法
基于HTTP構建的服務標準模型包括兩個端，客戶端(Client)和服務端(Server)，這篇文章主要介紹了Go 語言HTTP標準庫的實現(xiàn)方法,需要的朋友可以參考下
2022-07-07
Go語言指針用法詳解
Go指針和C指針在許多方面非常相似，但其中也有一些不同。本文詳細講解了Go語言指針的用法，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2022-07-07
Golang使用bcrypt實現(xiàn)密碼加密和校驗的操作代碼
bcrypt可以用于數(shù)據(jù)庫中的用戶密碼保存,相比md5而言更加的安全可靠,這篇文章主要介紹了Golang使用bcrypt實現(xiàn)密碼加密和校驗的操作代碼,需要的朋友可以參考下
2024-05-05
聊聊Golang中很好用的viper配置模塊
這篇文章主要介紹了Golang中很好用的viper配置模塊用法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-12-12
go語言生成隨機數(shù)和隨機字符串的實現(xiàn)方法
隨機數(shù)在很多時候都可以用到，尤其是登錄時，本文就詳細的介紹一下go語言生成隨機數(shù)和隨機字符串的實現(xiàn)方法，具有一定的參考價值，感興趣的可以了解一下
2021-12-12
golang中channel+error來做異步錯誤處理有多香
官方推薦golang中錯誤處理當做值處理，既然是值那就可以在channel中傳輸，這篇文章主要介紹了golang 錯誤處理channel+error真的香,需要的朋友可以參考下
2023-01-01
Go排序算法通用qsort函數(shù)使用示例
這篇文章主要為大家介紹了Go排序算法通用qsort函數(shù)使用示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪
2023-11-11
Go源碼字符串規(guī)范檢查lint工具strchecker使用詳解
這篇文章主要為大家介紹了Go源碼字符串規(guī)范檢查lint工具strchecker使用詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪
2022-06-06
深入理解Golang中指針的用途與技巧
在 Go 語言中，指針是一種重要的概念，了解和正確使用指非常關鍵，因此本文小編就來和大家講講Golang 中指針的概念與用法，希望對大家有所幫助
2023-05-05
Go語言中的通道chan使用指南
Go語言的通道chan是實現(xiàn)并發(fā)編程的關鍵工具,主要用于goroutine之間的數(shù)據(jù)傳輸,本文主要介紹了通道的基本操作如創(chuàng)建、發(fā)送、接收和關閉數(shù)據(jù),以及使用select語句進行多路復用和超時控制,感興趣的可以了解一下
2024-10-10