当前位置：首页 > news >正文

Go语言--传输文件

news 来源：原创 2024/9/21 3:19:36

获取文件属性`Stat()`

info,err:=os.Stat(fileName)

Stat()底层如下：

func Stat(name string) (FileInfo, error) {testlog.Stat(name)return statNolog(name)
}

FileInfo 底层如下：

type FileInfo interface {Name() string       // base name of the fileSize() int64        // length in bytes for regular files; system-dependent for othersMode() FileMode     // file mode bitsModTime() time.Time // modification timeIsDir() bool        // abbreviation for Mode().IsDir()Sys() any           // underlying data source (can return nil)
}

在这里插入图片描述

功能介绍

分别实现客户端和接收端，使得客户端发送的各种文件信息，接收方能够以相同的形式以及大小拷贝到当前目录下。

客户端

package mainimport ("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, err1 := f.Read(buf)if err1 != nil {if err1 == io.EOF {fmt.Println("文件发送完毕")} else {fmt.Println("f.Read.err1=", err1)}return}//给接收方发送内容conn.Write(buf[:n])}
}func main() {//提示传输文件fmt.Println("请输入需要传输的文件：")var path stringfmt.Scan(&path)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 intbuf := 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)}}

读取客户输入的文件路径 fmt.Scan(&path)
读取文件信息info, err := os.Stat(path)
主动连接服务器conn, err1 := net.Dial("tcp", "127.0.0.1:8000")
给接收方发送文件名_, err = conn.Write([]byte(info.Name()))
接收对方的回复，回复ok 就可以发n, err = conn.Read(buf)
发送文件内容SendFile(path, conn)
以只读方式打开f, err := os.Open(path)
读文件内容，读多少发多少buf := make([]byte, 1024*4)
给接收方发送内容conn.Write(buf[:n])

接收端

package mainimport ("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)var n int//接收多少，写多少for {n, err = conn.Read(buf)if err != nil {if err == io.EOF {fmt.Println("文件接收完毕")} else {fmt.Println("conn.Read.err=", err)}return}f.Write(buf[:n]) //往文件写入内容}}func main() {//监听listener, err := net.Listen("tcp", "127.0.0.1:8000")if err != nil {fmt.Println("net.Listen.err=", err)return}defer listener.Close()//阻塞等待用户连接conn, err1 := listener.Accept()if err1 != nil {fmt.Println("listener.Accept.err1=", 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])//回复okconn.Write([]byte("ok"))//接收文件内容RecvFile(fileName, conn)
}