TestController.cs 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442
  1. using Azure.Cosmos;
  2. using Microsoft.AspNetCore.Http;
  3. using Microsoft.AspNetCore.Mvc;
  4. using Newtonsoft.Json;
  5. using OS.Funct;
  6. using System;
  7. using System.Collections.Generic;
  8. using System.IO;
  9. using System.Linq;
  10. using System.Text.Json;
  11. using System.Threading.Tasks;
  12. using TEAMModelOS.SDK.DI;
  13. using TEAMModelOS.SDK.Extension;
  14. using TEAMModelOS.SDK.Helper.Common.CollectionHelper;
  15. using TEAMModelOS.SDK.Models;
  16. using TEAMModelOS.SDK.Models.Cosmos.Common;
  17. namespace TEAMModelOS.Controllers.XTest
  18. {
  19. [Route("test")]
  20. [ApiController]
  21. public class TestController :ControllerBase
  22. {
  23. private readonly AzureStorageFactory _azureStorage;
  24. private readonly AzureRedisFactory _azureRedis;
  25. private readonly AzureCosmosFactory _azureCosmos;
  26. public TestController(AzureCosmosFactory azureCosmos, AzureRedisFactory azureRedis, AzureStorageFactory azureStorage) {
  27. _azureCosmos = azureCosmos;
  28. _azureRedis = azureRedis;
  29. _azureStorage = azureStorage;
  30. }
  31. /// <summary>
  32. /// 测试blob多线程写入同一个文件
  33. /// </summary>
  34. /// <returns></returns>
  35. [ProducesDefaultResponseType]
  36. [HttpGet("multiple-blob")]
  37. public async Task<IActionResult> MultipleBlob() {
  38. await _azureStorage.GetBlobContainerClient("hbcn").List("other");
  39. var dn=await _azureStorage.GetBlobContainerClient("hbcn").GetBlobClient("survey/2e44ee33-ba65-34b6-7e0f-b7e627c70a54/record.json").DownloadAsync();
  40. var jsonc = await JsonDocument.ParseAsync( dn.Value.Content);
  41. var Recordc = jsonc.RootElement.ToObject<JsonElement>();
  42. return Ok();
  43. }
  44. /// <summary>
  45. /// 测试redis通配符
  46. /// </summary>
  47. /// <param name="request"></param>
  48. /// <returns></returns>
  49. [ProducesDefaultResponseType]
  50. [HttpGet("test-redis")]
  51. public async Task<IActionResult> TestRedis( ) {
  52. var db = _azureRedis.GetRedisClient(8);
  53. await db.StringSetAsync("1111", "111111");
  54. return Ok() ;
  55. }
  56. /// <summary>
  57. /// 删除
  58. /// </summary>
  59. /// <param name="request"></param>
  60. /// <returns></returns>
  61. [ProducesDefaultResponseType]
  62. //[AuthToken(Roles = "teacher")]
  63. [HttpPost("delete")]
  64. public async Task<IActionResult> Delete(JsonElement request)
  65. {
  66. try
  67. {
  68. if (!request.TryGetProperty("code", out JsonElement code)) return BadRequest();
  69. if (!request.TryGetProperty("scope", out JsonElement scope)) return BadRequest();
  70. var client = _azureCosmos.GetCosmosClient();
  71. List<Task> tasksFiles = new List<Task>();
  72. var query = $"select c.id from c ";
  73. if (scope.ToString().Equals("school"))
  74. {
  75. var ids = client.GetContainer("TEAMModelOS", "School").GetItemQueryStreamIterator(queryText: query, requestOptions: new QueryRequestOptions() { PartitionKey = new PartitionKey($"{code}") });
  76. await foreach (var id in ids)
  77. {
  78. using var json = await JsonDocument.ParseAsync(id.ContentStream);
  79. var jsonList = json.RootElement.GetProperty("Documents").EnumerateArray();
  80. //批量删除
  81. foreach (var obj in jsonList)
  82. {
  83. tasksFiles.Add(client.GetContainer("TEAMModelOS", "School").DeleteItemStreamAsync(obj.GetProperty("id").ToString(), new PartitionKey($"{code}")));
  84. }
  85. }
  86. }
  87. else if (scope.ToString().Equals("teacher")) {
  88. var ids = client.GetContainer("TEAMModelOS", "Teacher").GetItemQueryStreamIterator(queryText: query, requestOptions: new QueryRequestOptions() { PartitionKey = new PartitionKey($"{code}") });
  89. await foreach (var id in ids)
  90. {
  91. using var json = await JsonDocument.ParseAsync(id.ContentStream);
  92. var jsonList = json.RootElement.GetProperty("Documents").EnumerateArray();
  93. //批量删除
  94. foreach (var obj in jsonList)
  95. {
  96. //tasksFiles.Add(client.GetContainer("TEAMModelOS", "Teacher").DeleteItemStreamAsync(obj.GetProperty("id").ToString(), new PartitionKey($"{code}")));
  97. }
  98. }
  99. }
  100. await Task.WhenAll(tasksFiles);
  101. return Ok(new { code = 1 });
  102. }
  103. catch (Exception e)
  104. {
  105. return BadRequest(e.StackTrace);
  106. }
  107. }
  108. [ProducesDefaultResponseType]
  109. //[AuthToken(Roles = "teacher")]
  110. [HttpPost("fix-stu-course")]
  111. public async Task<IActionResult> fixStuCourse(JsonElement request) {
  112. string originCode = request.GetProperty("originCode").GetString();
  113. List<Course> courses = new List<Course>();
  114. var client = _azureCosmos.GetCosmosClient();
  115. var query = $"select * from c ";
  116. await foreach (var item in client.GetContainer("TEAMModelOS", "School").GetItemQueryIterator<Course>(queryText: query,
  117. requestOptions: new QueryRequestOptions() { PartitionKey = new Azure.Cosmos.PartitionKey($"Course-{originCode}") }))
  118. {
  119. courses.Add(item);
  120. }
  121. await foreach (var item in client.GetContainer("TEAMModelOS", "Teacher").GetItemQueryIterator<Course>(queryText: query,
  122. requestOptions: new QueryRequestOptions() { PartitionKey = new Azure.Cosmos.PartitionKey($"Course-{originCode}") }))
  123. {
  124. courses.Add(item);
  125. }
  126. //2.获取课程的id 并尝试添加或移除对应的学生课程记录StuCourse。
  127. foreach (var course in courses)
  128. {
  129. if (course.schedule.IsNotEmpty())
  130. {
  131. foreach (var sc in course.schedule)
  132. {
  133. if (!string.IsNullOrEmpty(sc.stulist))
  134. {
  135. (List<string> tmdids, List<Students> studentss) = await TriggerStuActivity.GetStuList(client, new List<string>() { sc.stulist }, course.school);
  136. foreach (var addStu in studentss)
  137. {
  138. var stuCourse = new StuCourse
  139. {
  140. id = course.id,
  141. scode = course.code,
  142. name = course.name,
  143. code = $"StuCourse-{course.school}-{addStu.id}",
  144. scope = course.scope,
  145. school = course.school,
  146. creatorId = course.creatorId,
  147. pk = "StuCourse"
  148. };
  149. await client.GetContainer("TEAMModelOS", "Student").UpsertItemAsync(stuCourse, new PartitionKey(stuCourse.code));
  150. }
  151. foreach (var addTmd in tmdids)
  152. {
  153. var tmdCourse = new StuCourse
  154. {
  155. id = course.id,
  156. scode = course.code,
  157. name = course.name,
  158. code = $"StuCourse-{addTmd}",
  159. scope = course.scope,
  160. //school = courseChange.school,
  161. creatorId = course.creatorId,
  162. pk = "StuCourse"
  163. };
  164. await client.GetContainer("TEAMModelOS", "Teacher").UpsertItemAsync(tmdCourse, new PartitionKey(tmdCourse.code));
  165. }
  166. }
  167. }
  168. }
  169. }
  170. return new OkObjectResult(new { });
  171. }
  172. [ProducesDefaultResponseType]
  173. //[AuthToken(Roles = "teacher")]
  174. [HttpPost("fix-exam-activity")]
  175. public async Task<IActionResult> ExamActivity(JsonElement request)
  176. {
  177. var datas = request.ToObject<List<string>>();
  178. var client = _azureCosmos.GetCosmosClient();
  179. var query = $"select * from c ";
  180. foreach (string data in datas)
  181. {
  182. List<ExamInfo> exams = new List<ExamInfo>();
  183. await foreach (var item in client.GetContainer("TEAMModelOS", "Common").GetItemQueryStreamIterator(
  184. queryText: query, requestOptions: new QueryRequestOptions() { PartitionKey = new PartitionKey($"Exam-{data}") }))
  185. {
  186. using var json = await JsonDocument.ParseAsync(item.ContentStream);
  187. if (json.RootElement.TryGetProperty("_count", out JsonElement count) && count.GetUInt16() > 0)
  188. {
  189. foreach (var obj in json.RootElement.GetProperty("Documents").EnumerateArray())
  190. {
  191. exams.Add(obj.ToObject<ExamInfo>());
  192. }
  193. }
  194. }
  195. foreach (var info in exams)
  196. {
  197. if (!info.classes.IsNotEmpty())
  198. {
  199. continue;
  200. }
  201. List<string> sub = new List<string>();
  202. foreach (ExamSubject subject in info.subjects)
  203. {
  204. sub.Add(subject.id);
  205. }
  206. (List<string> tmdids, List<Students> studentss) = await TriggerStuActivity.GetStuList(client, info.classes, info.school);
  207. List<StuActivity> stuActivities = new List<StuActivity>();
  208. List<StuActivity> tmdActivities = new List<StuActivity>();
  209. if (tmdids.IsNotEmpty())
  210. {
  211. tmdids.ForEach(x => {
  212. tmdActivities.Add(new StuActivity
  213. {
  214. pk = "Activity",
  215. id = info.id,
  216. code = $"Activity-{x}",
  217. type = "exam",
  218. name = info.name,
  219. startTime = info.startTime,
  220. endTime = info.endTime,
  221. scode = info.code,
  222. scope = info.scope,
  223. school = info.school,
  224. creatorId = info.creatorId,
  225. subjects = sub,
  226. blob = null,
  227. owner = info.owner
  228. });
  229. });
  230. }
  231. if (studentss.IsNotEmpty())
  232. {
  233. studentss.ForEach(x => {
  234. stuActivities.Add(new StuActivity
  235. {
  236. pk = "Activity",
  237. id = info.id,
  238. code = $"Activity-{info.school}-{x.id}",
  239. type = "exam",
  240. name = info.name,
  241. startTime = info.startTime,
  242. endTime = info.endTime,
  243. scode = info.code,
  244. scope = info.scope,
  245. school = info.school,
  246. creatorId = info.creatorId,
  247. subjects = sub,
  248. blob = null,
  249. owner = info.owner
  250. });
  251. });
  252. }
  253. await TriggerStuActivity.SaveStuActivity(client, stuActivities, tmdActivities);
  254. }
  255. }
  256. return new OkObjectResult(new { });
  257. }
  258. [ProducesDefaultResponseType]
  259. //[AuthToken(Roles = "teacher")]
  260. [HttpPost("fix-vote-activity")]
  261. public async Task<IActionResult> VoteActivity(JsonElement request)
  262. {
  263. var datas = request.ToObject<List<string>>();
  264. var client = _azureCosmos.GetCosmosClient();
  265. var query = $"select * from c ";
  266. foreach (string data in datas)
  267. {
  268. List<Vote> votes = new List<Vote>();
  269. await foreach (var item in client.GetContainer("TEAMModelOS", "Common").GetItemQueryStreamIterator(
  270. queryText: query, requestOptions: new QueryRequestOptions() { PartitionKey = new PartitionKey($"Vote-{data}") }))
  271. {
  272. using var json = await JsonDocument.ParseAsync(item.ContentStream);
  273. if (json.RootElement.TryGetProperty("_count", out JsonElement count) && count.GetUInt16() > 0)
  274. {
  275. foreach (var obj in json.RootElement.GetProperty("Documents").EnumerateArray())
  276. {
  277. votes.Add(obj.ToObject<Vote>());
  278. }
  279. }
  280. }
  281. foreach (var info in votes)
  282. {
  283. if (!info.classes.IsNotEmpty())
  284. {
  285. continue;
  286. }
  287. (List<string> tmdids, List<Students> studentss) = await TriggerStuActivity.GetStuList(client, info.classes, info.school);
  288. List<StuActivity> stuActivities = new List<StuActivity>();
  289. List<StuActivity> tmdActivities = new List<StuActivity>();
  290. if (tmdids.IsNotEmpty())
  291. {
  292. tmdids.ForEach(x => {
  293. tmdActivities.Add(new StuActivity
  294. {
  295. pk = "Activity",
  296. id = info.id,
  297. code = $"Activity-{x}",
  298. type = "vote",
  299. name = info.name,
  300. startTime = info.startTime,
  301. endTime = info.endTime,
  302. scode = info.code,
  303. scope = info.scope,
  304. school = info.school,
  305. creatorId = info.creatorId,
  306. subjects = new List<string>() { "" },
  307. blob = null,
  308. owner = info.owner
  309. });
  310. });
  311. }
  312. if (studentss.IsNotEmpty())
  313. {
  314. studentss.ForEach(x => {
  315. stuActivities.Add(new StuActivity
  316. {
  317. pk = "Activity",
  318. id = info.id,
  319. code = $"Activity-{info.school}-{x.id}",
  320. type = "vote",
  321. name = info.name,
  322. startTime = info.startTime,
  323. endTime = info.endTime,
  324. scode = info.code,
  325. scope = info.scope,
  326. school = info.school,
  327. creatorId = info.creatorId,
  328. subjects = new List<string>() { "" },
  329. blob = null,
  330. owner = info.owner
  331. });
  332. });
  333. }
  334. await TriggerStuActivity.SaveStuActivity(client, stuActivities, tmdActivities);
  335. }
  336. }
  337. return new OkObjectResult(new { });
  338. }
  339. [ProducesDefaultResponseType]
  340. //[AuthToken(Roles = "teacher")]
  341. [HttpPost("fix-survey-activity")]
  342. public async Task<IActionResult> SurveyActivity(JsonElement request)
  343. {
  344. var datas = request.ToObject<List<string>>();
  345. var client = _azureCosmos.GetCosmosClient();
  346. var query = $"select * from c ";
  347. foreach (string data in datas)
  348. {
  349. List<Survey> surveys = new List<Survey>();
  350. await foreach (var item in client.GetContainer("TEAMModelOS", "Common").GetItemQueryStreamIterator(
  351. queryText: query, requestOptions: new QueryRequestOptions() { PartitionKey = new PartitionKey($"Survey-{data}") }))
  352. {
  353. using var json = await JsonDocument.ParseAsync(item.ContentStream);
  354. if (json.RootElement.TryGetProperty("_count", out JsonElement count) && count.GetUInt16() > 0)
  355. {
  356. foreach (var obj in json.RootElement.GetProperty("Documents").EnumerateArray())
  357. {
  358. surveys.Add(obj.ToObject<Survey>());
  359. }
  360. }
  361. }
  362. foreach (var info in surveys)
  363. {
  364. if (!info.classes.IsNotEmpty())
  365. {
  366. continue;
  367. }
  368. (List<string> tmdids, List<Students> studentss) = await TriggerStuActivity.GetStuList(client, info.classes, info.school);
  369. List<StuActivity> stuActivities = new List<StuActivity>();
  370. List<StuActivity> tmdActivities = new List<StuActivity>();
  371. if (tmdids.IsNotEmpty())
  372. {
  373. tmdids.ForEach(x => {
  374. tmdActivities.Add(new StuActivity
  375. {
  376. pk = "Activity",
  377. id = info.id,
  378. code = $"Activity-{x}",
  379. type = "survey",
  380. name = info.name,
  381. startTime = info.startTime,
  382. endTime = info.endTime,
  383. scode = info.code,
  384. scope = info.scope,
  385. school = info.school,
  386. creatorId = info.creatorId,
  387. subjects = new List<string>() { "" },
  388. blob = info.blob
  389. });
  390. });
  391. }
  392. if (studentss.IsNotEmpty())
  393. {
  394. studentss.ForEach(x => {
  395. stuActivities.Add(new StuActivity
  396. {
  397. pk = "Activity",
  398. id = info.id,
  399. code = $"Activity-{info.school}-{x.id}",
  400. type = "survey",
  401. name = info.name,
  402. startTime = info.startTime,
  403. endTime = info.endTime,
  404. scode = info.code,
  405. scope = info.scope,
  406. school = info.school,
  407. creatorId = info.creatorId,
  408. subjects = new List<string>() { "" },
  409. blob = info.blob
  410. });
  411. });
  412. }
  413. await TriggerStuActivity.SaveStuActivity(client, stuActivities, tmdActivities);
  414. }
  415. }
  416. return new OkObjectResult(new { });
  417. }
  418. }
  419. }