go 网络编程 文件传输 from 黑马程序员

文件传输的实现过程

示意简图

        传输文件实际上是先把文件名发送给接收端，接收端创建同名文件，传输段在把文件内容传过去，实现的文件传输

 

func main() {
	//os.args 获取在命令行输入对应程序后的内容，程序名加后续内容
	//例如 ： go run get_config.go a.txt
	list := os.Args
	fmt.Printf("list: %v\n", list)
	fmt.Printf("list: %T\n", list)
}

var Args []string        

Args保管了命令行参数，第一个是程序名 

PS D:\goLang\web\send-file\get-file-config> go run .\get_config.go D:\goLang\a.log

name=a.log
size=67

如果直接打印获取的内容和其类型，这里为list，可以得到以下结果：

list: [C:\Users\wang2\AppData\Local\Temp\go-build589202303\b001\exe\get_config.exe D:\goLang\a.log]
list: []string

可以看到其为字符串类型的切片，其第一个内容为临时的程序的exe文件，而第二个则为后面输入的内容

实现代码

发送端：

package main

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

func Sendfile(path string, c net.Conn) {
	//以只读方式打开文件
	f, err := os.Open(path)
	if err != nil {
		fmt.Printf("open file failed ,err: %v\n", err)
		return
	}

	defer f.Close()

	//读文件内容，读多少发多少
	buf := make([]byte, 1024)
	for {
		// 循环读取，每次最多1024个字
		n, err2 := f.Read(buf)
		//发送内容
		fmt.Printf("buf[:n]: %v\n", string(buf[:n]))
		c.Write(buf[:n])
		if err2 == io.EOF {
			fmt.Println("发送完毕")
			return
		}
	}

}

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

	//可以回顾scan的各种类型的区别，见 https://blog.csdn.net/weixin_45765795/article/details/112766580
	fmt.Scan(&path)
	fmt.Printf("path: %v\n", path)

	//获取文件名
	fi, err := os.Stat(path)
	if err != nil {
		fmt.Printf("err: %v\n", err)
		return
	}

	//主动连接服务器
	c, err2 := net.Dial("tcp", "127.0.0.1:20000")
	if err2 != nil {
		fmt.Printf("dial failed,err2: %v\n", err2)
		return
	}
	fmt.Println("连接成功！")
	defer c.Close()

	//给接收方先发送文件名
	_, err3 := c.Write([]byte(fi.Name()))
	if err3 != nil {
		fmt.Printf("write failed ,err3: %v\n", err3)
		return
	}

	//接受对方的回复。如果回复“ok”，说明准备好了
	buf := make([]byte, 1024)
	n, err4 := c.Read(buf)
	fmt.Printf("buf: %v\n", string(buf))

	//上文中的写法与这里有什么区别？
	// r := bufio.NewReader(c)
	// n3, err5 := r.Read(buf)

	if err4 != nil {
		fmt.Printf("read failed,err4: %v\n", err4)
		return
	}
	if string(buf[:n]) == "ok" {
		fmt.Println("接收方已准备好")
		//发送文件内容
		Sendfile(path, c)
	} else {
		fmt.Println("对面没有返回 ok ，请重新准备")
	}
}

接收端

package main

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

func receiveFile(filename string, conn net.Conn) {
	//创建文件
	fmt.Printf("filename: %v\n", filename)
	f, err := os.Create(filename)

	if err != nil {
		fmt.Printf("create failed,err: %v\n", err)
		return
	}
	defer f.Close()
	//接受多少，写入多少内容
	buf := make([]byte, 1024)
	for {
		n, err2 := conn.Read(buf)
		fmt.Printf("buf: %v\n", string(buf[:n]))

		if err2 != nil {
			if err2 == io.EOF {
				fmt.Println("读取完毕")
				return
			}

		}
		f.Write(buf[:n])
	}

}

func main() {
	//监听串口
	l, err := net.Listen("tcp", "127.0.0.1:20000")
	if err != nil {
		fmt.Printf("listen failed ,err: %v\n", err)
		return
	}

	//阻塞等待用户连接
	c, err2 := l.Accept()
	if err2 != nil {
		fmt.Printf("accept failed,err2: %v\n", err2)
		return
	}
	fmt.Println("连接成功")
	defer c.Close()

	//读取对方发送的文件名
	buf := make([]byte, 1024)

	r := bufio.NewReader(c)
	n, err3 := r.Read(buf)
	if err3 != nil {
		fmt.Printf("read failed,err3: %v\n", err3)
		return
	}
	fileName := string(buf[:n])
	fmt.Printf("fileName: %v\n", fileName)

	//返回ok.
	if n > 0 {
		c.Write([]byte("ok"))
		//介绍文件内容
		receiveFile(fileName, c)
	}

}

        代码过程中需要注意在读缓存时需要在原本切边的基础上切出有内容的切片，不然os.open和确认传输内容的过程中会出现不符合预期的错误