go语言实现聊天服务器的示例代码

在Golang实战开发的过程中，我们经常会遇到一些这样那样的问题，然后要卡好半天，等问题解决了才发现原来一些细节知识点还是没有掌握好。今天golang学习网就整理分享《go语言实现聊天服务器的示例代码》，聊聊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

<p><span style="color: #ff0000"><strong>客户端：</strong></span><br></p>
<p>支持断线重连；支持给特定其他客户端发信息</p>

<pre class="brush:plain;">
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, "

<p>今天带大家了解了go聊天服务器的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~</p>