TEAMModelOS/TEAMModelBI/Controllers/BISchool/AreaRelevantController.cs

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

namespace TEAMModeBI.Controllers.BISchool
{
    [Route("area")]
    [ApiController]
    public class AreaRelevantController : ControllerBase
    {
        //数据容器
        private readonly AzureCosmosFactory _azureCosmos;
        private readonly AzureStorageFactory _azureStorage;
        //钉钉提示信息
        private readonly DingDing _dingDing;
        private readonly Option _option;

        public AreaRelevantController(AzureCosmosFactory azureCosmos, AzureStorageFactory azureStorage, DingDing dingDing, IOptionsSnapshot<Option> option)
        {
            _azureCosmos = azureCosmos;
            _azureStorage = azureStorage;
            _dingDing = dingDing;
            _option = option?.Value;
        }

        /// <summary>
        /// 查询区域已经有的学校
        /// </summary>
        /// <param name="jsonElement"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [HttpPost("get-areaschools")]
        public async Task<IActionResult> GetAreaSchools(JsonElement jsonElement) 
        {
            try
            {
                if (!jsonElement.TryGetProperty("areaId", out JsonElement _areaId)) return BadRequest();
                var cosmosClient = _azureCosmos.GetCosmosClient();
                List<JoinAreaSchool> joinAreaSchools = new List<JoinAreaSchool>();
                string slqtxt = $"SELECT c.id,c.name,c.schoolCode,c.province,c.city,c.dist,c.picture,c.period FROM c WHERE c.areaId='{_areaId}'";
                await foreach (var item in cosmosClient.GetContainer(Constant.TEAMModelOS, "School").GetItemQueryStreamIterator(queryText: slqtxt,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())
                        {
                            JoinAreaSchool joinAreaSchool = new JoinAreaSchool
                            {
                                id = obj.GetProperty("id").GetString(),
                                name = obj.GetProperty("name").GetString(),
                                schoolCode = obj.GetProperty("schoolCode").GetString(),
                                province = obj.GetProperty("province").GetString(),
                                city = obj.GetProperty("city").GetString(),
                                dist = obj.GetProperty("dist").GetString(),
                                picture = obj.GetProperty("picture").GetString(),
                                period = obj.GetProperty("period").ToObject<List<Period>>().Select(x => x.name).ToList()
                            };
                            joinAreaSchools.Add(joinAreaSchool);
                        }                    
                    }
                }

                return Ok(new { state = 200, joinAreaSchools });
            }
            catch (Exception ex)
            {
                await _dingDing.SendBotMsg($"BI， {_option.Location} /area/get-areaschools  \n {ex.Message}{ex.StackTrace}", GroupNames.成都开发測試群組);
                return BadRequest();
            }        
        }

        /// <summary>
        /// 学校移出区域
        /// </summary>
        /// <param name="jsonElement"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [AuthToken(Roles = "assist")]
        [HttpPost("set-areashiftschool")]
        public async Task<IActionResult> SetAreaShiftSchool(JsonElement jsonElement)
        {
            try
            {
                if (!jsonElement.TryGetProperty("tmdId", out JsonElement _tmdId)) return BadRequest();   //醍摩豆账户
                if (!jsonElement.TryGetProperty("tmdName", out JsonElement _tmdName)) return BadRequest(); //醍摩豆账号名称
                //if (!jsonElement.TryGetProperty("areaId", out JsonElement areaId)) return BadRequest();
                if (!jsonElement.TryGetProperty("schoolId", out JsonElement schoolId)) return BadRequest();

                var cosmosClient = _azureCosmos.GetCosmosClient();
                School tempSchool = await cosmosClient.GetContainer(Constant.TEAMModelOS, "School").ReadItemAsync<School>($"{schoolId}", new PartitionKey("Base"));

                tempSchool.areaId = null;
                tempSchool.standard = null;

                School school = await cosmosClient.GetContainer(Constant.TEAMModelOS, "School").ReplaceItemAsync<School>(tempSchool, tempSchool.id, new PartitionKey("Base"));

                //保存操作记录
                await _azureStorage.SaveLog("area-update", $"{_tmdName}【{_tmdId}】已操作学校({schoolId})移除该区域", _dingDing, httpContext: HttpContext);

                return Ok(new { state = 200, school });
            }
            catch (Exception ex)
            {
                await _dingDing.SendBotMsg($"BI， {_option.Location}   /area/set-areashiftschool \n  {ex.Message}{ex.StackTrace}", GroupNames.成都开发測試群組);
                return BadRequest();
            }
        }

        /// <summary>
        /// 已加入区域的学校
        /// </summary>
        public record JoinAreaSchool
        {
            public string id { get; set; }

            public string name { get; set; }

            public string schoolCode { get; set; }

            public string picture { get; set; }

            public List<string> period { get; set; }

            public string province { get; set; }

            public string city { get; set; }

            public string dist { get; set; }


        }

    }
}