TEAMModelOS/TEAMModelBI/Controllers/QueryTool/SchoolController.cs

using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Options;
using System.Text.Json;
using System.Threading.Tasks;
using TEAMModelOS.Models;
using TEAMModelOS.SDK.DI;
using System.Text;
using TEAMModelOS.SDK.Models;
using Azure.Cosmos;
using System.Collections.Generic;
using TEAMModelOS.SDK.Extension;

namespace TEAMModelBI.Controllers.DataTool
{
    [Route("school")]
    [ApiController]
    public class SchoolController : ControllerBase
    {
        private readonly AzureCosmosFactory _azureCosmos;
        private readonly AzureStorageFactory _azureStorage;
        private readonly DingDing _dingDing;
        private readonly Option _option;
        public SchoolController(AzureCosmosFactory azureCosmos, AzureStorageFactory azureStorage, DingDing dingDing, IOptionsSnapshot<Option> option) 
        {
            _azureCosmos = azureCosmos;
            _azureStorage = azureStorage;
            _dingDing = dingDing;
            _option = option?.Value;        
        }

        /// <summary>
        /// 查询学校
        /// </summary>
        /// <returns></returns>
        [HttpPost("get-school")]
        public async Task<IActionResult> GetSchool(JsonElement jsonElement) 
        {
            jsonElement.TryGetProperty("schoolId", out JsonElement schoolId);
            var clientCosmosDB = _azureCosmos.GetCosmosClient();
            StringBuilder sqlTxt = new StringBuilder("select * from c");
            if (!string.IsNullOrEmpty($"{schoolId}")) 
            {
                sqlTxt.Append($" where c.id='{schoolId}'");
            }

            List<School> schools = new();

            await foreach (var item in clientCosmosDB.GetContainer("TEAMModelOS", "School").GetItemQueryStreamIterator(queryText: sqlTxt.ToString(), requestOptions: new QueryRequestOptions() { PartitionKey = new PartitionKey("Base") }))
            {
                using var json = await JsonDocument.ParseAsync(item.ContentStream);
                if (json.RootElement.TryGetProperty("_count", out JsonElement count) && count.GetUInt16() > 0)
                {
                    foreach (var obj in json.RootElement.GetProperty("Documents").EnumerateArray())
                    {
                        schools.Add(obj.ToObject<School>());
                    }
                }
            }

            return Ok(new { state = 200, schools });
        }




    }
}