C# FileStream实现多线程断点续传
一、前言
网上有许多的多线程断点续传操作,但总是写的很云里雾里,或者写的比较坑长。由于这几个月要负责公司的在线升级项目,所以正好顺便写了一下
代码如下:
usingSystem;
usingSystem.Collections.Generic;
usingSystem.IO;
usingSystem.Threading.Tasks;
namespaceTestCenter
{
classProgram
{
staticvoidMain(string[]args)
{
stringLocalSavePath=@"E:\Test\TestFile\Local\1.msi";//本地目标文件路径
FileInfoSeverFilePath=newFileInfo(@"E:\Test\TestFile\Server\1.msi");//服务器待文件路径
longFileLength=SeverFilePath.Length;//待下载文件大小
Console.WriteLine("StartConfiguration");
intPackCount=0;//初始化数据包个数
longPackSize=1024000;//数据包大小
if(FileLength%PackSize>0)
{
PackCount=(int)(FileLength/PackSize)+1;
}
else
{
PackCount=(int)(FileLength/PackSize);
}
Console.WriteLine("StartRecieve");
vartasks=newTask[PackCount];//多线程任务
for(intindex=0;index
{
stringtempfilepath=@"E:\Test\TestFile\Temp\"+"QS_"+Threadindex+"_"+PackCount;//临时文件路径
using(FileStreamtempstream=newFileStream(tempfilepath,FileMode.Create,FileAccess.Write,FileShare.Write))
{
intlength=(int)Math.Min(PackSize,FileLength-Threadindex*PackSize);
varbytes=GetFile(Threadindex*PackCount,length);
tempstream.Write(bytes,0,length);
tempstream.Flush();
tempstream.Close();
tempstream.Dispose();
}
});
tasks[Threadindex]=task;
task.Start();
}
Task.WaitAll(tasks);//等待所有线程完成
Console.WriteLine("RecieveEnd");
//检测有哪些数据包未下载
Console.WriteLine("StartCompare");
DirectoryInfoTempDir=newDirectoryInfo(@"E:\Test\TestFile\temp");//临时文件夹路径
ListComparefiles=newList();
for(inti=0;i0)
{
foreach(stringcom_indexinComparefiles)
{
stringtempfilepath=@"E:\Test\TestFile\Temp\"+"QS_"+com_index+"_"+PackCount;
using(FileStreamCompstream=newFileStream(tempfilepath,FileMode.Create,FileAccess.Write,FileShare.Write))
{
intlength=(int)Math.Min(PackSize,FileLength-Convert.ToInt32(com_index)*PackSize);
varbytes=GetFile(Convert.ToInt32(com_index)*PackCount,length);
Compstream.Write(bytes,0,length);
Compstream.Flush();
Compstream.Close();
Compstream.Dispose();
}
}
}
Console.WriteLine("CompareEnd");
//准备将临时文件融合并写到1.msi中
Console.WriteLine("StartWrite");
using(FileStreamwritestream=newFileStream(LocalSavePath,FileMode.Create,FileAccess.Write,FileShare.Write))
{
foreach(FileInfoTempfileinTempDir.GetFiles())
{
using(FileStreamreadTempStream=newFileStream(Tempfile.FullName,FileMode.Open,FileAccess.Read,FileShare.ReadWrite))
{
longonefileLength=Tempfile.Length;
byte[]buffer=newbyte[Convert.ToInt32(onefileLength)];
readTempStream.Read(buffer,0,Convert.ToInt32(onefileLength));
writestream.Write(buffer,0,Convert.ToInt32(onefileLength));
}
}
writestream.Flush();
writestream.Close();
writestream.Dispose();
}
Console.WriteLine("WriteEnd");
//删除临时文件
Console.WriteLine("StartDeleteTempFiles");
foreach(FileInfoTempfileinTempDir.GetFiles())
{
Tempfile.Delete();
}
Console.WriteLine("DeleteSuccess");
Console.ReadKey();
}
//这个方法可以放到Remoting或者WCF服务中去,然后本地调用该方法即可实现多线程断点续传
publicstaticbyte[]GetFile(intstart,intlength)
{
stringSeverFilePath=@"E:\Test\TestFile\Server\1.msi";
using(FileStreamServerStream=newFileStream(SeverFilePath,FileMode.Open,FileAccess.Read,FileShare.ReadWrite,1024*80,true))
{
byte[]buffer=newbyte[length];
ServerStream.Position=start;
//ServerStream.Seek(start,SeekOrigin.Begin);
ServerStream.Read(buffer,0,length);
returnbuffer;
}
}
}
}
二、讨论
1)需要注意的是第44行,不能直接使用index变量在Task()里进行操作,而是要将它赋给Threadindex,让Threadindex在Task()里,不然会直接报错,为什么呢?查看链接
2)70至108行代码可以在外面再套一层while循环,循环检测临时文件是否下完整了,然后再定义一个检测最大上限,超过这个上限就放弃本次更新,当用户的网络恢复正常后下次再做更新操作。所以说放临时文件的文件夹最好要包含版本信息,不会把2.0.0的临时文件和1.0.0的临时文件搞混。
3)FileStream.Position与FileStream.Seek(longoffset,SeekOriginseekorigin)的作用都是获取流的指针位置,当文件路径使用绝对路径时使用Position;相对路径时使用Seek方法,查看链接
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持毛票票。