Golang实现断点续传功能

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

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

2、读写过程中断时，读写的中断位置会记录在“pic_des.jpgtemp.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.读取临时文件中的数据，根据seek
    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文件里保存的字节数刚好等于传输文件大小，说明文件完整复制。

如下图：

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