Go語言之進階篇文件傳輸

 一、文件傳輸

1、文件傳輸原理

2、文件傳輸

示例：

發送方： send_file.go

package main

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, err1 := net.Dial("tcp", "127.0.0.1:8000")
	if err1 != nil {
		fmt.Println("net.Dial err1 = ", err1)
		return
	}

	defer conn.Close()

	//給接收方，先發送文件名
	_, err = conn.Write([]byte(info.Name()))
	if err != nil {
		fmt.Println("conn.Write 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(path, conn)
	}
}

 

接收方： recv_file.go

package main

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

//接收文件內容
func RecvFile(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 {
		n, err := conn.Read(buf) //接收對方發過來的文件內容
		if err != nil {
			if err == io.EOF {
				fmt.Println("文件接收完畢")
			} else {
				fmt.Println("conn.Read err = ", err)
			}
			return
		}

		if n == 0 {
			fmt.Println("n == 0 文件接收完畢")
			break
		}

		f.Write(buf[:n]) //往文件寫入內容
	}

}

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

	defer listenner.Close()

	//阻塞等待用戶連接
	conn, err1 := listenner.Accept()
	if err1 != nil {
		fmt.Println("listenner.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 = ", err)
		return
	}

	fileName := string(buf[:n])

	//回復"ok"
	conn.Write([]byte("ok"))

	//接收文件內容
	RecvFile(fileName, conn)

}

執行結果：