123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486 |
- //------------------------------------------------------------
- // Game Framework
- // Copyright © 2013-2021 loyalsoft. All rights reserved.
- // Homepage: http://www.game7000.com/
- // Feedback: http://www.game7000.com/
- //------------------------------------------------------------
- using System;
- using System.Collections.Generic;
- namespace GameFramework.Download
- {
- /// <summary>
- /// 下载管理器。
- /// </summary>
- internal sealed partial class DownloadManager : GameFrameworkModule, IDownloadManager
- {
- private const int OneMegaBytes = 1024 * 1024;
- private readonly TaskPool<DownloadTask> m_TaskPool;
- private readonly DownloadCounter m_DownloadCounter;
- private int m_FlushSize;
- private float m_Timeout;
- private EventHandler<DownloadStartEventArgs> m_DownloadStartEventHandler;
- private EventHandler<DownloadUpdateEventArgs> m_DownloadUpdateEventHandler;
- private EventHandler<DownloadSuccessEventArgs> m_DownloadSuccessEventHandler;
- private EventHandler<DownloadFailureEventArgs> m_DownloadFailureEventHandler;
- /// <summary>
- /// 初始化下载管理器的新实例。
- /// </summary>
- public DownloadManager()
- {
- m_TaskPool = new TaskPool<DownloadTask>();
- m_DownloadCounter = new DownloadCounter(1f, 10f);
- m_FlushSize = OneMegaBytes;
- m_Timeout = 30f;
- m_DownloadStartEventHandler = null;
- m_DownloadUpdateEventHandler = null;
- m_DownloadSuccessEventHandler = null;
- m_DownloadFailureEventHandler = null;
- }
- /// <summary>
- /// 获取游戏框架模块优先级。
- /// </summary>
- /// <remarks>优先级较高的模块会优先轮询,并且关闭操作会后进行。</remarks>
- internal override int Priority
- {
- get
- {
- return 5;
- }
- }
- /// <summary>
- /// 获取或设置下载是否被暂停。
- /// </summary>
- public bool Paused
- {
- get
- {
- return m_TaskPool.Paused;
- }
- set
- {
- m_TaskPool.Paused = value;
- }
- }
- /// <summary>
- /// 获取下载代理总数量。
- /// </summary>
- public int TotalAgentCount
- {
- get
- {
- return m_TaskPool.TotalAgentCount;
- }
- }
- /// <summary>
- /// 获取可用下载代理数量。
- /// </summary>
- public int FreeAgentCount
- {
- get
- {
- return m_TaskPool.FreeAgentCount;
- }
- }
- /// <summary>
- /// 获取工作中下载代理数量。
- /// </summary>
- public int WorkingAgentCount
- {
- get
- {
- return m_TaskPool.WorkingAgentCount;
- }
- }
- /// <summary>
- /// 获取等待下载任务数量。
- /// </summary>
- public int WaitingTaskCount
- {
- get
- {
- return m_TaskPool.WaitingTaskCount;
- }
- }
- /// <summary>
- /// 获取或设置将缓冲区写入磁盘的临界大小。
- /// </summary>
- public int FlushSize
- {
- get
- {
- return m_FlushSize;
- }
- set
- {
- m_FlushSize = value;
- }
- }
- /// <summary>
- /// 获取或设置下载超时时长,以秒为单位。
- /// </summary>
- public float Timeout
- {
- get
- {
- return m_Timeout;
- }
- set
- {
- m_Timeout = value;
- }
- }
- /// <summary>
- /// 获取当前下载速度。
- /// </summary>
- public float CurrentSpeed
- {
- get
- {
- return m_DownloadCounter.CurrentSpeed;
- }
- }
- /// <summary>
- /// 下载开始事件。
- /// </summary>
- public event EventHandler<DownloadStartEventArgs> DownloadStart
- {
- add
- {
- m_DownloadStartEventHandler += value;
- }
- remove
- {
- m_DownloadStartEventHandler -= value;
- }
- }
- /// <summary>
- /// 下载更新事件。
- /// </summary>
- public event EventHandler<DownloadUpdateEventArgs> DownloadUpdate
- {
- add
- {
- m_DownloadUpdateEventHandler += value;
- }
- remove
- {
- m_DownloadUpdateEventHandler -= value;
- }
- }
- /// <summary>
- /// 下载成功事件。
- /// </summary>
- public event EventHandler<DownloadSuccessEventArgs> DownloadSuccess
- {
- add
- {
- m_DownloadSuccessEventHandler += value;
- }
- remove
- {
- m_DownloadSuccessEventHandler -= value;
- }
- }
- /// <summary>
- /// 下载失败事件。
- /// </summary>
- public event EventHandler<DownloadFailureEventArgs> DownloadFailure
- {
- add
- {
- m_DownloadFailureEventHandler += value;
- }
- remove
- {
- m_DownloadFailureEventHandler -= value;
- }
- }
- /// <summary>
- /// 下载管理器轮询。
- /// </summary>
- /// <param name="elapseSeconds">逻辑流逝时间,以秒为单位。</param>
- /// <param name="realElapseSeconds">真实流逝时间,以秒为单位。</param>
- internal override void Update(float elapseSeconds, float realElapseSeconds)
- {
- m_TaskPool.Update(elapseSeconds, realElapseSeconds);
- m_DownloadCounter.Update(elapseSeconds, realElapseSeconds);
- }
- /// <summary>
- /// 关闭并清理下载管理器。
- /// </summary>
- internal override void Shutdown()
- {
- m_TaskPool.Shutdown();
- m_DownloadCounter.Shutdown();
- }
- /// <summary>
- /// 增加下载代理辅助器。
- /// </summary>
- /// <param name="downloadAgentHelper">要增加的下载代理辅助器。</param>
- public void AddDownloadAgentHelper(IDownloadAgentHelper downloadAgentHelper)
- {
- DownloadAgent agent = new DownloadAgent(downloadAgentHelper);
- agent.DownloadAgentStart += OnDownloadAgentStart;
- agent.DownloadAgentUpdate += OnDownloadAgentUpdate;
- agent.DownloadAgentSuccess += OnDownloadAgentSuccess;
- agent.DownloadAgentFailure += OnDownloadAgentFailure;
- m_TaskPool.AddAgent(agent);
- }
- /// <summary>
- /// 根据下载任务的序列编号获取下载任务的信息。
- /// </summary>
- /// <param name="serialId">要获取信息的下载任务的序列编号。</param>
- /// <returns>下载任务的信息。</returns>
- public TaskInfo GetDownloadInfo(int serialId)
- {
- return m_TaskPool.GetTaskInfo(serialId);
- }
- /// <summary>
- /// 根据下载任务的标签获取下载任务的信息。
- /// </summary>
- /// <param name="tag">要获取信息的下载任务的标签。</param>
- /// <returns>下载任务的信息。</returns>
- public TaskInfo[] GetDownloadInfos(string tag)
- {
- return m_TaskPool.GetTaskInfos(tag);
- }
- /// <summary>
- /// 根据下载任务的标签获取下载任务的信息。
- /// </summary>
- /// <param name="tag">要获取信息的下载任务的标签。</param>
- /// <param name="results">下载任务的信息。</param>
- public void GetDownloadInfos(string tag, List<TaskInfo> results)
- {
- m_TaskPool.GetTaskInfos(tag, results);
- }
- /// <summary>
- /// 获取所有下载任务的信息。
- /// </summary>
- /// <returns>所有下载任务的信息。</returns>
- public TaskInfo[] GetAllDownloadInfos()
- {
- return m_TaskPool.GetAllTaskInfos();
- }
- /// <summary>
- /// 获取所有下载任务的信息。
- /// </summary>
- /// <param name="results">所有下载任务的信息。</param>
- public void GetAllDownloadInfos(List<TaskInfo> results)
- {
- m_TaskPool.GetAllTaskInfos(results);
- }
- /// <summary>
- /// 增加下载任务。
- /// </summary>
- /// <param name="downloadPath">下载后存放路径。</param>
- /// <param name="downloadUri">原始下载地址。</param>
- /// <returns>新增下载任务的序列编号。</returns>
- public int AddDownload(string downloadPath, string downloadUri)
- {
- return AddDownload(downloadPath, downloadUri, null, Constant.DefaultPriority, null);
- }
- /// <summary>
- /// 增加下载任务。
- /// </summary>
- /// <param name="downloadPath">下载后存放路径。</param>
- /// <param name="downloadUri">原始下载地址。</param>
- /// <param name="tag">下载任务的标签。</param>
- /// <returns>新增下载任务的序列编号。</returns>
- public int AddDownload(string downloadPath, string downloadUri, string tag)
- {
- return AddDownload(downloadPath, downloadUri, tag, Constant.DefaultPriority, null);
- }
- /// <summary>
- /// 增加下载任务。
- /// </summary>
- /// <param name="downloadPath">下载后存放路径。</param>
- /// <param name="downloadUri">原始下载地址。</param>
- /// <param name="priority">下载任务的优先级。</param>
- /// <returns>新增下载任务的序列编号。</returns>
- public int AddDownload(string downloadPath, string downloadUri, int priority)
- {
- return AddDownload(downloadPath, downloadUri, null, priority, null);
- }
- /// <summary>
- /// 增加下载任务。
- /// </summary>
- /// <param name="downloadPath">下载后存放路径。</param>
- /// <param name="downloadUri">原始下载地址。</param>
- /// <param name="userData">用户自定义数据。</param>
- /// <returns>新增下载任务的序列编号。</returns>
- public int AddDownload(string downloadPath, string downloadUri, object userData)
- {
- return AddDownload(downloadPath, downloadUri, null, Constant.DefaultPriority, userData);
- }
- /// <summary>
- /// 增加下载任务。
- /// </summary>
- /// <param name="downloadPath">下载后存放路径。</param>
- /// <param name="downloadUri">原始下载地址。</param>
- /// <param name="tag">下载任务的标签。</param>
- /// <param name="priority">下载任务的优先级。</param>
- /// <returns>新增下载任务的序列编号。</returns>
- public int AddDownload(string downloadPath, string downloadUri, string tag, int priority)
- {
- return AddDownload(downloadPath, downloadUri, tag, priority, null);
- }
- /// <summary>
- /// 增加下载任务。
- /// </summary>
- /// <param name="downloadPath">下载后存放路径。</param>
- /// <param name="downloadUri">原始下载地址。</param>
- /// <param name="tag">下载任务的标签。</param>
- /// <param name="userData">用户自定义数据。</param>
- /// <returns>新增下载任务的序列编号。</returns>
- public int AddDownload(string downloadPath, string downloadUri, string tag, object userData)
- {
- return AddDownload(downloadPath, downloadUri, tag, Constant.DefaultPriority, userData);
- }
- /// <summary>
- /// 增加下载任务。
- /// </summary>
- /// <param name="downloadPath">下载后存放路径。</param>
- /// <param name="downloadUri">原始下载地址。</param>
- /// <param name="priority">下载任务的优先级。</param>
- /// <param name="userData">用户自定义数据。</param>
- /// <returns>新增下载任务的序列编号。</returns>
- public int AddDownload(string downloadPath, string downloadUri, int priority, object userData)
- {
- return AddDownload(downloadPath, downloadUri, null, priority, userData);
- }
- /// <summary>
- /// 增加下载任务。
- /// </summary>
- /// <param name="downloadPath">下载后存放路径。</param>
- /// <param name="downloadUri">原始下载地址。</param>
- /// <param name="tag">下载任务的标签。</param>
- /// <param name="priority">下载任务的优先级。</param>
- /// <param name="userData">用户自定义数据。</param>
- /// <returns>新增下载任务的序列编号。</returns>
- public int AddDownload(string downloadPath, string downloadUri, string tag, int priority, object userData)
- {
- if (string.IsNullOrEmpty(downloadPath))
- {
- throw new GameFrameworkException("Download path is invalid.");
- }
- if (string.IsNullOrEmpty(downloadUri))
- {
- throw new GameFrameworkException("Download uri is invalid.");
- }
- if (TotalAgentCount <= 0)
- {
- throw new GameFrameworkException("You must add download agent first.");
- }
- DownloadTask downloadTask = DownloadTask.Create(downloadPath, downloadUri, tag, priority, m_FlushSize, m_Timeout, userData);
- m_TaskPool.AddTask(downloadTask);
- return downloadTask.SerialId;
- }
- /// <summary>
- /// 根据下载任务的序列编号移除下载任务。
- /// </summary>
- /// <param name="serialId">要移除下载任务的序列编号。</param>
- /// <returns>是否移除下载任务成功。</returns>
- public bool RemoveDownload(int serialId)
- {
- return m_TaskPool.RemoveTask(serialId);
- }
- /// <summary>
- /// 根据下载任务的标签移除下载任务。
- /// </summary>
- /// <param name="tag">要移除下载任务的标签。</param>
- /// <returns>移除下载任务的数量。</returns>
- public int RemoveDownloads(string tag)
- {
- return m_TaskPool.RemoveTasks(tag);
- }
- /// <summary>
- /// 移除所有下载任务。
- /// </summary>
- /// <returns>移除下载任务的数量。</returns>
- public int RemoveAllDownloads()
- {
- return m_TaskPool.RemoveAllTasks();
- }
- private void OnDownloadAgentStart(DownloadAgent sender)
- {
- if (m_DownloadStartEventHandler != null)
- {
- DownloadStartEventArgs downloadStartEventArgs = DownloadStartEventArgs.Create(sender.Task.SerialId, sender.Task.DownloadPath, sender.Task.DownloadUri, sender.CurrentLength, sender.Task.UserData);
- m_DownloadStartEventHandler(this, downloadStartEventArgs);
- ReferencePool.Release(downloadStartEventArgs);
- }
- }
- private void OnDownloadAgentUpdate(DownloadAgent sender, int deltaLength)
- {
- m_DownloadCounter.RecordDeltaLength(deltaLength);
- if (m_DownloadUpdateEventHandler != null)
- {
- DownloadUpdateEventArgs downloadUpdateEventArgs = DownloadUpdateEventArgs.Create(sender.Task.SerialId, sender.Task.DownloadPath, sender.Task.DownloadUri, sender.CurrentLength, sender.Task.UserData);
- m_DownloadUpdateEventHandler(this, downloadUpdateEventArgs);
- ReferencePool.Release(downloadUpdateEventArgs);
- }
- }
- private void OnDownloadAgentSuccess(DownloadAgent sender, long length)
- {
- if (m_DownloadSuccessEventHandler != null)
- {
- DownloadSuccessEventArgs downloadSuccessEventArgs = DownloadSuccessEventArgs.Create(sender.Task.SerialId, sender.Task.DownloadPath, sender.Task.DownloadUri, sender.CurrentLength, sender.Task.UserData);
- m_DownloadSuccessEventHandler(this, downloadSuccessEventArgs);
- ReferencePool.Release(downloadSuccessEventArgs);
- }
- }
- private void OnDownloadAgentFailure(DownloadAgent sender, string errorMessage)
- {
- if (m_DownloadFailureEventHandler != null)
- {
- DownloadFailureEventArgs downloadFailureEventArgs = DownloadFailureEventArgs.Create(sender.Task.SerialId, sender.Task.DownloadPath, sender.Task.DownloadUri, errorMessage, sender.Task.UserData);
- m_DownloadFailureEventHandler(this, downloadFailureEventArgs);
- ReferencePool.Release(downloadFailureEventArgs);
- }
- }
- }
- }
|