TEAMModelOS/TEAMModelBI/Controllers/Lesson/LessonSticsController.cs

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

namespace TEAMModelBI.Controllers.Lesson
{
    [Route("lesson")]
    [ApiController]
    public class LessonSticsController : ControllerBase
    {

        private readonly AzureCosmosFactory _azureCosmos;
        private readonly AzureStorageFactory _azureStorage;
        private readonly DingDing _dingDing;
        private readonly Option _option;

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

        /// <summary>
        /// 统计课例数量
        /// </summary>
        /// <returns></returns>
        [HttpPost("get-total")]
        public async Task<IActionResult> GetCount()
        {
            try
            {
                List<string> schoolIds = new List<string>();
                var cosmosClient = _azureCosmos.GetCosmosClient();
                string lessonSql = $"select c.id,c.tmid,c.scope from c where c.pk='LessonRecord'";
                int total = 0;
                await foreach (var item in cosmosClient.GetContainer(Constant.TEAMModelOS, "School").GetItemQueryStreamIterator(queryText: lessonSql, requestOptions: new QueryRequestOptions() { }))
                {
                    using var json = await JsonDocument.ParseAsync(item.ContentStream);
                    if (json.RootElement.TryGetProperty("_count", out JsonElement count) && count.GetInt32() > 0)
                    {
                        total = count.GetInt32();
                    }
                }

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

        /// <summary>
        /// 管家所关联的课例数据
        /// </summary>
        /// <param name="jsonElement"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [HttpPost("get-assiist")]
        public async Task<IActionResult> GetAssiist(JsonElement jsonElement) 
        {
            try
            {
                if (!jsonElement.TryGetProperty("tmdId", out JsonElement tmdId)) return BadRequest();
                var cosmosClient = _azureCosmos.GetCosmosClient();
                List<SchoolLen> schoolLens = new List<SchoolLen>();

                List<string> schoolIds = new List<string>();

                string schoolSql = $"SELECT DISTINCT REPLACE(c.code,'Teacher-','') AS schoolId,c.code,c.roles,c.id,c.name From c where ARRAY_CONTAINS(c.roles,'assist',true) AND c.pk = 'Teacher' AND c.status = 'join' AND c.id='{tmdId}'";

                await foreach (var item in cosmosClient.GetContainer(Constant.TEAMModelOS, "School").GetItemQueryStreamIterator(queryText: schoolSql, requestOptions: new QueryRequestOptions() { }))
                {
                    using var json = await JsonDocument.ParseAsync(item.ContentStream);
                    foreach (var obj in json.RootElement.GetProperty("Documents").EnumerateArray())
                    {
                        schoolIds.Add(obj.GetProperty("schoolId").GetString());
                    }
                }

                foreach (var itemId in schoolIds)
                {
                    SchoolLen schoolLen = new SchoolLen() { id = itemId };
                    School school = new();
                    try
                    {
                        school = await cosmosClient.GetContainer(Constant.TEAMModelOS, "School").ReadItemAsync<School>(itemId, new PartitionKey("Base"));
                    }
                    catch
                    {
                    }
                    schoolLen.name = school != null ? school.name : itemId;

                    await foreach (var item in cosmosClient.GetContainer(Constant.TEAMModelOS, "School").GetItemQueryStreamIterator(queryText: "SELECT DISTINCT c.id,c.code,c.name FROM c", requestOptions: new QueryRequestOptions() { PartitionKey = new PartitionKey($"LessonRecord-{itemId}") }))
                    {
                        using var json = await JsonDocument.ParseAsync(item.ContentStream);
                        if (json.RootElement.TryGetProperty("_count", out JsonElement count) && count.GetInt32() > 0)
                        {
                            schoolLen.total = count.GetInt32();
                        }
                    }

                    schoolLens.Add(schoolLen);
                }

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

            }
            catch (Exception ex)
            {
                await _dingDing.SendBotMsg($"BI， {_option.Location}  /lesson/get-assiist  {ex.Message}{ex.StackTrace}", GroupNames.成都开发測試群組);
                return BadRequest();
            }
        }

        public record SchoolLen 
        {
            public string id { get; set; }

            public string name { get; set;}

            public int total { get; set; }
        }

    }
}