SchoolController.cs 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. using Microsoft.AspNetCore.Http;
  2. using Microsoft.AspNetCore.Mvc;
  3. using Microsoft.Extensions.Options;
  4. using System.Text.Json;
  5. using System.Threading.Tasks;
  6. using TEAMModelOS.Models;
  7. using TEAMModelOS.SDK.DI;
  8. using System.Text;
  9. using TEAMModelOS.SDK.Models;
  10. using Azure.Cosmos;
  11. using System.Collections.Generic;
  12. using TEAMModelOS.SDK.Extension;
  13. namespace TEAMModelBI.Controllers.DataTool
  14. {
  15. [Route("school")]
  16. [ApiController]
  17. public class SchoolController : ControllerBase
  18. {
  19. private readonly AzureCosmosFactory _azureCosmos;
  20. private readonly AzureStorageFactory _azureStorage;
  21. private readonly DingDing _dingDing;
  22. private readonly Option _option;
  23. public SchoolController(AzureCosmosFactory azureCosmos, AzureStorageFactory azureStorage, DingDing dingDing, IOptionsSnapshot<Option> option)
  24. {
  25. _azureCosmos = azureCosmos;
  26. _azureStorage = azureStorage;
  27. _dingDing = dingDing;
  28. _option = option?.Value;
  29. }
  30. /// <summary>
  31. /// 查询学校
  32. /// </summary>
  33. /// <returns></returns>
  34. [HttpPost("get-school")]
  35. public async Task<IActionResult> GetSchool(JsonElement jsonElement)
  36. {
  37. jsonElement.TryGetProperty("schoolId", out JsonElement schoolId);
  38. var clientCosmosDB = _azureCosmos.GetCosmosClient();
  39. StringBuilder sqlTxt = new StringBuilder("select * from c");
  40. if (!string.IsNullOrEmpty($"{schoolId}"))
  41. {
  42. sqlTxt.Append($" where c.id='{schoolId}'");
  43. }
  44. List<School> schools = new();
  45. await foreach (var item in clientCosmosDB.GetContainer("TEAMModelOS", "School").GetItemQueryStreamIterator(queryText: sqlTxt.ToString(), requestOptions: new QueryRequestOptions() { PartitionKey = new PartitionKey("Base") }))
  46. {
  47. using var json = await JsonDocument.ParseAsync(item.ContentStream);
  48. if (json.RootElement.TryGetProperty("_count", out JsonElement count) && count.GetUInt16() > 0)
  49. {
  50. foreach (var obj in json.RootElement.GetProperty("Documents").EnumerateArray())
  51. {
  52. schools.Add(obj.ToObject<School>());
  53. }
  54. }
  55. }
  56. return Ok(new { state = 200, schools });
  57. }
  58. }
  59. }