网络通信3：TCP交互通信

服务端实现

import (
  "fmt"
  "net"
  "os"
  "strings"
)

func CheckErrorS(err error) {
  if err != nil {
    fmt.Println("网络错误", err.Error())
    os.Exit(1)
  }
}

func Processinfo(conn net.Conn) {
  buffer := make([]byte, 1024) //开创缓冲区
  defer conn.Close()           //关闭连接

  for {
    n, err := conn.Read(buffer) //读取数据
    CheckErrorS(err)

    if n != 0 {
      //拿到客户端地址
      remoteAddr := conn.RemoteAddr()
      msg := string(buffer[:n])
      fmt.Println("收到消息",msg, "来自", remoteAddr)

      if strings.Contains(msg,"钱") {
        conn.Write([]byte("fuckoff"))
        break
      }
      conn.Write([]byte("已阅："+msg))
    }
  }

}

func main() {
  //建立TCP服务器
  listener, err := net.Listen("tcp", "127.0.0.1:8898")
  CheckErrorS(err)
  defer listener.Close() //关闭网络

  fmt.Println("服务器正在等待")

  for {
    conn, err := listener.Accept() //新的客户端连接
    CheckErrorS(err)

    //处理每一个客户端
    go Processinfo(conn)
  }

}

客户端实现

import (
  "fmt"
  "net"
  "bufio"
  "os"
)

func CheckErrorC(err error) {
  if err != nil {
    fmt.Println("网络错误", err.Error())
    os.Exit(1)
  }
}

func MessageSend(conn net.Conn) {
  var msg string
  reader := bufio.NewReader(os.Stdin) //读取键盘输入

  for {
    lineBytes, _, _ := reader.ReadLine() //读取一行
    msg = string(lineBytes)              //键盘输入转化为字符串

    if msg == "exit" {
      conn.Close()
      fmt.Println("客户端关闭")
      break
    }

    _, err := conn.Write([]byte(msg)) //输入写入字符串
    if err != nil {
      conn.Close()
      fmt.Println("客户端关闭")
      break
    }

  }
}

func main() {
  conn, err := net.Dial("tcp", "127.0.0.1:8898") //建立TCP服务器
  CheckErrorC(err)                               //检查错误
  defer conn.Close()

  //发送消息中有阻塞读取标准输入的代码
  //为了避免阻塞住消息的接收，所以把它独立的协程中
  go MessageSend(conn)

  buffer := make([]byte, 1024)
  for {
    n, err := conn.Read(buffer)
    CheckErrorC(err)

    msg := string(buffer[:n])
    fmt.Println("收到服务器消息", msg)

    if msg=="fuckoff"{
      break
    }

  }

  fmt.Println("连接已断开")
}