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 == "lsn" {
      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)
  }
}

								 
 1/2    1 2 下一页 尾页