Golang: TCP 文件传输案例

基于 tcp 网络的文件读写案例，反复演练 c-s 通信过程。

专栏的介绍可以参考 《CaseGolang专栏》，代码可以看《宝库-case》。

流程和原理设计如下：

流程比较清晰，代码也比较简单，直接贴在下面了: 已经上传 github.

客户端代码: client.go

func main() {
  //获取命令行参数 (包含要发送的文件名)
  list := os.Args
  if len(list) != 2 {
    fmt.Println("usage : ./client filePath")
    return
  }

  filePath := list[1] //发送给 server

  info, err := os.Stat(filePath) //获取文件属性
  if err != nil {
    fmt.Println("err = ", err)
    return
  }

  //info.Name(), info.Size()

  //连接服务器
  conn, err1 := net.Dial("tcp", "127.0.0.1:8000")
  if err1 != nil {
    fmt.Println("dial error : ", err1)
    return
  }
  defer conn.Close()

  //先发送一次文件名
  _, err = conn.Write([]byte(info.Name()))
  if err != nil {
    fmt.Println("conn Write server err : ", err)
    return
  }

  //坐等接收对方的回应(最好是 "ok" )
  var n int
  buf := make([]byte, 1024)
  n, err = conn.Read(buf)
  if err != nil {
    fmt.Println("conn.Read err = ", err)
    return
  }
  if "ok" == string(buf[:n]) {
    //发送文件
    sendFile(filePath, conn)
  }
}

func sendFile(filePath string, conn net.Conn) {
  //读多少写多少
  f, err := os.Open(filePath)
  if err != nil {
    fmt.Println("os.Open err = ", err)
    return
  }
  defer f.Close()

  buf := make([]byte, 1024*4)
  for {
    n, err := f.Read(buf)
    if err != nil {
      if err == io.EOF && n ==0 {
        fmt.Println("文件发送完毕")
      } else {
        fmt.Println("f.Read err = ", err)
      }
      return 
    }
    conn.Write(buf[:n])
  }

}

服务端代码: server.go

func main() {
  listener, err := net.Listen("tcp", "127.0.0.1:8000")
  if err != nil {
    fmt.Println("net.Listen err =", err)
    return
  }
  defer listener.Close()

  conn, err1 := listener.Accept()
  if err1 != nil {
    fmt.Println("listener.Accept err = ", err1)
    return
  }
  defer conn.Close()

  //读取客户端请求
  buf := make([]byte, 1024)
  var n int
  n, err = conn.Read(buf)
  if err != nil {
    fmt.Println("conn Read err = ", err1)
    return
  }

  fileName := string(buf[:n])
  conn.Write([]byte("ok"))

  //介绍文件内容
  recvFile(fileName, conn)
}

func recvFile(fileName string, conn net.Conn) {
  //新建一个文件
  f ,err := os.Create(fileName)
  if err != nil {
    fmt.Println("create file err = ", err)
    return
  }
  defer f.Close()

  //从 conn 读取然后写入本地文件
  buf := make([]byte, 1024*4)

  //接收多少写多少
  for{
    n, err := conn.Read(buf)
    if err != nil && err != io.EOF {
      fmt.Println("conn.Read err = ", err)
    }
    if n == 0 {
      fmt.Println("文件接收完毕")
      return
    }
    f.Write(buf[:n])
  }
}

运行结果如下：

---

Merlin 上传 tcp 案例