脚本之家，脚本语言编程技术及教程分享平台！
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|

服务器之家 - 脚本之家 - Golang - go语言实现聊天服务器的示例代码

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

2020-05-17 12:09Y_xx Golang

这篇文章主要介绍了go语言实现聊天服务器的示例代码，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧

看了两天 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

  • GO语言
  • 聊天服务器

    • 延伸 · 阅读

    精彩推荐
    • Golanggolang json.Marshal 特殊html字符被转义的解决方法

      golang json.Marshal 特殊html字符被转义的解决方法

      今天小编就为大家分享一篇golang json.Marshal 特殊html字符被转义的解决方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧 ...

      李浩的life12792020-05-27
    • GolangGolang中Bit数组的实现方式

      Golang中Bit数组的实现方式

      这篇文章主要介绍了Golang中Bit数组的实现方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...

      天易独尊11682021-06-09
    • GolangGolang通脉之数据类型详情

      Golang通脉之数据类型详情

      这篇文章主要介绍了Golang通脉之数据类型，在编程语言中标识符就是定义的具有某种意义的词，比如变量名、常量名、函数名等等，Go语言中标识符允许由...

      羌4272021-11-24
    • Golanggolang如何使用struct的tag属性的详细介绍

      golang如何使用struct的tag属性的详细介绍

      这篇文章主要介绍了golang如何使用struct的tag属性的详细介绍，从例子说起，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看...

      Go语言中文网11352020-05-21
    • Golanggolang 通过ssh代理连接mysql的操作

      golang 通过ssh代理连接mysql的操作

      这篇文章主要介绍了golang 通过ssh代理连接mysql的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...

      a165861639710342021-03-08
    • Golanggolang的httpserver优雅重启方法详解

      golang的httpserver优雅重启方法详解

      这篇文章主要给大家介绍了关于golang的httpserver优雅重启的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，...

      helight2992020-05-14
    • Golanggo日志系统logrus显示文件和行号的操作

      go日志系统logrus显示文件和行号的操作

      这篇文章主要介绍了go日志系统logrus显示文件和行号的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...

      SmallQinYan12302021-02-02
    • Golanggo语言制作端口扫描器

      go语言制作端口扫描器

      本文给大家分享的是使用go语言编写的TCP端口扫描器，可以选择IP范围，扫描的端口，以及多线程，有需要的小伙伴可以参考下。 ...

      脚本之家3642020-04-25
    最近更新
    编辑推荐
    阅读排行
    热门标签
    gosublime 　 指针类型 　 new() 　 make() 　 并发模型 　 切片 　 静态WEB服务器 　 端口扫描器 　 Go语言入门教程 　 基础语 　 Slices 　 Maps 　 Golang编程 　 panic 　 接受者类型 　 嵌入类型 　 Slice 　 Golang教程 　 Golang项目 　 内存模型 　 开发环境搭建 　 大小端判断 　 代码组织结构 　 复合类型 　 GOPATH 　 工作目录 　 实例 　 struct类型 　 并发技术 　 buffered 　
    © 2019-2023 脚本之家 | 服务器之家(www.zzvips.com)旗下站点 版权所有关于我们联系我们版权申明网站地图