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)

}

執行結果:

 


注意!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系我们删除。



 
粤ICP备14056181号  © 2014-2021 ITdaan.com