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

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

}