golang socket断点续传大文件的实现方法

本篇文章主要是结合我之前面试的各种经历和实战开发中遇到的问题解决经验整理的，希望这篇《golang socket断点续传大文件的实现方法》对你有很大帮助！欢迎收藏，分享给更多的需要的朋友学习~

在日常编程中，我们肯定会遇到用socket传送文件内容，如果是大文件的，总不能传送到一半因某原因断掉了，又从新传送文件内容吧。对，我们需要续传，也就是接着上次传送的位置继续发送文件内容。

续传的话，其实并不难，我理解的思路大概如下：

客户端发送消息询问服务端，你上次接收到的文件内容位置

服务端告诉客户端上次接收到的文件内容位置

客户端就从上次断点的位置继续发送文件内容

客户端发送文件内容完毕后通知服务端，然后断开连接

下面我们看看代码的实现

服务端

// file name: server.go

package main

import (
 "os"
 "io"
 "net"
 "log"
 "strconv"
 // "time"
)

// 把接收到的内容append到文件
func writeFile(content []byte) {
 if len(content) != 0 {
  fp, err := os.OpenFile("test_1.txt", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0755)
  defer fp.Close()
  if err != nil {
   log.Fatalf("open file faild: %s\n", err)
  }
  _, err = fp.Write(content)
  if err != nil {
   log.Fatalf("append content to file faild: %s\n", err)
  }
  log.Printf("append content: 【%s】 success\n", string(content))
 }
}

// 获取已接收内容的大小
// (断点续传需要把已接收内容大下通知客户端从哪里开始发送文件内容)
func getFileStat() int64 {
 fileinfo, err := os.Stat("test_1.txt")
 if err != nil {
  // 如果首次没有创建test_1.txt文件，则直接返回0
  // 告诉客户端从头开始发送文件内容
  if os.IsNotExist(err) {
   log.Printf("file size: %d\n", 0)
   return int64(0)
  }
  log.Fatalf("get file stat faild: %s\n", err)
 }
 log.Printf("file size: %d\n", fileinfo.Size())
 return fileinfo.Size() 
}

func serverConn(conn net.Conn) {
 defer conn.Close()
 for {
  var buf = make([]byte, 10)
  n, err := conn.Read(buf)
  if err != nil {
   if err == io.EOF {
    log.Println("server io EOF\n")
    return
   }
   log.Fatalf("server read faild: %s\n", err)
  }
  log.Printf("recevice %d bytes, content is 【%s】\n", n, string(buf[:n]))
  // 判断客户端发送过来的消息
  // 如果是'start-->‘则表示需要告诉客户端从哪里开始读取文件数据发送
  switch string(buf[:n]) {
  case "start-->":
   off := getFileStat()
   // int conver string
   stringoff := strconv.FormatInt(off, 10)
   _, err = conn.Write([]byte(stringoff))
   if err != nil {
    log.Fatalf("server write faild: %s\n", err)
   }
   continue
  case "

<p><strong>客户端</strong></p>

<pre class="brush:plain;">
// file name: client.go

package main

import (
 "os"
 "io"
 "net"
 "log"
 "time"
 "strconv"
)

// 获取服务端发送的消息
func clientRead(conn net.Conn) int {
 buf := make([]byte, 5)
 n, err := conn.Read(buf)
 if err != nil {
  log.Fatalf("receive server info faild: %s\n", err)
 }
 // string conver int
 off, err := strconv.Atoi(string(buf[:n]))
 if err != nil {
  log.Fatalf("string conver int faild: %s\n", err)
 }
 return off
}

// 发送消息到服务端
func clientWrite(conn net.Conn, data []byte) {
 _, err := conn.Write(data)
 if err != nil {
  log.Fatalf("send 【%s】 content faild: %s\n", string(data), err)
 }
 log.Printf("send 【%s】 content success\n", string(data))
}

// client conn
func clientConn(conn net.Conn) {
 defer conn.Close()

 // 发送"start-->"消息通知服务端，我要开始发送文件内容了
 // 你赶紧告诉我你那边已经接收了多少内容,我从你已经接收的内容处开始继续发送
 clientWrite(conn, []byte("start-->"))
 off := clientRead(conn)

 // send file content
 fp, err := os.OpenFile("test.txt", os.O_RDONLY, 0755)
 if err != nil {
  log.Fatalf("open file faild: %s\n", err)
 }
 defer fp.Close()

 // set file seek
 // 设置从哪里开始读取文件内容
 _, err = fp.Seek(int64(off), 0)
 if err != nil {
  log.Fatalf("set file seek faild: %s\n", err)
 }
 log.Printf("read file at seek: %d\n", off)

 for {
  // 每次发送10个字节大小的内容
  data := make([]byte, 10)
  n, err := fp.Read(data)
  if err != nil {
   if err == io.EOF {
    // 如果已经读取完文件内容
    // 就发送'

<p>客户端读取文件test.txt内容发送到服务端，服务端把接收到的文件内容保存在test_1.txt文件中。我们模拟断点续传的方式是：</p>
<p>第一次先发送test.txt文件内容到服务端</p>
<p>修改test.txt文件，加一些内容</p>
<p>再次运行server socket以及client socket，观察客户端是不是只发送新增的文件内容到服务端<br></p>

<pre class="brush:plain;">
# 假设我的test.txt文件有以下内容
$ cat test.txt
hello golang.

# 先运行server socket再运行client socket（分别在两个终端窗口运行）
$ go run server.go
$ go run client.go

# 服务端会输出以下内容
2018/04/05 23:37:13 waiting accept.
2018/04/05 23:37:15 recevice 8 bytes, content is 【start-->】
2018/04/05 23:37:15 file size: 0
2018/04/05 23:37:15 recevice 10 bytes, content is 【hello gola】
2018/04/05 23:37:15 append content: 【hello gola】 success
2018/04/05 23:37:15 recevice 2 bytes, content is 【n.】
2018/04/05 23:37:15 append content: 【n.】 success
2018/04/05 23:37:16 recevice 6 bytes, content is 【】 content success
2018/04/05 23:37:15 read file at seek: 0
2018/04/05 23:37:15 send 【hello gola】 content success
2018/04/05 23:37:15 send 【n.】 content success
2018/04/05 23:37:16 send 【】
2018/04/05 23:44:34 file size: 12
2018/04/05 23:44:34 recevice 10 bytes, content is 【
hello pyt】
2018/04/05 23:44:34 append content: 【
hello pyt】 success
2018/04/05 23:44:34 recevice 4 bytes, content is 【hon.】
2018/04/05 23:44:34 append content: 【hon.】 success
2018/04/05 23:44:35 recevice 6 bytes, content is 【 信息后会获取上次接收到文件内容位置，并通知客户端（这里file size 是12）

# 客户端会输出以下内容
2018/04/05 23:44:34 send 【start-->】 content success
2018/04/05 23:44:34 read file at seek: 12
2018/04/05 23:44:34 send 【
hello pyt】 content success
2018/04/05 23:44:34 send 【hon.】 content success
2018/04/05 23:44:35 send 【

<p>以上这篇golang socket断点续传大文件的实现方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持golang学习网。</p><p>今天关于《golang socket断点续传大文件的实现方法》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于golang的内容请关注golang学习网公众号！</p>