DownloadTask.cs 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149
  1. using System;
  2. using System.Collections;
  3. using System.Collections.Generic;
  4. using System.IO;
  5. using UnityEngine;
  6. using UnityEngine.Networking;
  7. public enum DownloadTaskState
  8. {
  9. None,
  10. Downloading,
  11. DownloadSuccess,
  12. Error,
  13. }
  14. public class DownloadTask : MonoBehaviour
  15. {
  16. private UnityWebRequest webRqst;
  17. public UnityWebRequest WebRqst { get => webRqst; }
  18. public Action<DownloadTask> Callback { get; set; }
  19. [SerializeField]
  20. public DownloadDataEntity CurDownloadEntity { get; set; }
  21. [SerializeField]
  22. public float CurLoadProgress { get; private set; }
  23. public ulong downloadSize;
  24. private float timeOut;
  25. private DownloadTaskState state;
  26. public DownloadTaskState State {
  27. get => state;
  28. private set
  29. {
  30. state = value;
  31. if (CurDownloadEntity != null)
  32. {
  33. CurDownloadEntity.State = state;
  34. }
  35. }
  36. }
  37. public ulong DownLoadSize { get => webRqst.downloadedBytes; }
  38. public DownloadTask()
  39. {
  40. //webRqst = new UnityWebRequest();
  41. }
  42. public bool SetUrl(string url = "")
  43. {
  44. bool ret = true;
  45. if (string.IsNullOrEmpty(url) && CurDownloadEntity != null)
  46. {
  47. webRqst = UnityWebRequest.Get(DownloadMgr.Instance.DownloadUrl + CurDownloadEntity.FullName);
  48. }
  49. else if (!string.IsNullOrEmpty(url))
  50. {
  51. webRqst = UnityWebRequest.Get(url);
  52. }
  53. else
  54. {
  55. ret = false;
  56. }
  57. return ret;
  58. }
  59. public void StartTask()
  60. {
  61. StartCoroutine(Task());
  62. }
  63. public IEnumerator Task()
  64. {
  65. if (State == DownloadTaskState.Downloading)
  66. {
  67. yield break;
  68. }
  69. State = DownloadTaskState.Downloading;
  70. string savePath = DownloadMgr.Instance.LocalFilePath + CurDownloadEntity.FullName;
  71. //FileStream fs = new FileStream(savePath, FileMode.OpenOrCreate);
  72. //if (fs == null)
  73. //{
  74. // State = DownloadTaskState.Error;
  75. // Callback?.Invoke(this);
  76. // yield break;
  77. //}
  78. UnityWebRequestAsyncOperation asyncOperation = webRqst.SendWebRequest();
  79. yield return null;
  80. timeOut = Time.time;
  81. CurLoadProgress = 0;
  82. downloadSize = 0;
  83. while (!webRqst.isDone)
  84. {
  85. if (CurLoadProgress < webRqst.downloadProgress)
  86. {
  87. timeOut = Time.time;
  88. CurLoadProgress = webRqst.downloadProgress;
  89. //int count = (int)(webRqst.downloadedBytes - downloadSize);
  90. //fs.Write(webRqst.downloadHandler.data,(int)downloadSize, count);
  91. downloadSize = webRqst.downloadedBytes;
  92. CurDownloadEntity.UpDateCallback?.Invoke(this);
  93. }
  94. if (Time.time - timeOut > DownloadMgr.TimeOut)
  95. {
  96. Debug.LogWarning("下载超时!!!");
  97. //fs.Close();
  98. State = DownloadTaskState.Error;
  99. Callback?.Invoke(this);
  100. yield break;
  101. }
  102. yield return null;
  103. }
  104. CurLoadProgress = webRqst.downloadProgress;
  105. downloadSize = webRqst.downloadedBytes;
  106. CurDownloadEntity.UpDateCallback?.Invoke(this);
  107. //yield return asyncOperation;
  108. //fs.Flush();
  109. //fs.Close();
  110. if (webRqst != null && webRqst.error == null)
  111. {
  112. Debug.Log($"[{CurDownloadEntity.FullName}] 下载完成");
  113. State = DownloadTaskState.DownloadSuccess;
  114. //FileHelper.WriteToFileAsync( savePath,webRqst.downloadHandler.data,()=> { Callback?.Invoke(this); });
  115. FileHelper.WirteToFile( savePath,webRqst.downloadHandler.data);
  116. Callback?.Invoke(this);
  117. }
  118. else
  119. {
  120. State = DownloadTaskState.Error;
  121. Debug.Log($"下载失败;Error = {webRqst.error} === {CurDownloadEntity.FullName} ");
  122. Callback?.Invoke(this);
  123. }
  124. }
  125. }