TaskService.cs 7.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215
  1. using NHibernate;
  2. using NHibernate.Criterion;
  3. using System;
  4. using System.Collections.Generic;
  5. using System.Linq;
  6. using System.Text;
  7. using System.Threading.Tasks;
  8. using TaskDatabase.Model;
  9. namespace TaskDatabase
  10. {
  11. public class TaskService
  12. {
  13. public bool AddTask(TaskModel task) {
  14. ITransaction transaction = null;
  15. try
  16. {
  17. //如果存在该任务, 则不允许添加
  18. if (isContainsTask(task.Id))
  19. {
  20. return false;
  21. }
  22. using (ISession session = NHibernateHelper.OpenSession()) {
  23. using (transaction = session.BeginTransaction()) {
  24. try
  25. {
  26. session.Save(task);
  27. if(task.DownloadTaskModelList != null)
  28. {
  29. foreach(var download in task.DownloadTaskModelList)
  30. {
  31. session.Save(download);
  32. }
  33. }
  34. transaction.Commit();
  35. }
  36. catch (Exception ex){
  37. if (transaction != null && transaction.IsActive)
  38. {
  39. transaction.Rollback();
  40. }
  41. Console.WriteLine(ex.Message);
  42. return false;
  43. }
  44. }
  45. }
  46. return true;
  47. }
  48. catch {}
  49. return false;
  50. }
  51. public bool UpdateTask(TaskModel task)
  52. {
  53. ITransaction transaction = null;
  54. try
  55. {
  56. using (ISession session = NHibernateHelper.OpenSession())
  57. {
  58. using (transaction = session.BeginTransaction())
  59. {
  60. session.Update(task);
  61. transaction.Commit();
  62. }
  63. }
  64. return true;
  65. }
  66. catch (Exception ex)
  67. {
  68. Console.WriteLine(ex.Message+ex.StackTrace);
  69. }
  70. return false;
  71. }
  72. public bool UpdateDownload(DownloadTaskModel download)
  73. {
  74. ITransaction transaction = null;
  75. try
  76. {
  77. using (ISession session = NHibernateHelper.OpenSession())
  78. {
  79. using (transaction = session.BeginTransaction())
  80. {
  81. session.Update(download);
  82. transaction.Commit();
  83. }
  84. }
  85. return true;
  86. }
  87. catch { }
  88. return false;
  89. }
  90. public IList<TaskModel> GetTasksByStatus(int status)
  91. {
  92. ITransaction transaction = null;
  93. try
  94. {
  95. using (ISession session = NHibernateHelper.OpenSession())
  96. {
  97. using (transaction = session.BeginTransaction())
  98. {
  99. var queryResult = session.CreateCriteria(typeof(TaskModel))
  100. .Add(Restrictions.Eq("Task_status", status))
  101. .List<TaskModel>();
  102. return queryResult;
  103. }
  104. }
  105. }
  106. catch (Exception ex){
  107. Console.WriteLine(ex.Message);
  108. }
  109. return null;
  110. }
  111. public TaskModel GetTasksByTid(int tid)
  112. {
  113. ITransaction transaction = null;
  114. try
  115. {
  116. using (ISession session = NHibernateHelper.OpenSession())
  117. {
  118. using (transaction = session.BeginTransaction())
  119. {
  120. var queryResult = session.CreateCriteria(typeof(TaskModel))
  121. .Add(Restrictions.Eq("Tid", tid))
  122. .UniqueResult<TaskModel>();
  123. return queryResult;
  124. }
  125. }
  126. }
  127. catch (Exception ex)
  128. {
  129. Console.WriteLine(ex.Message);
  130. }
  131. return null;
  132. }
  133. public IList<TaskModel> GetTasksByTaskId(int taskId)
  134. {
  135. ITransaction transaction = null;
  136. try
  137. {
  138. using (ISession session = NHibernateHelper.OpenSession())
  139. {
  140. using (transaction = session.BeginTransaction())
  141. {
  142. var queryResult = session.CreateCriteria(typeof(TaskModel))
  143. .Add(Restrictions.Eq("Id", taskId))
  144. .List<TaskModel>();
  145. return queryResult;
  146. }
  147. }
  148. }
  149. catch (Exception ex)
  150. {
  151. Console.WriteLine(ex.Message);
  152. }
  153. return null;
  154. }
  155. public bool isContainsTask(int taskId)
  156. {
  157. ITransaction transaction = null;
  158. try
  159. {
  160. using (ISession session = NHibernateHelper.OpenSession())
  161. {
  162. using (transaction = session.BeginTransaction())
  163. {
  164. var queryResult = session.CreateCriteria(typeof(TaskModel))
  165. .Add(Restrictions.Eq("Id", taskId))
  166. .List<TaskModel>();
  167. return queryResult.Count > 0;
  168. }
  169. }
  170. }
  171. catch (Exception ex)
  172. {
  173. Console.WriteLine(ex.Message);
  174. }
  175. return true;
  176. }
  177. public void removeErrorTask()
  178. {
  179. ITransaction transaction = null;
  180. try
  181. {
  182. using (ISession session = NHibernateHelper.OpenSession())
  183. {
  184. using (transaction = session.BeginTransaction())
  185. {
  186. var queryResult = session.CreateCriteria(typeof(TaskModel))
  187. .Add(Restrictions.Le("Task_status", -1))
  188. .List<TaskModel>();
  189. for (int i = 0; i < queryResult.Count; ++i) {
  190. ICollection<DownloadTaskModel> downloadTaskModels = queryResult[i].DownloadTaskModelList;
  191. for (int j = 0; j < downloadTaskModels.Count; ++j) {
  192. session.Delete(downloadTaskModels.ElementAt(j));
  193. }
  194. session.Delete(queryResult[i]);
  195. }
  196. transaction.Commit();
  197. }
  198. }
  199. }
  200. catch (Exception ex)
  201. {
  202. Console.WriteLine(ex.Message);
  203. }
  204. }
  205. }
  206. }