Go语言之TCP文件传输

小山儿 發表於 2019-7-7 19:49:00

服务端实现流程大致如下：
<ol>
<li>创建监听listener，程序结束时关闭。</li>
<li>阻塞等待客户端连接，程序结束时关闭conn。</li>
<li>读取客户端发送文件名。保存fileName。</li>
<li>回发“ok”给客户端做应答</li>
<li>封装函数 RecvFile接收客户端发送的文件内容。传参fileName 和conn</li>
<li>按文件名Create文件，结束时Close</li>
<li>循环Read客户端发送的文件内容，当读到EOF说明文件读取完毕。</li>
<li>将读到的内容原封不动Write到创建的文件中</li>
</ol>
<div class="cnblogs_code">
<pre>package main

import (
"fmt"
"net"
"os"
"runtime"
)

func Handler(conn net.Conn) {
buf := make([]byte, 2048)
//读取客户端发送的内容
n, err := conn.Read(buf)
if err != nil {
 fmt.Println(err)
 return
}
fileName := string(buf[:n])
//获取客户端ip+port
addr := conn.RemoteAddr().String()
fmt.Println(addr + ": 客户端传输的文件名为--" + fileName)
//告诉客户端已经接收到文件名
conn.Write([]byte("ok"))
//创建文件
f, err := os.Create(fileName)
if err != nil {
 fmt.Println(err)
 return
}
//循环接收客户端传递的文件内容
for {
 buf := make([]byte, 2048)
 n, _ := conn.Read(buf)
 //结束协程
 if string(buf[:n]) == "finish" {
 fmt.Println(addr + ": 协程结束")
 runtime.Goexit()
 }
 f.Write(buf[:n])
}
defer conn.Close()
defer f.Close()
}

func main() {
//创建tcp监听
listen, err := net.Listen("tcp", ":8000")
if err != nil {
 fmt.Println(err)
 return
}
defer listen.Close()

for {
 //阻塞等待客户端
 conn, err := listen.Accept()
 if err != nil {
 fmt.Println(err)
 return
 }
 //创建协程
 go Handler(conn)
}
}</pre>
</div>
客户端实现流程大致如下：
<ol>
<li>提示用户输入文件名。接收文件名path（含访问路径）</li>
<li>使用os.Stat()获取文件属性，得到纯文件名（去除访问路径）</li>
<li>主动连接服务器，结束时关闭连接</li>
<li>给接收端（服务器）发送文件名conn.Write()</li>
<li>读取接收端回发的确认数据conn.Read()</li>
<li>判断是否为“ok”。如果是，封装函数SendFile() 发送文件内容。传参path和conn</li>
<li>只读Open文件, 结束时Close文件</li>
<li>循环读文件，读到EOF终止文件读取</li>
<li>将读到的内容原封不动Write给接收端（服务器）</li>
</ol>
<div class="cnblogs_code">
<pre>package main

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

//发送文件到服务端
func SendFile(filePath string, fileSize int64, conn net.Conn) {
f, err := os.Open(filePath)
if err != nil {
 fmt.Println(err)
 return
}
defer f.Close()
var count int64
for {
 buf := make([]byte, 2048)
 //读取文件内容
 n, err := f.Read(buf)
 if err != nil && io.EOF == err {
 fmt.Println("文件传输完成")
 //告诉服务端结束文件接收
 conn.Write([]byte("finish"))
 return
 }
 //发送给服务端
 conn.Write(buf[:n])

 count += int64(n)
 sendPercent := float64(count) / float64(fileSize) * 100
 value := fmt.Sprintf("%.2f", sendPercent)
 //打印上传进度
 fmt.Println("文件上传：" + value + "%")
}
}

func main() {
fmt.Print("请输入文件的完整路径：")
//创建切片，用于存储输入的路径
var str string
fmt.Scan(&str)
//获取文件信息
fileInfo, err := os.Stat(str)
if err != nil {
 fmt.Println(err)
 return
}
//创建客户端连接
conn, err := net.Dial("tcp", ":8000")
if err != nil {
 fmt.Println(err)
 return
}
defer conn.Close()
//文件名称
fileName := fileInfo.Name()
//文件大小
fileSize := fileInfo.Size()
//发送文件名称到服务端
conn.Write([]byte(fileName))
buf := make([]byte, 2048)
//读取服务端内容
n, err := conn.Read(buf)
if err != nil {
 fmt.Println(err)
 return
}
revData := string(buf[:n])
if revData == "ok" {
 //发送文件数据
 SendFile(str, fileSize, conn)
}
}</pre>
</div>
  
来源：https://www.cnblogs.com/yang-2018/p/11147418.html

頁: [1]

琼殿技术论坛's Archiver

Go语言 之TCP文件传输

Go语言之TCP文件传输