375 lines
15 KiB
C#
375 lines
15 KiB
C#
|
using System.Net;
|
|||
|
|
using System.IO;
|
|||
|
|
|
|||
|
|
namespace Zerolauncher.Manager
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
public class DownloadManager
|
|||
|
|
{
|
|||
|
|
/// <summary>
|
|||
|
|
/// 主线程
|
|||
|
|
/// </summary>
|
|||
|
|
private SynchronizationContext _mainThreadSynContext;
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 下载网址
|
|||
|
|
/// </summary>
|
|||
|
|
public string Url;
|
|||
|
|
|
|||
|
|
public event Action<Exception> OnError;
|
|||
|
|
|
|||
|
|
private static object errorlock = new object();
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 主要用于关闭线程
|
|||
|
|
/// </summary>
|
|||
|
|
private bool _isDownload = false;
|
|||
|
|
public DownloadManager(string url)
|
|||
|
|
{
|
|||
|
|
// 主线程赋值
|
|||
|
|
_mainThreadSynContext = SynchronizationContext.Current;
|
|||
|
|
// 突破Http协议的并发连接数限制
|
|||
|
|
ServicePointManager.DefaultConnectionLimit = 512;
|
|||
|
|
|
|||
|
|
Url = url;
|
|||
|
|
}
|
|||
|
|
/// <summary>
|
|||
|
|
/// 查询文件大小
|
|||
|
|
/// </summary>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
public long GetFileSize()
|
|||
|
|
{
|
|||
|
|
HttpWebRequest request;
|
|||
|
|
HttpWebResponse response;
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
request = (HttpWebRequest)WebRequest.CreateHttp(new Uri(Url));
|
|||
|
|
request.Method = "HEAD";
|
|||
|
|
response = (HttpWebResponse)request.GetResponse();
|
|||
|
|
// 获得文件长度
|
|||
|
|
long contentLength = response.ContentLength;
|
|||
|
|
|
|||
|
|
response.Close();
|
|||
|
|
request.Abort();
|
|||
|
|
|
|||
|
|
return contentLength;
|
|||
|
|
}
|
|||
|
|
catch (Exception ex)
|
|||
|
|
{
|
|||
|
|
onError(ex);
|
|||
|
|
// throw;
|
|||
|
|
return -1;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
/// <summary>
|
|||
|
|
/// 异步查询文件大小
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="onTrigger"></param>
|
|||
|
|
public void GetFileSizeAsyn(Action<long> onTrigger = null)
|
|||
|
|
{
|
|||
|
|
ThreadStart threadStart = new ThreadStart(() =>
|
|||
|
|
{
|
|||
|
|
PostMainThreadAction<long>(onTrigger, GetFileSize());
|
|||
|
|
});
|
|||
|
|
Thread thread = new Thread(threadStart);
|
|||
|
|
thread.Start();
|
|||
|
|
}
|
|||
|
|
/// <summary>
|
|||
|
|
/// 多线程下载文件至本地
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="threadCount">线程总数</param>
|
|||
|
|
/// <param name="filePath">保存文件路径</param>
|
|||
|
|
/// <param name="onDownloading">下载过程回调（已下载文件大小、总文件大小）</param>
|
|||
|
|
/// <param name="onTrigger">下载完毕回调（下载文件数据）</param>
|
|||
|
|
public void DownloadToFile(int threadCount, string filePath, Action<long, long> onDownloading = null, Action<byte[]> onTrigger = null)
|
|||
|
|
{
|
|||
|
|
_isDownload = true;
|
|||
|
|
|
|||
|
|
long csize = 0; //已下载大小
|
|||
|
|
int ocnt = 0; //完成线程数
|
|||
|
|
|
|||
|
|
|
|||
|
|
// 下载逻辑
|
|||
|
|
GetFileSizeAsyn((size) =>
|
|||
|
|
{
|
|||
|
|
if (size == -1) return;
|
|||
|
|
// 准备工作
|
|||
|
|
var tempFilePaths = new string[threadCount];
|
|||
|
|
var tempFileFileStreams = new FileStream[threadCount];
|
|||
|
|
var dirPath = Path.GetDirectoryName(filePath);
|
|||
|
|
var fileName = Path.GetFileName(filePath);
|
|||
|
|
// 下载根目录不存在则创建
|
|||
|
|
if (!Directory.Exists(dirPath))
|
|||
|
|
{
|
|||
|
|
Directory.CreateDirectory(dirPath);
|
|||
|
|
}
|
|||
|
|
// 查看下载临时文件是否可以继续断点续传
|
|||
|
|
var fileInfos = new DirectoryInfo(dirPath).GetFiles(fileName + "*.temp");
|
|||
|
|
if (fileInfos.Length != threadCount)
|
|||
|
|
{
|
|||
|
|
// 下载临时文件数量不相同，则清理
|
|||
|
|
foreach (var info in fileInfos)
|
|||
|
|
{
|
|||
|
|
info.Delete();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// 创建下载临时文件，并创建文件流
|
|||
|
|
for (int i = 0; i < threadCount; i++)
|
|||
|
|
{
|
|||
|
|
tempFilePaths[i] = filePath + i + ".temp";
|
|||
|
|
if (!File.Exists(tempFilePaths[i]))
|
|||
|
|
{
|
|||
|
|
File.Create(tempFilePaths[i]).Dispose();
|
|||
|
|
}
|
|||
|
|
tempFileFileStreams[i] = File.OpenWrite(tempFilePaths[i]);
|
|||
|
|
tempFileFileStreams[i].Seek(tempFileFileStreams[i].Length, System.IO.SeekOrigin.Current);
|
|||
|
|
|
|||
|
|
csize += tempFileFileStreams[i].Length;
|
|||
|
|
}
|
|||
|
|
// 单线程下载过程回调函数
|
|||
|
|
Action<int, long, byte[], byte[]> t_onDownloading = (index, rsize, rbytes, data) =>
|
|||
|
|
{
|
|||
|
|
csize += rsize;
|
|||
|
|
tempFileFileStreams[index].Write(rbytes, 0, (int)rsize);
|
|||
|
|
PostMainThreadAction<long, long>(onDownloading, csize, size);
|
|||
|
|
};
|
|||
|
|
// 单线程下载完毕回调函数
|
|||
|
|
Action<int, byte[]> t_onTrigger = (index, data) =>
|
|||
|
|
{
|
|||
|
|
// 关闭文件流
|
|||
|
|
tempFileFileStreams[index].Close();
|
|||
|
|
ocnt++;
|
|||
|
|
if (ocnt >= threadCount)
|
|||
|
|
{
|
|||
|
|
// 将临时文件转为下载文件
|
|||
|
|
if (!File.Exists(filePath))
|
|||
|
|
{
|
|||
|
|
File.Create(filePath).Dispose();
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
File.WriteAllBytes(filePath, new byte[] { });
|
|||
|
|
}
|
|||
|
|
FileStream fs = File.OpenWrite(filePath);
|
|||
|
|
fs.Seek(fs.Length, System.IO.SeekOrigin.Current);
|
|||
|
|
foreach (var tempPath in tempFilePaths)
|
|||
|
|
{
|
|||
|
|
var tempData = File.ReadAllBytes(tempPath);
|
|||
|
|
fs.Write(tempData, 0, tempData.Length);
|
|||
|
|
File.Delete(tempPath);
|
|||
|
|
}
|
|||
|
|
fs.Close();
|
|||
|
|
PostMainThreadAction<byte[]>(onTrigger, File.ReadAllBytes(filePath));
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
// 分割文件尺寸，多线程下载
|
|||
|
|
long[] sizes = SplitFileSize(size, threadCount);
|
|||
|
|
for (int i = 0; i < sizes.Length; i = i + 2)
|
|||
|
|
{
|
|||
|
|
long from = sizes[i];
|
|||
|
|
long to = sizes[i + 1];
|
|||
|
|
|
|||
|
|
// 断点续传
|
|||
|
|
from += tempFileFileStreams[i / 2].Length;
|
|||
|
|
if (from >= to)
|
|||
|
|
{
|
|||
|
|
t_onTrigger(i / 2, null);
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
_threadDownload(i / 2, from, to, t_onDownloading, t_onTrigger);
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
/// <summary>
|
|||
|
|
/// 多线程下载文件至内存
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="threadCount">线程总数</param>
|
|||
|
|
/// <param name="onDownloading">下载过程回调（已下载文件大小、总文件大小）</param>
|
|||
|
|
/// <param name="onTrigger">下载完毕回调（下载文件数据）</param>
|
|||
|
|
public void DownloadToMemory(int threadCount, Action<long, long> onDownloading = null, Action<byte[]> onTrigger = null)
|
|||
|
|
{
|
|||
|
|
_isDownload = true;
|
|||
|
|
|
|||
|
|
long csize = 0; // 已下载大小
|
|||
|
|
int ocnt = 0; // 完成线程数
|
|||
|
|
byte[] cdata; // 已下载数据
|
|||
|
|
// 下载逻辑
|
|||
|
|
GetFileSizeAsyn((size) =>
|
|||
|
|
{
|
|||
|
|
cdata = new byte[size];
|
|||
|
|
// 单线程下载过程回调函数
|
|||
|
|
Action<int, long, byte[], byte[]> t_onDownloading = (index, rsize, rbytes, data) =>
|
|||
|
|
{
|
|||
|
|
csize += rsize;
|
|||
|
|
PostMainThreadAction<long, long>(onDownloading, csize, size);
|
|||
|
|
};
|
|||
|
|
// 单线程下载完毕回调函数
|
|||
|
|
Action<int, byte[]> t_onTrigger = (index, data) =>
|
|||
|
|
{
|
|||
|
|
long dIndex = (long)Math.Ceiling((double)(size * index / threadCount));
|
|||
|
|
Array.Copy(data, 0, cdata, dIndex, data.Length);
|
|||
|
|
|
|||
|
|
ocnt++;
|
|||
|
|
if (ocnt >= threadCount)
|
|||
|
|
{
|
|||
|
|
PostMainThreadAction<byte[]>(onTrigger, cdata);
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
// 分割文件尺寸，多线程下载
|
|||
|
|
long[] sizes = SplitFileSize(size, threadCount);
|
|||
|
|
for (int i = 0; i < sizes.Length; i = i + 2)
|
|||
|
|
{
|
|||
|
|
long from = sizes[i];
|
|||
|
|
long to = sizes[i + 1];
|
|||
|
|
_threadDownload(i / 2, from, to, t_onDownloading, t_onTrigger);
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
/// <summary>
|
|||
|
|
/// 单线程下载
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="index">线程ID</param>
|
|||
|
|
/// <param name="from">下载起始位置</param>
|
|||
|
|
/// <param name="to">下载结束位置</param>
|
|||
|
|
/// <param name="onDownloading">下载过程回调（线程ID、单次下载数据大小、单次下载数据缓存区、已下载文件数据）</param>
|
|||
|
|
/// <param name="onTrigger">下载完毕回调（线程ID、下载文件数据）</param>
|
|||
|
|
private void _threadDownload(int index, long from, long to, Action<int, long, byte[], byte[]> onDownloading = null, Action<int, byte[]> onTrigger = null)
|
|||
|
|
{
|
|||
|
|
Thread thread = new Thread(new ThreadStart(() =>
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
var request = (HttpWebRequest)WebRequest.Create(new Uri(Url));
|
|||
|
|
request.AddRange(from, to);
|
|||
|
|
|
|||
|
|
HttpWebResponse response = (HttpWebResponse)request.GetResponse();
|
|||
|
|
Stream ns = response.GetResponseStream();
|
|||
|
|
|
|||
|
|
byte[] rbytes = new byte[8 * 1024];
|
|||
|
|
int rSize = 0;
|
|||
|
|
MemoryStream ms = new MemoryStream();
|
|||
|
|
while (true)
|
|||
|
|
{
|
|||
|
|
if (!_isDownload) return;
|
|||
|
|
rSize = ns.Read(rbytes, 0, rbytes.Length);
|
|||
|
|
if (rSize <= 0) break;
|
|||
|
|
ms.Write(rbytes, 0, rSize);
|
|||
|
|
if (onDownloading != null) onDownloading(index, rSize, rbytes, ms.ToArray());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ns.Close();
|
|||
|
|
response.Close();
|
|||
|
|
request.Abort();
|
|||
|
|
|
|||
|
|
if (ms.Length == (to - from) + 1)
|
|||
|
|
{
|
|||
|
|
if (onTrigger != null) onTrigger(index, ms.ToArray());
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
lock (errorlock)
|
|||
|
|
{
|
|||
|
|
if (_isDownload)
|
|||
|
|
{
|
|||
|
|
onError(new Exception("文件大小校验不通过"));
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
catch (Exception ex)
|
|||
|
|
{
|
|||
|
|
onError(ex);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
}));
|
|||
|
|
thread.Start();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void Close()
|
|||
|
|
{
|
|||
|
|
_isDownload = false;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 分割文件大小
|
|||
|
|
/// </summary>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
private long[] SplitFileSize(long size, int count)
|
|||
|
|
{
|
|||
|
|
long[] result = new long[count * 2];
|
|||
|
|
for (int i = 0; i < count; i++)
|
|||
|
|
{
|
|||
|
|
long from = (long)Math.Ceiling((double)(size * i / count));
|
|||
|
|
long to = (long)Math.Ceiling((double)(size * (i + 1) / count)) - 1;
|
|||
|
|
result[i * 2] = from;
|
|||
|
|
result[i * 2 + 1] = to;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return result;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void onError(Exception ex)
|
|||
|
|
{
|
|||
|
|
Close();
|
|||
|
|
PostMainThreadAction<Exception>(OnError, ex);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 通知主线程回调
|
|||
|
|
/// </summary>
|
|||
|
|
private void PostMainThreadAction(Action action)
|
|||
|
|
{
|
|||
|
|
_mainThreadSynContext.Post(new SendOrPostCallback((o) =>
|
|||
|
|
{
|
|||
|
|
Action e = (Action)o.GetType().GetProperty("action").GetValue(o);
|
|||
|
|
if (e != null) e();
|
|||
|
|
}), new { action = action });
|
|||
|
|
}
|
|||
|
|
private void PostMainThreadAction<T>(Action<T> action, T arg1)
|
|||
|
|
{
|
|||
|
|
_mainThreadSynContext.Post(new SendOrPostCallback((o) =>
|
|||
|
|
{
|
|||
|
|
Action<T> e = (Action<T>)o.GetType().GetProperty("action").GetValue(o);
|
|||
|
|
T t1 = (T)o.GetType().GetProperty("arg1").GetValue(o);
|
|||
|
|
if (e != null) e(t1);
|
|||
|
|
}), new { action = action, arg1 = arg1 });
|
|||
|
|
}
|
|||
|
|
public void PostMainThreadAction<T1, T2>(Action<T1, T2> action, T1 arg1, T2 arg2)
|
|||
|
|
{
|
|||
|
|
_mainThreadSynContext.Post(new SendOrPostCallback((o) =>
|
|||
|
|
{
|
|||
|
|
Action<T1, T2> e = (Action<T1, T2>)o.GetType().GetProperty("action").GetValue(o);
|
|||
|
|
T1 t1 = (T1)o.GetType().GetProperty("arg1").GetValue(o);
|
|||
|
|
T2 t2 = (T2)o.GetType().GetProperty("arg2").GetValue(o);
|
|||
|
|
if (e != null) e(t1, t2);
|
|||
|
|
}), new { action = action, arg1 = arg1, arg2 = arg2 });
|
|||
|
|
}
|
|||
|
|
public void PostMainThreadAction<T1, T2, T3>(Action<T1, T2, T3> action, T1 arg1, T2 arg2, T3 arg3)
|
|||
|
|
{
|
|||
|
|
_mainThreadSynContext.Post(new SendOrPostCallback((o) =>
|
|||
|
|
{
|
|||
|
|
Action<T1, T2, T3> e = (Action<T1, T2, T3>)o.GetType().GetProperty("action").GetValue(o);
|
|||
|
|
T1 t1 = (T1)o.GetType().GetProperty("arg1").GetValue(o);
|
|||
|
|
T2 t2 = (T2)o.GetType().GetProperty("arg2").GetValue(o);
|
|||
|
|
T3 t3 = (T3)o.GetType().GetProperty("arg3").GetValue(o);
|
|||
|
|
if (e != null) e(t1, t2, t3);
|
|||
|
|
}), new { action = action, arg1 = arg1, arg2 = arg2, arg3 = arg3 });
|
|||
|
|
}
|
|||
|
|
public void PostMainThreadAction<T1, T2, T3, T4>(Action<T1, T2, T3, T4> action, T1 arg1, T2 arg2, T3 arg3, T4 arg4)
|
|||
|
|
{
|
|||
|
|
_mainThreadSynContext.Post(new SendOrPostCallback((o) =>
|
|||
|
|
{
|
|||
|
|
Action<T1, T2, T3, T4> e = (Action<T1, T2, T3, T4>)o.GetType().GetProperty("action").GetValue(o);
|
|||
|
|
T1 t1 = (T1)o.GetType().GetProperty("arg1").GetValue(o);
|
|||
|
|
T2 t2 = (T2)o.GetType().GetProperty("arg2").GetValue(o);
|
|||
|
|
T3 t3 = (T3)o.GetType().GetProperty("arg3").GetValue(o);
|
|||
|
|
T4 t4 = (T4)o.GetType().GetProperty("arg4").GetValue(o);
|
|||
|
|
if (e != null) e(t1, t2, t3, t4);
|
|||
|
|
}), new { action = action, arg1 = arg1, arg2 = arg2, arg3 = arg3, arg4 = arg4 });
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|