【GO】TCP交替通信

package main
import (
  "fmt"
  "net"
  "os"
)
/**
处理异常
*/
func JtHandleError(err error, when string) {
  if err != nil {
    fmt.Println(when, "错误原因：", err)
    os.Exit(1)
  }
}
func main() {
  listener, err := net.Listen("tcp", "127.0.0.1:8888")
  JtHandleError(err, "net.Listen")
  fmt.Println("listener...")
  defer listener.Close()
  for {
    // 接入一个客户端
    conn, err := listener.Accept()
    JtHandleError(err, "listener.Accept")
    // 给每一个客户端开启一个线程
    go JtIoWithConn(conn)
  }
}
func JtIoWithConn(conn net.Conn) {
  defer conn.Close()
  // 创建缓存区
  buffer := make([]byte, 1024)
  // 这个获取的是客户端的地址
  clientAddr := conn.RemoteAddr()
  for {
    // 读入消息
    n, err := conn.Read(buffer)
    JtHandleError(err, "conn.Read")
    // 客户端消息
    clientMsg := string(buffer[:n])
    // 打印出客户端发过来的消息
    fmt.Println(clientAddr, clientMsg)
    // 服务端给客户端写的消息
    conn.Write([]byte("服务端以阅：" + clientMsg))
    if clientMsg == "over" {
      fmt.Printf("%v以下线\n", clientAddr)
      break
    }
  }
}

package main
import (
  "fmt"
  "net"
  "os"
)
/**
处理异常
*/
func JcHandleError(err error, when string) {
  if err != nil {
    fmt.Println(when, "错误原因：", err)
    os.Exit(1)
  }
}
func main() {
  // 拨号连接,得到专线conn
  conn, err := net.Dial("tcp", "127.0.0.1:8888")
  JcHandleError(err, "net.Dial")
  defer conn.Close()
  // 在专线中处理io
  ioInteractive(conn)
}
func ioInteractive(conn net.Conn) {
  defer conn.Close()
  // 在一条独立的协程里接收服务端的消息
  go func() {
    // 创建一个缓冲区
    buffer := make([]byte, 1024)
    // 循环读取
    for {
      // 从缓冲区读取消息
      n, err := conn.Read(buffer)
      JcHandleError(err, "conn.Read")
      serverMsg := string(buffer[:n])
      fmt.Println("服务端消息:", serverMsg)
    }
  }()
  // 在主协程里向服务端写消息
  var userInput string
  for {
    // 标准输入与输出
    fmt.Scan(&userInput)
    // 向服务端写出
    conn.Write([]byte(userInput))
    if userInput == "exit" {
      return
    }
  }
}