本文实例为大家分享了Golang实现断点续传的具体代码,供大家参考,具体内容如下
1、将文件pic_src.jpg复制到pic_des.jpg文件;
2、读写过程中断时,读写的中断位置会记录在“pic_des.jpgtemp.txt”文件里;
3、下次再启动程序时,程序会接着“pic_des.jpgtemp.txt”文件里记录的中断位置,继续读、写文件,直到完成传输;
具体代码实现,如下:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
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文件里保存的字节数刚好等于传输文件大小,说明文件完整复制。
如下图:
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/yanghaitao5000/article/details/119280485