1 package main
 2 
 3 import (
 4     "fmt"
 5     "io"
 6     "log"
 7     "os"
 8     "strconv"
 9     "strings"
10 )
11 
12 func HandleErr(err error) {
13     if err != nil {
14         log.Fatal(err)
15     }
16 }
17 
18 func main() {
19     /*
20         断点续传：
21             文件传递：文件复制
22 
23             "D:\\Go\\web2\\test.png"
24 
25             复制到当前的工程下：
26 
27         思路：边复制，边记录复制的总量
28 
29     */
30 
31     srcFile := "D:\\Go\\web2\\test.png"
32     destFile := srcFile[strings.LastIndex(srcFile, "\\")+1:]
33     fmt.Println(destFile) // test.png
34 
35     tempFile := destFile + "temp.txt"
36     fmt.Println(tempFile) // test.pngtemp.txt
37 
38     file1, err := os.Open(srcFile)
39     HandleErr(err)
40     file2, err := os.OpenFile(destFile, os.O_CREATE|os.O_WRONLY, os.ModePerm)
41     HandleErr(err)
42     file3, err := os.OpenFile(tempFile, os.O_CREATE|os.O_RDWR, os.ModePerm)
43     HandleErr(err)
44 
45     defer file1.Close()
46     defer file2.Close()
47 
48     // step1：先读取临时文件中的数据，再seek
49     file3.Seek(0, io.SeekStart)
50     bs := make([]byte, 100, 100)
51     n1, err := file3.Read(bs)
52     // HandleErr(err)
53     countStr := string(bs[:n1])
54     fmt.Println("countstr: ", countStr)
55     count, err := strconv.ParseInt(countStr, 10, 64)
56     // HandleErr(err)
57     fmt.Println("count: ", count)
58 
59     //step2：设置读，写的位置：
60     file1.Seek(count, io.SeekStart)
61     file2.Seek(count, io.SeekStart)
62     data := make([]byte, 1024, 1024)
63     n2 := -1            // 读取的数据量
64     n3 := -1            // 写入的数据量
65     total := int(count) // 读取的总量
66 
67     //step3：复制文件
68     for {
69         n2, err = file1.Read(data)
70         if err == io.EOF || n2 == 0 {
71             fmt.Println("文件复制完毕。。", total)
72             file3.Close()
73             os.Remove(tempFile)
74             break
75         }
76         n3, err = file2.Write(data[:n2])
77         total += n3
78 
79         // 将复制的总量，存储到临时文件中
80         file3.Seek(0, io.SeekStart)
81         file3.WriteString(strconv.Itoa(total))
82 
83         fmt.Printf("total:%d\n", total)
84 
85         // 假装断电
86         if total > 8000 {
87             panic("假装断电了。。。")
88         }
89 
90     }
91 
92 }