使用go语言实现传输文件

使用go语言实现传输文件

发送方（客户端）

package main

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

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:8080")
	if err1 != nil{
		fmt.Println("net.Dail err1:",err1)
		return
	}
	defer conn.Close()
	
	//给接收方先发送文件名
	_ , err2 := conn.Write([]byte(info.Name()))
	if err2 != nil{
		fmt.Println("conn.Write err",err2)
		return
	}
	
	//接受对方的回复，如果回复“ok”，说明对方准备好，可以发送文件，相当于一次简单的握手过程
	buf := make([]byte,1024)
	n,err3 := conn.Write(buf)
	if err != nil{
		fmt.Println("conn.Write err:",err3)
	}
	if "ok" == string(buf[:n]){
	//	发送文件内容
	SendFile(path,conn)
	}
}

func SendFile(path string,conn net.Conn){
	//以只读方式打开文件
	f ,err4 := os.Open(path)
	if err4 != nil{
		fmt.Println("os.Open err:",err4)
		return
	}
	defer f.Close()
	//读文件内容，读多少内容就发送多少内容
	buf := make([]byte,1024*4)
	for {
		n,err5 := f.Read(buf)	//从文件读取内容
		if err5 != nil {
			if err5 == io.EOF {
				fmt.Println("文件发送完毕")
			}else {
				fmt.Println("f.Read err:",err5)
			}
			return
		}
		//发送内容
		conn.Write(buf[:n])		//给服务器发送内容
	}
}

接收方

package main

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

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,err1 := listener.Accept()
	if err1 != nil {
		fmt.Println("listener.Accept err:",err1)
		return
	}
	defer conn.Close()

	buf := make([]byte,1024)
	n,err2 := conn.Read(buf)	//读取对方发送的文件名
	if err2 != nil {
		fmt.Println("conn.Read err",err2)
		return
	}
	fileName := string(buf[:n])
	//回复“ok”
	conn.Write([]byte("ok"))

	//接受文件内容
	RecvFile(fileName,conn)
}

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

	buf := make([]byte,1024*4)
	//接受多少，写多少
	for {
		n,err4 := conn.Read(buf)	//接受对方发送过来的文件内容
		if err4 != nil {
			if err4 == io.EOF{
				fmt.Println("文件接受完毕")
			}else {
				fmt.Println("conn.Read err",err4)
			}
			return
		}

		if n == 0 {
			fmt.Println("n == 0,文件接受完毕")
			return
		}
		f.Write(buf[:n])
	}
}