Golang:实现断点续传

金刀大菜牙 · · 1561 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

\color{red}{喜欢本篇,欢迎点赞分享}

Seeker 接口


Seeker 是包装基本 Seek 方法的接口。

type Seeker interface {
    Seek(offset int64, whence int) (int64, error)
}

seek(offset,whence),设置指针光标的位置,随机读写文件:
第一个参数:偏移量
​第二个参数:如何设置
0:seekStart表示相对于文件开始,
​1:seekCurrent表示相对于当前偏移量,
​2:seek end表示相对于结束。

const (
    SeekStart   = 0 // seek relative to the origin of the file
    SeekCurrent = 1 // seek relative to the current offset
    SeekEnd     = 2 // seek relative to the end
)

我们要读取本地 /Users/ruby/Documents/pro/a 目录下的 aa.txt 文件,文件中的内容是:abcdefghij 这几个字符。



示例代码:

package main

import (
    "os"
    "fmt"
    "io"
)

func main() {
    /*
    seek(offset,whence),设置指针光标的位置
    第一个参数:偏移量
    第二个参数:如何设置
        0:seekStart表示相对于文件开始,
        1:seekCurrent表示相对于当前偏移量,
        2:seek end表示相对于结束。

    const (
    SeekStart   = 0 // seek relative to the origin of the file
    SeekCurrent = 1 // seek relative to the current offset
    SeekEnd     = 2 // seek relative to the end
)

    随机读取文件:
        可以设置指针光标的位置
     */

    file,_:=os.OpenFile("/Users/ruby/Documents/pro/a/aa.txt",os.O_RDWR,0)
    defer file.Close()
    bs :=[]byte{0}

    file.Read(bs)
    fmt.Println(string(bs))

    file.Seek(4,io.SeekStart)
    file.Read(bs)
    fmt.Println(string(bs))
    file.Seek(2,0) //也是SeekStart
    file.Read(bs)
    fmt.Println(string(bs))

    file.Seek(3,io.SeekCurrent)
    file.Read(bs)
    fmt.Println(string(bs))

    file.Seek(0,io.SeekEnd)
    file.WriteString("ABC")
}

运行结果:



本地文件:


断点续传


首先思考几个问题
Q1:如果你要传的文件,比较大,那么是否有方法可以缩短耗时?
Q2:如果在文件传递过程中,程序因各种原因被迫中断了,那么下次再重启时,文件是否还需要重头开始?
Q3:传递文件的时候,支持暂停和恢复么?即使这两个操作分布在程序进程被杀前后。
通过断点续传可以实现,不同的语言有不同的实现方式。我们看看 Go 语言中,通过 Seek() 方法如何实现:
先说一下思路:想实现断点续传,主要就是记住上一次已经传递了多少数据,那我们可以创建一个临时文件,记录已经传递的数据量,当恢复传递的时候,先从临时文件中读取上次已经传递的数据量,然后通过Seek()方法,设置到该读和该写的位置,再继续传递数据。
示例代码:

package main

import (
    "fmt"
    "os"
    "strconv"
    "io"
)

func main() {
    /*
    断点续传:
        文件传递:文件复制
            /Users/ruby/Documents/pro/a/guliang.jpeg

        复制到
            guliang4.jpeg

    思路:
        边复制,边记录复制的总量
     */

    srcFile:="/Users/ruby/Documents/pro/a/guliang.jpeg"
    destFile:="guliang4.jpeg"
    tempFile:=destFile+"temp.txt"
    //fmt.Println(tempFile)
    file1,_:=os.Open(srcFile)
    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.Atoi(countStr)
    count,_:=strconv.ParseInt(countStr,10,64)
    fmt.Println(count)

    //2. 设置读,写的偏移量
    file1.Seek(count,0)
    file2.Seek(count,0)
    data:=make([]byte,1024,1024)
    n2:=-1// 读取的数据量
    n3:=-1//写出的数据量
    total :=int(count)//读取的总量

    for{
        //3.读取数据
        n2,err=file1.Read(data)
        if err ==io.EOF{
            fmt.Println("文件复制完毕。。")
            file3.Close()
            os.Remove(tempFile)
            break
        }
        //将数据写入到目标文件
        n3,_=file2.Write(data[:n2])
        total += n3
        //将复制总量,存储到临时文件中
        file3.Seek(0,io.SeekStart)
        file3.WriteString(strconv.Itoa(total))

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

加入这里(https://t.zsxq.com/zBAQNbu),每天都有 Golang 初级到高级进阶视频更新

\color{red}{喜欢本篇,欢迎点赞分享}


有疑问加站长微信联系(非本文作者)

本文来自:简书

感谢作者:金刀大菜牙

查看原文:Golang:实现断点续传

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

1561 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传