Golang实现断点续传功能

2022-07-26 18:12:47

本文实例为大家分享了Golang实现断点续传的具体代码,供大家参考,具体内容如下1、将文件pic_src.jpg复制到pic_des.jpg文件;2、读写过程中断时,读写的中断位置会记录在pic_de...

本文实例为大家分享了golang实现断点续传的具体代码,供大家参考,具体内容如下

1、将文件pic_src.jpg复制到pic_des.jpg文件;

2、读写过程中断时,读写的中断位置会记录在“pic_des.jpgqTwJwtemp.txt”文件里;

3、下次再启动程序时,程序会接着“pic_des.jpgtemp.txt”文件里记录的中断位置,继续读、写文件,直到完成传输;

具体代码实现,如下:

package main

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

func main() {
 /*
 断点续传:
  文件传递:文件复制
  srcFile 复制到 destFile
 思路:
  边复制,边记录复制的总量(记录总复制字节数保存至tempFile)
  复制一半中断后,若再次复制,会先读取tempFile文件里记录的已完成字节总数
  然后将这个字节数作为文件读写offset(偏移量),
    来结合whence(偏移位置)来决定后面读写文件的起点位置。
  seek(offset,whence),设置指针光标的位置
  第一个参数:偏移量
  第二个参数:如何设置
   0:seekStart表示相对于文件开始,
     1:seekCurrent表示相对于当前偏移量,
     2:seek end表示相对于结束。
  */

 srcFile:="./0731/ddxc/pic_src.jpg"
 destFile:="./0731/ddxc/pic_des.jpg"
 tempFile:=destFile+"temp.txt"
 file1,err:=os.Open(srcFile)
 if err !=nil {
  log.Fatalln(err)
 }
 file2,_:=os.OpenFile(destFile,os.O_CREATE|os.O_WRONLY,os.ModePerm)
 file3,_:=os.OpenFile(tempFile,os.O_CREATE|os.O_RdwR,os.ModePerm)

 defer file1.Close()
 defer file2.Close()
 //1.读取临时文件中的数据,根据seekhttp://www.cppcns.com
 file3.Seek(0,io.SeekStart)
 bs:=make([]byte,100,100)
 n1,err:=file3.Read(bs)
 fmt.Println(n1)
 countStr:=string(bs[:n1])
 fmt.Println(countStr)
 count,_:=strconv.ParseInt(countStr,10,64)
 fmt.Println(count)

 //2. 设置读,写的偏移量
 file1.Seek(count,0) // 设置file1下一次读或者写的起点
 file2.Seek(count,0) // 设置file2下一次读或者写的起点
 data:=make([]byte,1024,1024)
 n2:=-1// 读取的数据量, 默认值
 n3:=-1//写出的数据量
 total :=int(count)//读取的总量

 for{
  //3.读取数据
  // 基于上面的起点,读取file1文件len(data)个字节,
  // n2:实际读取的字节数(小于等于len(data),将读取的字节存入data。
  n2,err=file1.Read(data)
  if err ==io.EOF{
   fmt.Println("文件复制完毕。。")
   file3.Close()
   //os.Remove(tempFile) //复制完,先不删除,验证最后存储的字节总数是不是跟复制的文件大小一致。
   break
  }
  //将数据写入到目标文件
  // 基于上面的起点,向file2文件写入len(data[:n2])个字节,也就是写入data中前n2个元素;
  // n3:实际写入的字节数(小于等于n2)。
  n3,_=file2.Write(data[:n2])
  total += n3
  //将复制总量,存储到临时文件中
  file3.Seek(0,io.SeekStart) // 设置file3的下次读写起点为源点:0点,即覆盖重写。
  file3.WriteString(strconv.Itoa(total))

  //假装断电
  //if total>1800000{
  // panic("假装断电了。。。,假装的。。。")
  //}
 }
}

 4、编程客栈检验下: 当文件复制完成后,pic_des.jpgtemp.txt文件里保存的字节数刚好等于传输文件大小,说明文件完整复制。

如下图:

Golang实现断点续传功能

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持我们。