下面由golang教學欄位給大家分享golang實作檔案傳輸小demo ，希望對需要的朋友有幫助！
取得檔案資訊需要用到os. Stat接口，發送檔案前開啟接收者（服務端），啟動客戶端先傳送檔案名稱給接收者，接收者收到檔案名稱回傳確認訊息"ok ",才讀取本地文件 發送給接收者。
發送者：
package main
import ( "fmt"
"io"
"net"
"os")
func main() {
fmt.Println("请输入接收者地址(IP:PORT)：") var addr string
fmt.Scan(&addr)
fmt.Println("请输入需要传输的文件：") var path string
fmt.Scan(&path)
info, errf := os.Stat(path) if errf != nil{
fmt.Println("os.Stat errf =", errf) return
}
conn, err := net.Dial("tcp", addr) if err != nil{
fmt.Println("net.Dial err =",err) return
}
defer conn.Close()
_, err = conn.Write([]byte(info.Name())) if err != nil{
fmt.Println("conn.Write info.Name err =",err) return
} var n int
buf := make([]byte, 1024)
n, err = conn.Read(buf) if err != nil{
fmt.Println("conn.Read ok err =", err) return
} if "ok" == string(buf[:n]){
fmt.Println("ok")
SendFile(path, conn)
}
}
func SendFile(path string, conn net.Conn){
file , err := os.Open(path) if err != nil{
fmt.Println("os.Open err =", err) return
}
defer file.Close()
buf := make([]byte, 1024 * 4) for {
n, err := file.Read(buf) if err != nil{ if err == io.EOF{
fmt.Println("文件发送完毕")
} else{
fmt.Println("file.Read err =",err)
} return
} if n == 0{
fmt.Println("文件发送完毕") break
}
conn.Write(buf[:n])
}
}接收者：
package main
import ( "fmt"
"io"
"net"
"os")
func main() {
fmt.Println("请你的地址(IP:PORT)：") var addr string
fmt.Scan(&addr)
listenner, err := net.Listen("tcp", addr) if err != nil{
fmt.Println("net.Listen err =", err) return
}
defer listenner.Close()
conn, errl := listenner.Accept() if errl != nil{
fmt.Println("listenner.Accept err =", errl) return
} var n int
buf := make([]byte, 1024)
n, err = conn.Read(buf) if err != nil{
fmt.Println("conn.Read fileName err =", err) return
}
fileName := string(buf[:n])
n, err = conn.Write([]byte("ok")) if err != nil{
fmt.Println("conn.Write ok err =", err) return
}
RecvFile(fileName, conn)
}
func RecvFile(fileName string, conn net.Conn){
file , err := os.Create(fileName) if err != nil{
fmt.Println("os.Create err =", err) return
}
defer file.Close()
buf := make([]byte, 1024 * 4) for{
n, err := conn.Read(buf) if err != nil{ if err == io.EOF{
fmt.Println("文件接收完成")
} else {
fmt.Println("conn.Read err =", err)
} return
}
n, err = file.Write(buf[:n]) if err != nil{
fmt.Println("file.Write err =", err) break
}
}
}以上是分享golang實作檔案傳輸小demo的詳細內容。更多資訊請關注PHP中文網其他相關文章！
