【GO语言】实现网络文件传输

项目总览：

1.开发语言：GO语言
2.IDE：Goland
3.开发用时：一个小时
4.源码链接：https://pan.baidu.com/s/1FbyrWv2sopbjM9fzRhpfhQ
提取码：0000 --来自百度网盘超级会员V1的分享

---

以下是本篇文章正文内容，欢迎朋友们进行指正，一起探讨，共同进步。——来自考研路上的lwj。QQ：2394799692

一、项目开发日志

文件传输——发送端(客户端)：
1.提示用户使用命令行参数输入文件名。接收文件名filepath(含访问路径)
2.使用os.stat()获取文件属性，得到纯文件名filename(去除访问路径)
3.主动发起连接服务器请求，结束时关闭连接
4.发送文件名到接收端conn.write()
5.读取接收端回发的确认数据conn.read()
6.判断是否“ok”。如果是，封装函数sendfile()发送文件内容，传参filepath和conn
7.只读open文件，结束时close文件
8.循环读本地文件，读到EOF，读取完毕。
9.将读到的内容原封不动conn.w给接收端

文件传输——接收端：
1.创建监听listener，程序结束时关闭。
2.阻塞等待客户端连接conn，程序结束时关闭conn。
3.读取客户端发送文件名。保存filename。
4.回发“ok”。
5.封装函数recvfile接收客户端发送的文件内容。传参filename和conn
6.按文件名create文件，结束时close
7.循环read发送端网络文件内容，当读到说明文件读取完毕。
8.将读到的内容原封不动write到创建的文件中

二、图解传输过程

三、发送端端代码展示

package main

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

func sendFile(conn net.Conn, filePath string) {
	//只读打开文件
	f, err := os.Open(filePath)
	if err != nil {
		fmt.Println("os.Open err:", err)
		return
	}
	defer f.Close()
	//从本文件中写数据，写给网络接收端。读多少写多少，原封不动
	buf := make([]byte, 4096)
	for {
		n, err := f.Read(buf)
		if err != nil {
			if err == io.EOF {
				fmt.Println("发送文件完毕。")
			} else {
				fmt.Println("f.Read err:", err)
			}
			return
		}
		//写到网络socket中
		_, err = conn.Write(buf[:n])
		if err != nil {
			fmt.Println("conn.Write err:", err)
			return
		}
	}
}
func main() {
	list := os.Args //获取命令行参数
	//提取文件的绝对路径
	if len(list) != 2 {
		fmt.Println("格式为：go run xxx.go 文件绝对路径")
		return
	}
	//提取文件的绝对路径
	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:8008")
	if err != nil {
		fmt.Println("net.Dial err:", err)
		return
	}
	defer conn.Close()
	//发送文件名给接收端
	_, err = conn.Write([]byte(fileName))
	if err != nil {
		fmt.Println("conn.Write err:", err)
		return
	}
	//读取服务器回发的OK
	buf := make([]byte, 16)
	n, err := conn.Read(buf)
	if err != nil {
		fmt.Println("conn.Read err:", err)
		return
	}
	if "ok" == string(buf[:n]) {
		//写文件内容给服务器——借助conn
		sendFile(conn, filePath)
	}
}

四、接收端代码展示

package main

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

func recvFile(conn net.Conn, fileName string) {
	//按照文件名创建新文件
	f, err := os.Create(fileName)
	if err != nil {
		fmt.Println("os.crreate err:", err)
		return
	}
	defer f.Close()
	//从网络中读数据，写入本地文件
	buf := make([]byte, 4096)
	for {
		n, _ := conn.Read(buf)
		if n == 0 {
			fmt.Println("接收文件完成。")
			return
		}
		//写入本地文件，读多少，写多少
		f.Write(buf[:n])
	}

}
func main() {
	//创建用于监听的socket
	listener, err := net.Listen("tcp", "127.0.0.1:8008")
	if err != nil {
		fmt.Println("net.Listen err:", err)
		return
	}
	defer listener.Close()
	//阻塞监听
	conn, err := listener.Accept()
	if err != nil {
		fmt.Println("listener.Accept err:", err)
		return
	}
	defer conn.Close()
	//获取文件名，保存
	buf := make([]byte, 4096)
	n, err := conn.Read(buf)
	if err != nil {
		fmt.Println("conn.Read err:", err)
		return
	}
	fileName := string(buf[:n])
	//回写ok给发送端
	conn.Write([]byte("ok"))
	//获取文件内容
	recvFile(conn, fileName)
}

---

五、效果展示：

启动接收端后，在根目录中打开cmd输入go run 文件传输-发送端.go F:\web作业\news.sql，客户端开始进行文件传输，传输完毕后，返回发送文件完毕。