TEAMModelOS/TEAMModeBI/Controllers/BIAbility/AbilityTaskMgmtController.cs

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

namespace TEAMModeBI.Controllers.BIAbility
{
    [Route("biabilitytask")]
    [ApiController]
    public class AbilityTaskMgmtController : ControllerBase
    {
        //获取配置信息
        private readonly IConfiguration _configuration;
        //数据容器
        private readonly AzureCosmosFactory _azureCosmos;
        //文件容器
        private readonly AzureStorageFactory _azureStorage;
        //钉钉提示信息
        private readonly DingDing _dingDing;
        private readonly Option _option;

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

        /// <summary>
        /// 依据区级标准和能力编号查询能力点信息 章节集合
        /// </summary>
        /// <param name="jsonElement"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [HttpPost("get-abilitytasksid")]
        public async Task<IActionResult> GetAbilityTasksId(JsonElement jsonElement)
        {
            try
            {
                if (!jsonElement.TryGetProperty("standard", out JsonElement standard)) return BadRequest();
                if (!jsonElement.TryGetProperty("abilityId", out JsonElement abilityId)) return BadRequest();
                var cosmosClient = _azureCosmos.GetCosmosClient();
                Ability ability = null;
                List<AbilityTaskTreeNode> abilityTaskTreeNodes = new List<AbilityTaskTreeNode>();//原始数据未排序
                List<AbilityTaskTreeNode> redtAbilityTaskTreeNodes = new List<AbilityTaskTreeNode>();//排序后的数据

                int rnodeCount = 0;
                ability = await cosmosClient.GetContainer(Constant.TEAMModelOS, "Normal").ReadItemAsync<Ability>($"{abilityId}", new PartitionKey($"Ability-{standard}"));
                await foreach (var item in cosmosClient.GetContainer(Constant.TEAMModelOS, "Normal").GetItemQueryIterator<AbilityTask>(queryText: $"select value(c) from c where c.abilityId='{abilityId}'", requestOptions: new QueryRequestOptions() { PartitionKey = new PartitionKey($"AbilityTask-{standard}") }))
                {
                    var rns = item.children.Select(x => x.rnodes);
                    if (rns != null)
                    {
                        var list = rns.ToList();
                        list.ForEach(x => { if (x.IsNotEmpty()) { rnodeCount += 1; } });
                    }
                    List<AbilityTaskTree> trees = AbilityService.ListToTree(item.children);
                    AbilityTaskTreeNode abilityTaskTreeNode = new AbilityTaskTreeNode() { id = item.id, scope = item.scope, trees = trees, abilityId = item.abilityId, auth = item.auth, codeval = ability.school };
                    abilityTaskTreeNodes.Add(abilityTaskTreeNode);
                }

                if (ability.abilityTaskIds.IsNotEmpty())
                {
                    ability.abilityTaskIds.ForEach(x =>
                    {
                        for (int i = 0; i < abilityTaskTreeNodes.Count; i++)
                        {
                            if (abilityTaskTreeNodes[i].id.Equals(x))
                            {
                                redtAbilityTaskTreeNodes.Add(abilityTaskTreeNodes[i]);
                                abilityTaskTreeNodes.RemoveAt(i);
                            }
                        }
                    });
                    redtAbilityTaskTreeNodes.AddRange(abilityTaskTreeNodes);
                    return Ok(new { state = 200, abilityTaskTreeNodes = redtAbilityTaskTreeNodes });
                }
                else
                {
                    return Ok(new { state = 200, abilityTaskTreeNodes = abilityTaskTreeNodes, rnodeCount });
                }
            }
            catch (Exception ex)
            {
                await _dingDing.SendBotMsg($"BI,{_option.Location} /biabilitytask/get-abilitytasksid \n {ex.Message}{ex.StackTrace}", GroupNames.醍摩豆服務運維群組);
                return BadRequest();
            }
        }

        /// <summary>
        /// 修改和新增章节下的节点数据
        /// </summary>
        /// <param name="abilityTask"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [HttpPost("upd-abilitytask")]
        public async Task<IActionResult> UpdAbilityTask(RecordAbilityTask recordAbilityTask)//AbilityTask abilityTask) 
        {
            AbilityTask abilityTask = recordAbilityTask.abilityTask;
            var cosmosClient = _azureCosmos.GetCosmosClient();
            //操作记录
            OperateLog operateLog = new OperateLog();
            string blobOrTable = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds().ToString();

            operateLog.PartitionKey = "OperateLog-BI";
            operateLog.RowKey = blobOrTable;
            operateLog.recordID = blobOrTable;
            operateLog.platformSource = "BI";
            operateLog.tmdId = $"{recordAbilityTask.tmdId}";
            operateLog.tmdName = $"{recordAbilityTask.tmdName}";

            //修改
            if (abilityTask.id != null)
            {
                try
                {
                    //AbilityTask tempAbilityTask = await cosmosClient.GetContainer(Constant.TEAMModelOS, "Normal").ReadItemAsync<AbilityTask>(abilityTask.id, new PartitionKey($"AbilityTask-{abilityTask.standard}"));


                    AbilityTask AbilityTask = await cosmosClient.GetContainer(Constant.TEAMModelOS, "Normal").ReplaceItemAsync<AbilityTask>(abilityTask,abilityTask.id, new PartitionKey($"AbilityTask-{abilityTask.standard}"));
                     
                    operateLog.operateDescribe = $"{recordAbilityTask.tmdName}【{recordAbilityTask.tmdId}】已操作修改章节功能，章节ID和分区键{abilityTask.id}And{abilityTask.code} ";
                    operateLog.visitApi = "biabilitytask/upd-abilitytask";
                    operateLog.operateTime = DateTime.Now;
                    await _azureStorage.Save<OperateLog>(operateLog);

                    return Ok(new { state = 200, abilityTask = AbilityTask });
                }
                catch (Exception ex)
                {
                    await _dingDing.SendBotMsg($"BI,{_option.Location}  biabilitytask/upd-abilitytask \n {ex.Message}{ex.StackTrace}", GroupNames.醍摩豆服務運維群組);
                    return BadRequest();
                }
            }
            //新增
            else
            {
                try
                {
                    AbilityTask tempAbilityTask = await cosmosClient.GetContainer(Constant.TEAMModelOS, "Normal").CreateItemAsync<AbilityTask>(abilityTask, new PartitionKey($"AbilityTask-{abilityTask.standard}"));
                    operateLog.operateDescribe = $"{recordAbilityTask.tmdName}【{recordAbilityTask.tmdId}】已操作新增章节功能，章节ID和分区键{tempAbilityTask.id}And{tempAbilityTask.code} ";
                    operateLog.visitApi = "biabilitytask/upd-abilitytask";
                    operateLog.operateTime = DateTime.Now;
                    await _azureStorage.Save<OperateLog>(operateLog);

                    return Ok(new { state = 200, abilityTask = tempAbilityTask });
                }
                catch (Exception ex)
                {
                    await _dingDing.SendBotMsg($"BI,{_option.Location}  biabilitytask/upd-abilitytask \n {ex.Message}{ex.StackTrace}", GroupNames.醍摩豆服務運維群組);
                    return BadRequest();
                }
            }
        }

        /// <summary>
        /// 依据id删除章节
        /// </summary>
        /// <param name="jsonElement"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [HttpPost("del-abilitytask")]
        public async Task<IActionResult> DelAbilityTask(JsonElement jsonElement) 
        {
            try
            {
                if (!jsonElement.TryGetProperty("standard", out JsonElement standard)) return BadRequest();
                if (!jsonElement.TryGetProperty("id", out JsonElement id)) return BadRequest();
                if (!jsonElement.TryGetProperty("tmdId", out JsonElement _tmdId)) return BadRequest();   //醍摩豆账户
                if (!jsonElement.TryGetProperty("tmdName", out JsonElement _tmdName)) return BadRequest(); //醍摩豆账号名称

                //操作记录
                string blobOrTable = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds().ToString();
                OperateLog operateLog = new OperateLog
                {
                    PartitionKey = "OperateLog-BI",
                    RowKey = blobOrTable,
                    recordID = blobOrTable,
                    platformSource = "BI",
                    tmdId = $"{_tmdId}",
                    tmdName = $"{_tmdName}",
                    visitApi = "/biabilitytask/delabilitytask",
                    operateTime = DateTime.Now,
                    operateDescribe = $"{_tmdName}【{_tmdId}】已操作删除册别，删除ID:{id}"
                };

                var cosmosClient = _azureCosmos.GetCosmosClient();
                var response = await cosmosClient.GetContainer(Constant.TEAMModelOS, "Normal").DeleteItemStreamAsync(id.ToString(), new PartitionKey($"AbilityTask-{standard}"));

                await _azureStorage.Save<OperateLog>(operateLog);//保存操作记录

                return Ok(new { state = response.Status }); 
            }
            catch (Exception ex)
            {
                await _dingDing.SendBotMsg($"BI,{_option.Location}   /biabilitytask/delabilitytask    \n  {ex.Message} {ex.StackTrace}", GroupNames.醍摩豆服務運維群組);
                return BadRequest();
            }
        }


        public record RecordAbilityTask() 
        {
            /// <summary>
            /// 醍摩豆账户ID
            /// </summary>
            public string tmdId { get; set; }

            /// <summary>
            /// 醍摩豆账户名称
            /// </summary>
            public string tmdName { get; set; }

            /// <summary>
            /// 章节信息
            /// </summary>
            public AbilityTask abilityTask { get; set; }
        }

    }
}