golang学习2---网络传送文件

client.go

 

package main

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

func sendFile(conn net.Conn,filePath string) {
	//只读打开文件
	f, err := os.Open(filePath)
	if err != nil {
		fmt.Println("os.Open err:",err)
		return
	}
	defer f.Close()

	//从本文件中，读数据，写给网络接收端。读多少，写多少。原封不动
	buf := make([]byte,4096)
	for {
		n,err := f.Read(buf)
		if err != nil {
			if err == io.EOF {
				fmt.Println("发送文件完成。")
				return
			}else {
				fmt.Println("os.Open err:",err)
				return
			}

		}
		//写道网络socket中
		_,err = conn.Write(buf[:n])
		if err != nil {
			fmt.Println("conn.Write err:",err)
			return
		}

	}
}
func main() {
	list := os.Args
	if len(list) != 2 {
		fmt.Println("格式为：go run xxx.go 文件绝对路径")
		return
	}
	//提取文件绝对路径
	filePath := list[1]
	//提取文件名
	fileInfo, err := os.Stat(filePath)
	if err != nil {
		fmt.Println("os.Stat err:",err)
		return
	}
	fileName := fileInfo.Name()

	//主动发起连接请求
	conn,err := net.Dial("tcp","127.0.0.1:8008")
	if err != nil {
		fmt.Println("net.Dial err:",err)
		return
	}
	defer conn.Close()

	//发送文件名给接收端
	n,err := conn.Write([]byte(fileName))
	if err != nil{
		fmt.Println("conn.Write err:",err)
		return
	}
	//读取服务器回发的 OK
	buf := make([]byte,16)
	n, err = conn.Read(buf)
	if err != nil{
		fmt.Println("conn.Read err:",err)
		return
	}
	if "ok" == string(buf[:n]) {
		//写文件内容给服务器-- 借助conn
		sendFile(conn,filePath)
	}


}

 

server.go

package main

import (
	"fmt"
	"net"
	"os"
)
func recvFile(conn net.Conn, fileName string) {
	//按照文件名创建新文件
	f, err := os.Create(fileName)
	if err != nil {
		fmt.Println("os.Create err:",err)
		return
	}
	defer f.Close()

	//从网络中读数据，写入本地文件
	buf := make([]byte,4096)
	for {
		n,_ := conn.Read(buf)
		if n == 0 {
			fmt.Println("接收文件完成。")
			return
		}
		//写入本地文件，都多少，写多少
		f.Write(buf[:n])
	}

}
func main(){
	//创建用于监听的socket
	listenner,err := net.Listen("tcp","127.0.0.1:8008")
	if err != nil{
		fmt.Println("net.Listen err: ",err)
		return
	}
	defer listenner.Close()

	//阻塞监听
	conn,err := listenner.Accept()
	if err != nil{
		fmt.Println("listenner.Accept err:",err)
		return
	}
	defer conn.Close()

	//获取文件名，保存
	buf := make([]byte,4096)
	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(conn,fileName)
}