go网络编程40——文件传输

发送端

package main

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

func SendFile(path string, conn net.Conn) {
	//只读方式打开文件
	file, err := os.Open(path)
	if err != nil {
		fmt.Println("os.Open err = ", err)
		return
	}

	defer file.Close()

	//读文件内容
	buf := make([]byte, 4*1024)
	for {
		num, err := file.Read(buf)
		if err != nil {
			if err == io.EOF {
				fmt.Println("文件发送完毕")
			} else {
				fmt.Println("file.Read err = ", err)
			}

			return
		}

		//发送内容
		conn.Write(buf[:num])
	}
}

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

	info, err := os.Stat(path)
	if err != nil {
		fmt.Println("os.Stat err = ", err)
		return
	}

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

	defer conn.Close()

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

	//接收对方的回复，如果ok,说明对方准备好了，可以发文件
	buf := make([]byte, 1024)
	num, err3 := conn.Read(buf)
	if err3 != nil {
		fmt.Println("conn.Write Read = ", err3)
		return
	}

	if "ok" == string(buf[:num]) {
		//发送文件内容
		SendFile(path, conn)
	}
}

接收端

package main

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

func RecvFile(filename string, conn net.Conn) {
	//新建文件，file相当于文件句柄(可读可写)
	file, err := os.Create(filename)
	if err != nil {
		fmt.Println("os.Create err = ", err)
		return
	}

	buf := make([]byte, 4*1024)
	for {
		num, err := conn.Read(buf)
		if err != nil {
			if err == io.EOF {
				fmt.Println("文件接收完毕")
				return
			} else {
				fmt.Println("conn.Read err = ", err)
				return
			}
		}

		//写入内容
		file.Write(buf[:num])
	}
}

func main() {
	//监听
	listen, err := net.Listen("tcp", "127.0.0.1:8000")
	if err != nil {
		fmt.Println("net.Listen err = ", err)
		return
	}

	defer listen.Close()

	//阻塞用户等待连接
	conn, err1 := listen.Accept()
	if err1 != nil {
		fmt.Println("listen.Accept err1 = ", err1)
		return
	}

	defer conn.Close()

	//读取对方发送的文件名
	buf := make([]byte, 1024)
	num, err2 := conn.Read(buf)
	if err2 != nil {
		fmt.Println("conn.Read err2 = ", err2)
		return
	}

	filename := string(buf[:num])

	//成功接收到文件名 并返回ok
	conn.Write([]byte("ok"))

	//接收文件内容
	RecvFile(filename, conn)
}