Go语言Seeker接口与文件断点续传实战教程

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

package main
import (
    "fmt"
    "io"
    "os"
)
func main() {
    // 读取文件
    file, _ := os.OpenFile("F:\\goworks\\src\\jingtian\\yufa\\io操作\\a.txt", os.O_RDWR, os.ModePerm)
    // defer close
    defer file.Close()
    // 测试seek
    // 相对开始位置。io.SeekStart
    // 相对于文件末尾， io.SeekEnd
    // func (f *File) Seek(offset int64, whence int) (ret int64, err error)
    //相对于开始位置，光标偏移两个字节
    file.Seek(2, io.SeekStart)
    //创建一个字节的buffer
    buf := []byte{0}
    file.Read(buf)
    fmt.Println(string(buf)) // n
    //Read读了一个字节， 光标现在在3这个位置
    // 相对于当前位置
    file.Seek(3, io.SeekCurrent)
    file.Read(buf)
    fmt.Println(string(buf)) // a
    // 在结尾追加内容
    //相对于结束位置偏移0，光标就设在了结束位置
    file.Seek(0, io.SeekEnd)
    //写入内容
    file.WriteString("hahahaha")
}

package main
import (
    "fmt"
    "io"
    "os"
    "strconv"
)
// 断点续传
func main() {
    // 传输源文件地址
    srcFile := "D:\\downloads\\mingw.7z"
    // 传输的目标位置
    destFile := "F:\\goworks\\src\\jingtian\\yufa\\io操作\\server\\mingw.7z"
    // 临时记录文件
    tempFile := "F:\\goworks\\src\\jingtian\\yufa\\io操作\\temp.txt"
    // 创建对应的file对象，连接起来
    file1, _ := os.Open(srcFile)
    file2, _ := os.OpenFile(destFile, os.O_CREATE|os.O_RDWR, os.ModePerm)
    file3, _ := os.OpenFile(tempFile, os.O_CREATE|os.O_RDWR, os.ModePerm)
    defer file1.Close()
    defer file2.Close()
    fmt.Println("file1/2/3 文件连接建立完毕")
    // 1、读取temp.txt
    file3.Seek(0, io.SeekStart)
    buf := make([]byte, 1024)
    n, _ := file3.Read(buf) //这里的n是读取file3中的字符的个数，比如1024，得到的n是4.所以要借助string转
    //查看返回的n的数据类型
    fmt.Printf("查看n的数据类型%T\n", n)
    fmt.Println("n的值为", n)
    // 2、先转换成string，然后再转换成数字。
    countStr := string(buf[:n])
    count, _ := strconv.ParseInt(countStr, 10, 64)
    fmt.Println("temp.txt中记录的值为：", count) // 5120
    // 3、设置读写的偏移量,offset是int64数据类型
    file1.Seek(count, io.SeekStart)
    file2.Seek(count, io.SeekStart)
    fmt.Println("file1/2 光标已经移动到了目标位置")
    // 4、开始读写（复制、上传）
    bufData := make([]byte, 1024)
    // 5、需要记录读取了多少个字节
    total := int(count)
    for {
        fmt.Println("传输了,", total)
        // 读取数据
        readNum, err := file1.Read(bufData)
        if err == io.EOF || readNum == 0 { // file1 读取完毕了
            fmt.Println("文件传输完毕了")
            //上传完文件再关闭临时文件file3
            file3.Close()
            os.Remove(tempFile)
            break
        }
        // 向目标文件中写入数据，返回写的字节数和错误
        writeNum, err := file2.Write(bufData[:readNum])
        // 将写入数据放到 total中, 在这里total 就是传输的进度
        total = total + writeNum
        // temp.txt 存放临时记录数据
        file3.Seek(0, io.SeekStart) // 将光标重置到开头
        //将数字转换成字符串写入，这里total逐渐变大不存在覆盖不完的问题。如果存在覆盖不完问题，使用os.Truncate(fileName, 0)来清空文件内容
        // os.Truncate(fileName, 0) 截取指定长度字节的内容，其余内容会被删除
        file3.WriteString(strconv.Itoa(total))
        //模拟断电
        //if total > 10000 {
        //    panic("断电了")
        //}
    }
}