站长资源脚本专栏

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

}