亲宝软件园·资讯

展开

Golang断点续传功能

开心码农1号 人气:0

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文件里保存的字节数刚好等于传输文件大小,说明文件完整复制。

如下图:

加载全部内容

相关教程
猜你喜欢
用户评论