TaskHandler.cs 8.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using System.Text;
  5. using System.Threading;
  6. using System.Threading.Tasks;
  7. using HttpDownload;
  8. using NettyClient;
  9. using NettyClient.proto;
  10. using Newtonsoft.Json;
  11. using TaskDatabase;
  12. using TaskDatabase.Model;
  13. namespace Client
  14. {
  15. class TaskHandler
  16. {
  17. private SimpleMessageHandler simpleHandler;
  18. private TaskService taskService;
  19. public IList<TaskModel> taskModels;
  20. private TaskDownloadManager taskDownloadManager;
  21. private string defaultFileDir;
  22. IRevitCommandExcutor revitCommandExcutor;
  23. public void RunTasks(object obj) {
  24. while (true) {
  25. try
  26. {
  27. for (int i = taskModels.Count - 1; i > -1; --i) {
  28. TaskModel task = taskModels[i];
  29. TaskModel downloadException = null, commandException = null;
  30. bool isAllDownloaded = true;
  31. // 看当前任务依赖的文件下载状况
  32. foreach (DownloadTaskModel downloadTaskModel in task.DownloadTaskModelList) {
  33. DownloadTask download = taskDownloadManager.getDownloadTaskByDownloadTaskId(downloadTaskModel.Tid);
  34. if (download == null) // 如果还未加入下载队列
  35. {
  36. // 开始下载
  37. if (downloadTaskModel.Local_dir == null) {
  38. downloadTaskModel.Local_dir = defaultFileDir + downloadTaskModel.Tid + ".rvt";
  39. }
  40. download = taskDownloadManager.enqueueTask(task.Id, downloadTaskModel.Tid, downloadTaskModel.Task_url,
  41. downloadTaskModel.Local_dir, downloadTaskModel.Task_md5, downloadTaskModel.Downloaded_bytes);
  42. taskDownloadManager.runTask(download);
  43. isAllDownloaded &= false;
  44. }
  45. else {
  46. // 如果该下载发生异常
  47. if (download.Status == DownloadStatus.ExceptionStopped) {
  48. isAllDownloaded &= false;
  49. downloadException = task;
  50. break;
  51. }
  52. // 同步当前状态到数据库
  53. if (download.Status == DownloadStatus.Completed)
  54. {
  55. downloadTaskModel.Finish_time = GetTimeStampSeconds();
  56. isAllDownloaded &= true;
  57. }
  58. else {
  59. isAllDownloaded &= false;
  60. }
  61. syncSingleDownTask(download, downloadTaskModel);
  62. }
  63. } // foreach end
  64. // 如果该下载任务下载失败
  65. if (downloadException != null)
  66. {
  67. task.Task_status = -1;
  68. taskService.UpdateTask(task);
  69. taskModels.RemoveAt(i);
  70. simpleHandler.WriteMessage(MessageUtil.generateMessage(Command.DownloadError.ToString(), task.Id, ""));
  71. continue;
  72. }
  73. // 如果该任务的所有下载文件均已完成, 开始执行Revit命令
  74. if (isAllDownloaded) {
  75. try
  76. {
  77. HashSet<string> fileSet = new HashSet<string>();
  78. foreach(var downloadModel in task.DownloadTaskModelList) {
  79. fileSet.Add(downloadModel.Local_dir);
  80. }
  81. try
  82. {
  83. string resultJson = revitCommandExcutor.ExecuteCmd(task.Task_cmd, task.Task_param, fileSet);
  84. simpleHandler.WriteMessage(MessageUtil.generateMessage(Command.TaskSuccess.ToString(), task.Id, resultJson));
  85. task.Task_status = 1;
  86. task.Task_result_json = resultJson;
  87. taskService.UpdateTask(task);
  88. taskModels.RemoveAt(i);
  89. }
  90. catch(Exception ex) {
  91. Console.WriteLine(ex.StackTrace);
  92. commandException = task;
  93. }
  94. }
  95. catch(Exception ex) {
  96. Console.WriteLine(ex.StackTrace);
  97. }
  98. }
  99. if (commandException != null)
  100. {
  101. task.Task_status = -2;
  102. taskService.UpdateTask(task);
  103. taskModels.RemoveAt(i);
  104. simpleHandler.WriteMessage(MessageUtil.generateMessage(Command.CommandError.ToString(), task.Id, ""));
  105. continue;
  106. }
  107. }
  108. taskService.removeErrorTask();
  109. }
  110. catch (Exception ex){
  111. Console.WriteLine(ex.StackTrace);
  112. }
  113. Thread.Sleep(1000);
  114. }
  115. }
  116. private void syncSingleDownTask(DownloadTask downTask, DownloadTaskModel downloadModel)
  117. {
  118. bool changed = false;
  119. if (downloadModel.Downloaded_bytes != downTask.BytesWritten)
  120. {
  121. downloadModel.Downloaded_bytes = downTask.BytesWritten;
  122. changed = true;
  123. }
  124. if (downloadModel.File_bytes != downTask.totalFileSize)
  125. {
  126. downloadModel.File_bytes = downTask.totalFileSize;
  127. changed = true;
  128. }
  129. if(changed)
  130. taskService.UpdateDownload(downloadModel);
  131. }
  132. public void setSimpleMessageHandler(SimpleMessageHandler simpleHandler)
  133. {
  134. this.simpleHandler = simpleHandler;
  135. }
  136. private TaskHandler(SimpleMessageHandler simpleHandler, TaskService taskService)
  137. {
  138. this.simpleHandler = simpleHandler;
  139. this.taskService = taskService;
  140. }
  141. public TaskHandler(SimpleMessageHandler simpleHandler, TaskService taskService, TaskDownloadManager taskDownloadManager, string defaultFileDir, IRevitCommandExcutor revitCommandExcutor) : this(simpleHandler, taskService)
  142. {
  143. this.taskDownloadManager = taskDownloadManager;
  144. this.defaultFileDir = defaultFileDir;
  145. this.revitCommandExcutor = revitCommandExcutor;
  146. }
  147. internal void SyncAllTask()
  148. {
  149. IList<TaskModel> tasks = taskService.GetTasksByStatus(0); // 任务状态 0 未执行完的任务, 1 执行完成 -1 下载出错 -2 revit命令执行异常
  150. taskModels = tasks;
  151. }
  152. internal bool addOneTask(int taskId, string msgContent) {
  153. try
  154. {
  155. TaskModel taskModel = JsonConvert.DeserializeObject<TaskModel>(msgContent);
  156. taskService.AddTask(taskModel);
  157. TaskModel task = taskService.GetTasksByTaskId(taskId)[0];
  158. taskModels.Add(task);
  159. foreach (DownloadTaskModel download in taskModel.DownloadTaskModelList) {
  160. download.Local_dir = defaultFileDir + download.Tid + ".rvt";
  161. taskService.UpdateDownload(download);
  162. }
  163. return true;
  164. }
  165. catch {
  166. return false;
  167. }
  168. }
  169. internal bool isContainTask(int taskId)
  170. {
  171. try
  172. {
  173. return taskService.isContainsTask(taskId);
  174. }
  175. catch (Exception ex){
  176. Console.WriteLine(ex.StackTrace);
  177. return true;
  178. }
  179. }
  180. public static long GetTimeStampSeconds() {
  181. TimeSpan ts = DateTime.UtcNow - new DateTime(1970, 1, 1, 0, 0, 0, 0);
  182. return (long)ts.TotalSeconds;
  183. }
  184. }
  185. }