MonitorCosmosDB.cs 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Net.Http;
  4. using System.Text.Json;
  5. using System.Threading.Tasks;
  6. using Azure.Cosmos;
  7. using Azure.Messaging.ServiceBus;
  8. using Microsoft.Azure.Documents;
  9. using Microsoft.Azure.WebJobs;
  10. using Microsoft.Extensions.Logging;
  11. using TEAMModelOS.SDK.DI;
  12. using TEAMModelOS.SDK.Extension;
  13. namespace TEAMModelFunction
  14. {
  15. public class MonitorCosmosDB
  16. {
  17. private readonly IHttpClientFactory _clientFactory;
  18. private readonly AzureCosmosFactory _azureCosmos;
  19. private readonly AzureServiceBusFactory _serviceBus;
  20. private readonly AzureStorageFactory _azureStorage;
  21. public MonitorCosmosDB(IHttpClientFactory clientFactory, AzureCosmosFactory azureCosmos,AzureServiceBusFactory azureServiceBus, AzureStorageFactory azureStorage)
  22. {
  23. _clientFactory = clientFactory;
  24. _azureCosmos = azureCosmos;
  25. _serviceBus = azureServiceBus;
  26. _azureStorage = azureStorage;
  27. }
  28. [FunctionName("Common")]
  29. public async Task Common([CosmosDBTrigger(
  30. databaseName: "TEAMModelOS",
  31. collectionName: "Common",
  32. ConnectionStringSetting = "Azure:Cosmos:ConnectionString",
  33. LeaseCollectionName = "leases")]IReadOnlyList<Document> inputs, ILogger log)
  34. {
  35. if (inputs != null && inputs.Count > 0)
  36. {
  37. log.LogInformation("Documents modified " + inputs.Count);
  38. log.LogInformation("First document Id " + inputs[0].Id);
  39. var client = _azureCosmos.GetCosmosClient();
  40. foreach (var input in inputs)
  41. {
  42. string pk = input.GetPropertyValue<string>("pk");
  43. if (!string.IsNullOrWhiteSpace(pk))
  44. {
  45. long stime = input.GetPropertyValue<long>("startTime");
  46. long etime = input.GetPropertyValue<long>("endTime");
  47. string school = input.GetPropertyValue<string>("school");
  48. string code = input.GetPropertyValue<string>("code");
  49. switch (pk)
  50. {
  51. case "Exam":
  52. ExamInfo info = await client.GetContainer("TEAMModelOS", "Common").ReadItemAsync<ExamInfo>(input.Id, new Azure.Cosmos.PartitionKey($"{code}"));
  53. List<ExamClassResult> examClassResults = new List<ExamClassResult>();
  54. await foreach (var item in client.GetContainer("TEAMModelOS", "Common").GetItemQueryStreamIterator(queryText: $"select value(c) from c where c.examId = '{info.id}'", requestOptions: new QueryRequestOptions() { PartitionKey = new Azure.Cosmos.PartitionKey($"ExamClassResult-{school}") }))
  55. {
  56. using var json = await JsonDocument.ParseAsync(item.ContentStream);
  57. if (json.RootElement.TryGetProperty("_count", out JsonElement count) && count.GetUInt16() > 0)
  58. {
  59. foreach (var obj in json.RootElement.GetProperty("Documents").EnumerateArray())
  60. {
  61. examClassResults.Add(obj.ToObject<ExamClassResult>());
  62. }
  63. }
  64. }
  65. List<ChangeRecord> records = await _azureStorage.FindListByDict<ChangeRecord>(new Dictionary<string, object>() { { "RowKey", input.Id }, { "PartitionKey", info.progress } });
  66. //ChangeRecord record = await client.GetContainer("TEAMModelOS", "Common").ReadItemAsync<ChangeRecord>(input.Id, new Azure.Cosmos.PartitionKey($"{info.progress}"));
  67. switch (info.progress)
  68. {
  69. case "pending":
  70. var message = new ServiceBusMessage(new { id = input.Id, progress = "going", code = code }.ToJsonString());
  71. message.ApplicationProperties.Add("name", "Exam");
  72. if (records.Count > 0)
  73. {
  74. await _serviceBus.GetServiceBusClient().cancelMessage("active-task", records[0].sequenceNumber);
  75. long start = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", message, DateTimeOffset.FromUnixTimeMilliseconds(stime));
  76. records[0].sequenceNumber = start;
  77. await _azureStorage.SaveOrUpdate<ChangeRecord>(records[0]);
  78. //await client.GetContainer("TEAMModelOS", "Common").ReplaceItemAsync(record, record.id, new Azure.Cosmos.PartitionKey($"{record.code}"));
  79. }
  80. else
  81. {
  82. long start = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", message, DateTimeOffset.FromUnixTimeMilliseconds(stime));
  83. ChangeRecord changeRecord = new ChangeRecord
  84. {
  85. RowKey = input.Id,
  86. PartitionKey = "pending",
  87. sequenceNumber = start,
  88. msgId = message.MessageId
  89. };
  90. await _azureStorage.Save<ChangeRecord>(changeRecord);
  91. //await client.GetContainer("TEAMModelOS", "Common").CreateItemAsync(changeRecord, new Azure.Cosmos.PartitionKey($"{changeRecord.code}"));
  92. }
  93. break;
  94. case "going":
  95. if (examClassResults.Count == 0)
  96. {
  97. for (int j = 0; j < info.subjects.Count; j++)
  98. {
  99. for (int k = 0; k < info.targetClassIds.Count; k++)
  100. {
  101. ExamClassResult result = new ExamClassResult();
  102. result.code = "ExamClassResult-" + info.school;
  103. result.examId = info.id;
  104. result.id = Guid.NewGuid().ToString();
  105. result.subjectId = info.subjects[j].id;
  106. result.year = info.year;
  107. result.ttl = -1;
  108. result.scope = info.scope;
  109. result.pk = typeof(ExamClassResult).Name;
  110. result.info.id = info.targetClassIds[k];
  111. var sresponse = await client.GetContainer("TEAMModelOS", "School").ReadItemStreamAsync(info.targetClassIds[k], new Azure.Cosmos.PartitionKey($"Class-{info.school}"));
  112. if (sresponse.Status == 200)
  113. {
  114. using var json = await JsonDocument.ParseAsync(sresponse.ContentStream);
  115. Classroom classroom = json.ToObject<Classroom>();
  116. result.info.name = classroom.name;
  117. List<List<string>> ans = new List<List<string>>();
  118. List<double> ansPoint = new List<double>();
  119. foreach (double p in info.papers[j].point)
  120. {
  121. ans.Add(new List<string>());
  122. ansPoint.Add(-1);
  123. }
  124. foreach (StudentSimple stu in classroom.students)
  125. {
  126. result.studentIds.Add(stu.id);
  127. result.studentAnswers.Add(ans);
  128. result.studentScores.Add(ansPoint);
  129. }
  130. }
  131. //result.progress = info.progress;
  132. result.school = info.school;
  133. await client.GetContainer("TEAMModelOS", "Common").CreateItemAsync(result, new Azure.Cosmos.PartitionKey($"{result.code}"));
  134. }
  135. }
  136. var messageEnd = new ServiceBusMessage(new { id = input.Id, progress = "finish", code = code }.ToJsonString());
  137. messageEnd.ApplicationProperties.Add("name", "Exam");
  138. if (records.Count > 0)
  139. {
  140. long end = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", messageEnd, DateTimeOffset.FromUnixTimeMilliseconds(etime));
  141. await _serviceBus.GetServiceBusClient().cancelMessage("active-task", records[0].sequenceNumber);
  142. records[0].sequenceNumber = end;
  143. await _azureStorage.SaveOrUpdate<ChangeRecord>(records[0]);
  144. //await client.GetContainer("TEAMModelOS", "Common").ReplaceItemAsync(record, record.id, new Azure.Cosmos.PartitionKey($"{record.code}"));
  145. }
  146. else
  147. {
  148. long end = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", messageEnd, DateTimeOffset.FromUnixTimeMilliseconds(etime));
  149. ChangeRecord changeRecord = new ChangeRecord
  150. {
  151. RowKey = input.Id,
  152. PartitionKey = "going",
  153. sequenceNumber = end,
  154. msgId = messageEnd.MessageId
  155. };
  156. await _azureStorage.Save<ChangeRecord>(changeRecord);
  157. //await client.GetContainer("TEAMModelOS", "Common").CreateItemAsync(changeRecord, new Azure.Cosmos.PartitionKey($"{changeRecord.code}"));
  158. }
  159. }
  160. break;
  161. case "finish":
  162. for (int j = 0; j < info.subjects.Count; j++)
  163. {
  164. if (info.subjects[j].classCount == info.targetClassIds.Count)
  165. {
  166. ExamResult result = new ExamResult();
  167. //人数总和
  168. int Count = 0;
  169. int m = 0;
  170. List<ClassRange> classRanges = new List<ClassRange>();
  171. foreach (ExamClassResult classResult in examClassResults)
  172. {
  173. if (classResult.subjectId.Equals(info.subjects[j].id))
  174. {
  175. foreach (List<double> scores in classResult.studentScores)
  176. {
  177. result.studentScores.Add(scores);
  178. }
  179. //处理班级信息
  180. ClassRange range = new ClassRange();
  181. range.id = classResult.info.id;
  182. range.name = classResult.info.name;
  183. List<int> ran = new List<int>();
  184. int stuCount = classResult.studentIds.Count;
  185. Count += stuCount;
  186. if (m == 0)
  187. {
  188. ran.Add(0);
  189. ran.Add(stuCount - 1);
  190. }
  191. else
  192. {
  193. ran.Add(Count - stuCount);
  194. ran.Add(Count - 1);
  195. }
  196. m++;
  197. range.range = ran;
  198. classRanges.Add(range);
  199. //处理学生ID
  200. foreach (string id in classResult.studentIds)
  201. {
  202. result.studentIds.Add(id);
  203. }
  204. }
  205. }
  206. result.classes = classRanges;
  207. result.ttl = -1;
  208. result.pk = typeof(ExamResult).Name;
  209. result.code = "ExamResult-" + info.id;
  210. result.school = info.school;
  211. result.id = info.subjects[j].id;
  212. result.examId = info.id;
  213. result.subjectId = info.subjects[j].id;
  214. result.year = info.year;
  215. result.paper = info.papers[j];
  216. //result.point = info.papers[j].point;
  217. result.scope = info.scope;
  218. result.name = info.name;
  219. result.time = info.startTime;
  220. await _azureCosmos.GetCosmosClient().GetContainer("TEAMModelOS", "Common").UpsertItemAsync(result, new Azure.Cosmos.PartitionKey($"ExamResult-{info.id}"));
  221. }
  222. }
  223. break;
  224. }
  225. break;
  226. case "Vote":
  227. Vote vote = await client.GetContainer("TEAMModelOS", "Common").ReadItemAsync<Vote>(input.Id, new Azure.Cosmos.PartitionKey($"{code}"));
  228. List<ChangeRecord> voteRecords = await _azureStorage.FindListByDict<ChangeRecord>(new Dictionary<string, object>() { { "RowKey", input.Id }, { "PartitionKey", vote.progress } });
  229. //ChangeRecord voteRecord = await client.GetContainer("TEAMModelOS", "Common").ReadItemAsync<ChangeRecord>(input.Id, new Azure.Cosmos.PartitionKey($"{vote.progress}"));
  230. switch (vote.progress)
  231. {
  232. case "pending":
  233. var messageVote = new ServiceBusMessage(new { id = input.Id, progress = "going", code = code }.ToJsonString());
  234. messageVote.ApplicationProperties.Add("name", "Vote");
  235. if (voteRecords.Count > 0)
  236. {
  237. long start = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", messageVote, DateTimeOffset.FromUnixTimeMilliseconds(stime));
  238. await _serviceBus.GetServiceBusClient().cancelMessage("active-task", voteRecords[0].sequenceNumber);
  239. voteRecords[0].sequenceNumber = start;
  240. await _azureStorage.SaveOrUpdate<ChangeRecord>(voteRecords[0]);
  241. //await client.GetContainer("TEAMModelOS", "Common").ReplaceItemAsync(voteRecord, voteRecord.id, new Azure.Cosmos.PartitionKey($"{voteRecord.code}"));
  242. }
  243. else
  244. {
  245. long start = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", messageVote, DateTimeOffset.FromUnixTimeMilliseconds(stime));
  246. ChangeRecord changeRecord = new ChangeRecord
  247. {
  248. RowKey = input.Id,
  249. PartitionKey = "pending",
  250. sequenceNumber = start,
  251. msgId = messageVote.MessageId
  252. };
  253. await _azureStorage.Save<ChangeRecord>(changeRecord);
  254. //await client.GetContainer("TEAMModelOS", "Common").CreateItemAsync(changeRecord, new Azure.Cosmos.PartitionKey($"{changeRecord.code}"));
  255. }
  256. break;
  257. case "going":
  258. var messageVoteEnd = new ServiceBusMessage(new { id = input.Id, progress = "finish", code = code }.ToJsonString());
  259. messageVoteEnd.ApplicationProperties.Add("name", "Vote");
  260. if (voteRecords.Count > 0)
  261. {
  262. long end = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", messageVoteEnd, DateTimeOffset.FromUnixTimeMilliseconds(etime));
  263. await _serviceBus.GetServiceBusClient().cancelMessage("active-task", voteRecords[0].sequenceNumber);
  264. voteRecords[0].sequenceNumber = end;
  265. await _azureStorage.SaveOrUpdate<ChangeRecord>(voteRecords[0]);
  266. //await client.GetContainer("TEAMModelOS", "Common").ReplaceItemAsync(voteRecord, voteRecord.id, new Azure.Cosmos.PartitionKey($"{voteRecord.code}"));
  267. }
  268. else
  269. {
  270. long end = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", messageVoteEnd, DateTimeOffset.FromUnixTimeMilliseconds(etime));
  271. ChangeRecord changeRecord = new ChangeRecord
  272. {
  273. RowKey = input.Id,
  274. PartitionKey = "going",
  275. sequenceNumber = end,
  276. msgId = messageVoteEnd.MessageId
  277. };
  278. await _azureStorage.Save<ChangeRecord>(changeRecord);
  279. //await client.GetContainer("TEAMModelOS", "Common").CreateItemAsync(changeRecord, new Azure.Cosmos.PartitionKey($"{changeRecord.code}"));
  280. }
  281. break;
  282. }
  283. break;
  284. case "Survey":
  285. Survey survey = await client.GetContainer("TEAMModelOS", "Common").ReadItemAsync<Survey>(input.Id, new Azure.Cosmos.PartitionKey($"{code}"));
  286. //messageSurvey.ScheduledEnqueueTime = DateTimeOffset.FromUnixTimeMilliseconds(stime);
  287. //string msgid = messageSurvey.MessageId;
  288. List<ChangeRecord> changeRecords = await _azureStorage.FindListByDict<ChangeRecord>(new Dictionary<string, object>() { { "RowKey", input.Id }, { "PartitionKey", survey.progress } });
  289. //ChangeRecord surveyRecord = await client.GetContainer("TEAMModelOS", "Common").ReadItemAsync<ChangeRecord>(input.Id, new Azure.Cosmos.PartitionKey($"{survey.progress}"));
  290. switch (survey.progress)
  291. {
  292. case "pending":
  293. var messageSurvey = new ServiceBusMessage(new { id = input.Id, progress = "going", code = code }.ToJsonString());
  294. messageSurvey.ApplicationProperties.Add("name", "Survey");
  295. if (changeRecords.Count > 0)
  296. {
  297. await _serviceBus.GetServiceBusClient().cancelMessage("active-task", changeRecords[0].sequenceNumber);
  298. long start = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", messageSurvey, DateTimeOffset.FromUnixTimeMilliseconds(stime));
  299. changeRecords[0].sequenceNumber = start;
  300. await _azureStorage.SaveOrUpdate<ChangeRecord>(changeRecords[0]);
  301. //await client.GetContainer("TEAMModelOS", "Common").ReplaceItemAsync(surveyRecord, surveyRecord.id, new Azure.Cosmos.PartitionKey($"{surveyRecord.code}"));
  302. }
  303. else
  304. {
  305. long start = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", messageSurvey, DateTimeOffset.FromUnixTimeMilliseconds(stime));
  306. ChangeRecord changeRecord = new ChangeRecord
  307. {
  308. RowKey = input.Id,
  309. PartitionKey = "pending",
  310. sequenceNumber = start,
  311. msgId = messageSurvey.MessageId
  312. };
  313. await _azureStorage.Save<ChangeRecord>(changeRecord);
  314. //await client.GetContainer("TEAMModelOS", "Common").CreateItemAsync(changeRecord, new Azure.Cosmos.PartitionKey($"{changeRecord.code}"));
  315. }
  316. break;
  317. case "going":
  318. var messageSurveyEnd = new ServiceBusMessage(new { id = input.Id, progress = "finish", code = code }.ToJsonString());
  319. messageSurveyEnd.ApplicationProperties.Add("name", "Survey");
  320. if (changeRecords.Count > 0)
  321. {
  322. long end = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", messageSurveyEnd, DateTimeOffset.FromUnixTimeMilliseconds(etime));
  323. await _serviceBus.GetServiceBusClient().cancelMessage("active-task", changeRecords[0].sequenceNumber);
  324. changeRecords[0].sequenceNumber = end;
  325. await _azureStorage.SaveOrUpdate<ChangeRecord>(changeRecords[0]);
  326. }
  327. else
  328. {
  329. long end = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", messageSurveyEnd, DateTimeOffset.FromUnixTimeMilliseconds(etime));
  330. ChangeRecord changeRecord = new ChangeRecord
  331. {
  332. RowKey = input.Id,
  333. PartitionKey = "going",
  334. sequenceNumber = end,
  335. msgId = messageSurveyEnd.MessageId
  336. };
  337. await _azureStorage.Save<ChangeRecord>(changeRecord);
  338. //await client.GetContainer("TEAMModelOS", "Common").CreateItemAsync(changeRecord, new Azure.Cosmos.PartitionKey($"{changeRecord.code}"));
  339. }
  340. break;
  341. }
  342. break;
  343. }
  344. }
  345. }
  346. }
  347. }
  348. }
  349. }