TEAMModelOS/TEAMModelBI/Controllers/Census/BlobLogController.cs

using Azure.Cosmos;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Options;
using StackExchange.Redis;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Text.Json;
using System.Threading.Tasks;
using TEAMModelBI.Tool;
using TEAMModelOS.Models;
using TEAMModelOS.SDK.Context.BI;
using TEAMModelOS.SDK.DI;
using TEAMModelOS.SDK.Extension;
using TEAMModelOS.SDK.Models.Cosmos.BI;
using TEAMModelOS.SDK.Models.Service.BI;

namespace TEAMModelBI.Controllers.Census
{
    [Route("bloblog")]
    [ApiController]
    public class BlobLogController : ControllerBase
    {
        private readonly AzureCosmosFactory _azureCosmos;
        private readonly AzureStorageFactory _azureStorage;
        private readonly DingDing _dingDing;
        private readonly Option _option;
        private readonly AzureRedisFactory _azureRedis;

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

        /// <summary>
        /// 空间统计数量统计   已对接
        /// </summary>
        /// <param name="jsonElement"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [HttpPost("get-area")]
        public async Task<IActionResult> GetArea(JsonElement jsonElement)
        {
            jsonElement.TryGetProperty("site", out JsonElement site);
            jsonElement.TryGetProperty("areaId", out JsonElement areaId);
            var cosmosClient = _azureCosmos.GetCosmosClient();
            if ($"{site}".Equals(BIConst.Global))
                cosmosClient = _azureCosmos.GetCosmosClient(name: BIConst.Global);
            StringBuilder sqlSize = new($"select value(sum(c.size)) from c ");
            long useSize = 0;  //使用大小
            Dictionary<string, int> typeCnt = new();

            List<string> schoolIds = new();
            if (!string.IsNullOrEmpty($"{areaId}"))
            {
                string sqlTxt = $"select c.id from c where c.areaId='{areaId}'";
                sqlSize.Append($" where c.areaId = '{areaId}'");
                schoolIds = await CommonFind.FindSchoolIds(cosmosClient, sqlTxt.ToString(), "Base");
                foreach (var item in StaticValue.fileType)
                {
                    int fileCnt = 0;
                    if (schoolIds.Count > 0)
                    {
                        string inScStr = BICommonWay.ManyScSql(" REPLACE(c.code, 'Bloblog-', '')", schoolIds);
                        string typeSql = $"select value(count(c.id)) from c where c.pk='Bloblog' and {inScStr} and c.type='{item}'";
                        fileCnt = await CommonFind.GetSqlValueCount(cosmosClient, new List<string>() { "School", "Teacher" }, typeSql);
                    }
                    typeCnt.Add(item, fileCnt);
                }

                foreach (var scId in schoolIds)
                {
                    long blobsize = 0;
                    RedisValue value = _azureRedis.GetRedisClient(8).HashGet($"Blob:Record", $"{scId}");
                    if (!value.IsNullOrEmpty)
                    {
                        JsonElement record = value.ToString().ToObject<JsonElement>();
                        if (record.TryGetInt64(out blobsize))
                        {
                            useSize += blobsize;
                        }
                    }
                }
            }
            else
            {
                foreach (var item in StaticValue.fileType)
                {
                    string typeSql = $"select value(count(c.id)) from c where c.pk='Bloblog' and c.type='{item}'";
                    int fileCnt = await CommonFind.GetSqlValueCount(cosmosClient, new List<string>() { "School", "Teacher" }, typeSql);

                    typeCnt.Add(item, fileCnt);
                }
                schoolIds = await CommonFind.FindSchoolIds(cosmosClient, $"select c.id from c", "Base");
                foreach (var id in schoolIds)
                {
                    long blobsize = 0;
                    RedisValue value = _azureRedis.GetRedisClient(8).HashGet($"Blob:Record", $"{id}");
                    if (!value.IsNullOrEmpty)
                    {
                        JsonElement record = value.ToString().ToObject<JsonElement>();
                        if (record.TryGetInt64(out blobsize))
                        {
                            useSize += blobsize;
                        }
                    }
                }
            }

            var areaSize = await CommonFind.GetSqlValueCount(cosmosClient, "School", sqlSize.ToString(), "Base");

            return Ok(new { state = 200, areaSize, useSize, typeCount = typeCnt.ToList() });
        }

        /// <summary>
        /// 依据教师获取教师个人的记录
        /// </summary>
        /// <param name="cosmosClient"></param>
        /// <param name="teacherIds"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        public async static Task<List<RecBlobFile>> GetBlobTeache(CosmosClient cosmosClient, List<string> teacherIds)
        {
            List<RecBlobFile> blobFiles = new();
            if (teacherIds.Count > 0)
            {
                foreach (var teacherId in teacherIds)
                {
                    await foreach (var item in cosmosClient.GetContainer("TEAMModelOS", "Teacher").GetItemQueryIterator<RecBlobFile>(queryText: $"SELECT c.id,c.code,c.name,c.size,c.type FROM c where c.pk='Bloblog'", requestOptions: new QueryRequestOptions() { PartitionKey = new PartitionKey($"Bloblog-{teacherId}") }))
                    {
                        blobFiles.Add(item);
                    }
                }
            }

            return blobFiles;
        }



        /// <summary>
        /// 空间内容信息
        /// </summary>
        public record RecBlobFile
        {
            public string id { get; set; }
            public string code { get; set; }
            public string name { get; set; }
            public int size { get; set; }
            public string type { get; set; }

        }


    }

}