Golang断点续传功能
开心码农1号 人气:01、将文件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文件里保存的字节数刚好等于传输文件大小,说明文件完整复制。
如下图:
加载全部内容