【golang】实现基于TCP协议的文件传输

利用tcp协议,建立发送端与接收端的连接,通过这个连接,以字节切片的形式,完成文件传输。

一、发送端

发送端主要步骤为:

1.建立与接收端的连接;
2.读取文件信息;
3.向接收端传递文件名;
4.读取接收端返回的指令,若为OK,则执行文件传输;
5.以字节切片循环读取文件内容,使用连接将文件内容传送到接收端。
package main

import (
    "fmt"
    "io"
    "net"
    "os"
)

func main() {
    //获取命令行参数,用命令传递文件go run send.go D:\1.mp3,参数为1:send.go,2:D:\1.mp3
    list := os.Args
    //文件路径
    filepath := list[1]
    //文件属性
    fileInfo, err := os.Stat(filepath)
    if err != nil {
        fmt.Println("os.Stat err", err)
        return
    }
    filename := fileInfo.Name()
    /**
    建立连接
    */
    conn, err := net.Dial("tcp", "127.0.0.1:8003")
    if err != nil {
        fmt.Println("net.Dialt err", err)
        return
    }
    //发送文件名到接收端
    _, err = conn.Write([]byte(filename))
    if err != nil {
        fmt.Println("conn.Write err", err)
        return
    }
    buf := make([]byte, 4096)
    //接收服务器返还的指令
    n, err := conn.Read(buf)
    if err != nil {
        fmt.Println("conn.Read err", err)
        return
    }
    //返回ok,可以传输文件
    if string(buf[:n]) == "ok" {
        sendFile(conn, filepath)
    }

}
func sendFile(conn net.Conn, filepath string) {
    //打开要传输的文件
    file, err := os.Open(filepath)
    if err != nil {
        fmt.Println("os.Open err", err)
        return
    }
    buf := make([]byte, 4096)
    //循环读取文件内容,写入远程连接
    for {
        n, err := file.Read(buf)
        if err == io.EOF {
            fmt.Println("文件读取完毕")
            return
        }
        if err != nil {
            fmt.Println("file.Read err:", err)
            return
        }
        _, err = conn.Write(buf[:n])
        if err != nil {
            fmt.Println("conn.Write err:", err)
            return
        }
    }
}

二、接收端

接收端的主要内容是:

1.建立监听服务,等待连接;
2.获取发送端的连接,读取请求数据,获取文件名,返回“ok”给发送端;
3.创建一个新的文件;
4.循环读取连接中发送来的信息,将其写入到创建的新文件中。
package main

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

func main() {
    //创建监听服务
    listener, err := net.Listen("tcp", "127.0.0.1:8003")
    if err != nil {
        fmt.Println("net.Listen err:", err)
        return
    }
    /**
    等待接受连接
    */
    conn, err := listener.Accept()
    if err != nil {
        fmt.Println("listener.Accept err:", err)
        return
    }
    /**
    读取文件名,向文件发送者返回OK
     */
    buf := make([]byte, 4096)
    n, err := conn.Read(buf)
    if err != nil {
        fmt.Println("conn.Read err:", err)
        return
    }
    filename := string(buf[:n])
    fmt.Println("filename:", filename)
    if filename != "" {
        _, err = conn.Write([]byte("ok"))
        if err != nil {
            fmt.Println("conn.Write err:", err)
            return
        }
    } else {
        return
    }
    /**
    创建文件并写入文件内容
    */
    fmt.Println(filename)
    file, err := os.Create(filename)
    if err != nil {
        fmt.Println("os.Create err:", err)
        return
    }

    for {
        n, err := conn.Read(buf)
        if n == 0 {
            fmt.Println("文件读取完毕")
            break
        }
        if err != nil {
            fmt.Println("conn.Read err:", err)
            return
        }
        file.Write(buf[:n])
    }

}

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • 国家电网公司企业标准(Q/GDW)- 面向对象的用电信息数据交换协议 - 报批稿:20170802 前言: 排版 ...
    庭说阅读 12,196评论 6 13
  • 一、TCP的可靠性 TCP向应用层提供与UDP完全不同的服务。它提供一种面向连接的、可靠的字节流服务。TCP通过下...
    ZMRWEGo阅读 1,422评论 0 0
  • 计算机网络概述 网络编程的实质就是两个(或多个)设备(例如计算机)之间的数据传输。 按照计算机网络的定义,通过一定...
    蛋炒饭_By阅读 1,350评论 0 10
  • 实时消息协议---流的分块 版权声明: 版权(c)2009 Adobe系统有限公司。全权所有。 摘要: 本备忘录描...
    一个人zy阅读 2,059评论 0 9
  • 计算机网络七层模型中,传输层有两个重要的协议:(1)用户数据报协议UDP (User Datagram Proto...
    Q南南南Q阅读 1,852评论 0 3

友情链接更多精彩内容