golang_文件传输: go实现C/S构架下的文件传输系统

版权声明：本文为博主原创文章，转载请附上博文链接！ https://blog.csdn.net/weixin_43851310/article/details/87993308

文件传输

流程(3步):

sender---------------receiver

发送文件名 —> 保存文件名

接收到"ok" <— 回复"ok"

发送文件内容 —> 接收文件

传输文件: 发送方

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

//发送文件内容
func SendFile(path string, conn net.Conn) {
	//以只读方式打开文件
	f, err := os.Open(path)
	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 {
				fmt.Println("文件发送完毕")
			}else {
				fmt.Println("f.Read err =",err)
			}
			return
		}
		//给服务器发送内容
		conn.Write(buf[:n])
	}
}

func main() {
	//提示输入文件
	fmt.Println("请输入需要传输的文件:")
	var path string
	fmt.Scan(&path)

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

	//主动连接服务器
	conn, err := net.Dial("tcp","127.0.0.1:8080")
	if err != nil {
		fmt.Println("net.Dial err =",err)
		return
	}
	defer conn.Close()

	//给接收方,先发送文件名
	_, err = conn.Write([]byte(info.Name()))
	if err != nil {
		fmt.Println("conn.Write err =",err)
		return
	}

	//若接收到"ok"的回复,说明对方准备好吗,可发送文件
	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(path, conn) //发送文件内容
	}
}

传输文件: 接收方

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

//接收文件内容
func RecFile(fileName string, conn net.Conn) {
	//新建文件
	f, err := os.Create(fileName)
	if err != nil {
		fmt.Println("os.Create err =",err)
		return
	}

	buf := make([]byte, 1024*4)
	//读文件内容
	for ; ; {
		//注意这里,接收的是发送方发来的信息,不是上面读取的fileName的信息(f.Read)
		n, err := conn.Read(buf)
		if err != nil {
			if err == io.EOF {
				fmt.Println("文件接收完毕")
			} else {
				fmt.Println("f.Read err =", err)
			}
			return
		}
		if n == 0 {
			fmt.Println("n == 0,文件接收完毕")
			break
		}
		f.Write(buf[:n]) //往文件写入内容
	}
}

func main() {
	//创建进行监听的套接字
	listener, err := net.Listen("tcp","127.0.0.1:8080")
	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, 1024)
	n, err := conn.Read(buf) //读取通讯方发送的文件名
	if err != nil {
		fmt.Println("conn.Read err =",err)
		return
	}

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

	//接收文件内容
	RecFile(fileName, conn)
}

Output:

终端1: go run 01-传输文件_发送方.go

请输入需要传输的文件:
E:\face.jpg
文件发送完毕

终端2: go run 02-传输文件_接收方.go

文件接收完毕

最终文件传输到,该程序代码目录.