服务器之家

服务器之家 > 正文

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

时间:2020-05-17 12:09     来源/作者:Y_xx

看了两天 go 语言,是时候练练手了。

go 的 routine(例程) 和 chan(通道) 简直是神器,实现多线程(在 go 里准确的来说是 多例程)简直不要太轻松。

于是动手码了一个傻瓜版的黑框聊天器。

server 端:

监听 TCP 连接;支持自定义客户端命令;支持消息分发;理论上支持广播;...

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
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)
  }
}

客户端:

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

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
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

相关文章

热门资讯

歪歪漫画vip账号共享2020_yy漫画免费账号密码共享
歪歪漫画vip账号共享2020_yy漫画免费账号密码共享 2020-04-07
沙雕群名称大全2019精选 今年最火的微信群名沙雕有创意
沙雕群名称大全2019精选 今年最火的微信群名沙雕有创意 2019-07-07
玄元剑仙肉身有什么用 玄元剑仙肉身境界等级划分
玄元剑仙肉身有什么用 玄元剑仙肉身境界等级划分 2019-06-21
男生常说24816是什么意思?女生说13579是什么意思?
男生常说24816是什么意思?女生说13579是什么意思? 2019-09-17
超A是什么意思 你好a表达的是什么
超A是什么意思 你好a表达的是什么 2019-06-06
返回顶部