服务端实现流程大致如下：
- 创建监听listener，程序结束时关闭。
- 阻塞等待客户端连接，程序结束时关闭conn。
- 读取客户端发送文件名。保存fileName。
- 回发“ok”给客户端做应答
- 封装函数 RecvFile接收客户端发送的文件内容。传参fileName 和conn
- 按文件名Create文件，结束时Close
- 循环Read客户端发送的文件内容，当读到EOF说明文件读取完毕。
- 将读到的内容原封不动Write到创建的文件中
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) } }
客户端实现流程大致如下：
- 提示用户输入文件名。接收文件名path（含访问路径）
- 使用os.Stat()获取文件属性，得到纯文件名（去除访问路径）
- 主动连接服务器，结束时关闭连接
- 给接收端（服务器）发送文件名conn.Write()
- 读取接收端回发的确认数据conn.Read()
- 判断是否为“ok”。如果是，封装函数SendFile() 发送文件内容。传参path和conn
- 只读Open文件, 结束时Close文件
- 循环读文件，读到EOF终止文件读取
- 将读到的内容原封不动Write给接收端（服务器）
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) } }
请发表评论