| 這篇文章主要為大家詳細介紹了Golang實作檔案傳輸功能，文中範例程式碼介紹的非常詳細，具有一定的參考價值，有興趣的夥伴們可以參考一下 |
本文實例為大家分享了Golang實作檔案傳輸的具體程式碼，供大家參考，具體內容如下
借助TCP完成檔案的傳輸，基本想法如下：
1、發送方（客戶端）向服務端發送檔案名，服務端保存該檔案名稱。
2、接收者（服務端）向客戶端傳回一個訊息ok，確認檔案名稱保存成功。
3、發送方（客戶端）收到訊息後，開始傳送文件資料到服務端。
4、接收者（服務端）讀取檔案內容，寫入到先前儲存好的檔案中。
首先取得檔案名稱。借助os包中的stat()函數來取得檔案屬性資訊。在函數傳回的檔案屬性中包含檔案名稱和檔案大小。 Stat參數name傳入的是檔案存取的絕對路徑。 FileInfo中的Name()函數可以將檔案名稱單獨提取出來。
func Stat(name string) (FileInfo, error)
type FileInfo interface {
Name() string
Size() int64
Mode() FileMode
ModTime() time.Time
IsDir() bool
Sys() interface{}
}發送端：
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, 1024)
for {
n, err := f.Read(buf)
if err != nil {
if err == io.EOF {
fmt.Println("发送文件完成。")
} 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:8000")
if err != nil {
fmt.Println("net.Dial err:", err)
return
}
defer conn.Close()
// 发送文件名给 接收端
_, err = conn.Write([]byte(fileName))
if err != nil {
fmt.Println("conn.Write err:", err)
return
}
// 读取服务器回发的 OK
buf := make([]byte, 1024)
n, err := conn.Read(buf)
if err != nil {
fmt.Println("conn.Read err:", err)
return
}
if "ok" == string(buf[:n]) {
// 写文件内容给服务器——借助conn
sendFile(conn, filePath)
}
}接收端：
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, 1024)
for {
n, _ := conn.Read(buf)
if n == 0 {
fmt.Println("接收文件完成。")
return
}
// 写入本地文件，读多少，写多少。
f.Write(buf[:n])
}
}
func main() {
// 创建用于监听的socket
listener, err := net.Listen("tcp", "127.0.0.1:8000")
if err != nil {
fmt.Println(" net.Listen err:", err)
return
}
defer listener.Close()
fmt.Println("接收端启动成功，等待发送端发送文件！")
// 阻塞监听
conn, err := listener.Accept()
if err != nil {
fmt.Println(" listener.Accept() err:", err)
return
}
defer conn.Close()
// 获取文件名，保存
buf := make([]byte, 1024)
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)
}以上就是本文的全部內容，希望對大家的學習有所幫助。
推薦學習：《go影片教學》
以上是Golang怎麼實現文件傳輸功能的詳細內容。更多資訊請關注PHP中文網其他相關文章！
