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 == "ls\n" {
     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)
 }
}

客户端：

支持断线重连；支持给特定其他客户端发信息

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, "<-")
   if len(msgs) < 2 {
     fmt.Println("发送的姿势不正确，应该像这样 1<-给1号发送消息\n")
     continue
   }

index, err := strconv.Atoi(msgs[0])
   if err != nil {
     fmt.Println("发送的姿势不正确，应该像这样 1<-给1号发送消息\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()
   }
 }()
}

来源：https://segmentfault.com/a/1190000015950984