Go语言Telnet回音服务器的实现

时间:2021-10-02 04:54:56

本文介绍了Go语言Telnet回音服务器,分享给大家,具体如下:

?
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
package main
 
import (
  "fmt"
  "net"
  "bufio"
  "strings"
  "os"
)
 
func main() {
 
  // 创建一个程序结束码的通道
  exitChan := make(chan int)
 
  // 将服务器并发运行
  go server("127.0.0.1:7001", exitChan)
 
  // 通道阻塞, 等待接收返回值
  code := <-exitChan
 
  // 标记程序返回值并退出
  os.Exit(code)
}
 
// 服务逻辑, 传入地址和退出的通道
func server(address string, exitChan chan int) {
 
  // 根据给定地址进行侦听
  l, err := net.Listen("tcp", address)
 
  // 如果侦听发生错误, 打印错误并退出
  if err != nil {
   fmt.Println(err.Error())
   exitChan <- 1
  }
 
  // 打印侦听地址, 表示侦听成功
  fmt.Println("listen: " + address)
 
  // 延迟关闭侦听器
  defer l.Close()
 
  // 侦听循环
  for {
 
   // 新连接没有到来时, Accept是阻塞的
   conn, err := l.Accept()
 
   // 发生任何的侦听错误, 打印错误并退出服务器
   if err != nil {
     fmt.Println(err.Error())
     continue
   }
 
   // 根据连接开启会话, 这个过程需要并行执行
   go handleSession(conn, exitChan)
  }
}
 
// 连接的会话逻辑
func handleSession(conn net.Conn, exitChan chan int) {
 
  fmt.Println("Session started:")
 
  // 创建一个网络连接数据的读取器
  reader := bufio.NewReader(conn)
 
  // 接收数据的循环
  for {
 
   // 读取字符串, 直到碰到回车返回
   str, err := reader.ReadString('\n')
 
   // 数据读取正确
   if err == nil {
 
     // 去掉字符串尾部的回车
     str = strings.TrimSpace(str)
 
     // 处理Telnet指令
     if !processTelnetCommand(str, exitChan) {
      conn.Close()
      break
     }
 
     // Echo逻辑, 发什么数据, 原样返回
     conn.Write([]byte(str + "\r\n"))
 
   } else {
     // 发生错误
     fmt.Println("Session closed")
     conn.Close()
     break
   }
  }
 
}
 
//命令处理
func processTelnetCommand(str string, exitChan chan int) bool {
 
  // @close指令表示终止本次会话
  if strings.HasPrefix(str, "@close") {
 
   fmt.Println("Session closed")
 
   // 告诉外部需要断开连接
   return false
 
   // @shutdown指令表示终止服务进程
  } else if strings.HasPrefix(str, "@shutdown") {
 
   fmt.Println("Server shutdown")
 
   // 往通道中写入0, 阻塞等待接收方处理
   exitChan <- 0
 
   // 告诉外部需要断开连接
   return false
  }
 
  // 打印输入的字符串
  fmt.Println(str)
 
  return true
 
}

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。

原文链接:https://studygolang.com/articles/26146