TEAMModelOS/TEAMModelFunction/TriggerSurvey.cs

using Azure.Cosmos;
using Azure.Messaging.ServiceBus;
using Azure.Storage.Blobs.Models;
using Azure.Storage.Sas;
using Microsoft.Azure.Documents;
using System;
using System.Collections.Generic;
using System.Net.Http;
using System.Text;
using System.Text.Json;
using System.Threading.Tasks;
using TEAMModelOS.SDK.DI;
using TEAMModelOS.SDK.Extension;
using TEAMModelOS.SDK.Helper.Common.CollectionHelper;
using TEAMModelOS.SDK.Models;
using TEAMModelOS.SDK.Models.Cosmos;
using TEAMModelOS.SDK.Models.Cosmos.Common.Inner;
using TEAMModelOS.SDK.Module.AzureBlob.Configuration;

namespace TEAMModelFunction
{
   public class TriggerSurvey
    {
        public static async void Trigger( AzureServiceBusFactory _serviceBus, AzureStorageFactory _azureStorage, DingDing _dingDing,
               CosmosClient client, Document input, TriggerData tdata,AzureRedisFactory _azureRedis)
        {
            if ((tdata.status != null && tdata.status.Value == 404) || tdata.ttl > 0)
            {
                return;
            }
            var adid = tdata.id;
            var adcode = "";
            string blobcntr = null;
            if (tdata.scope == "school")
            {
                  adcode = $"Activity-{tdata.school}";
                blobcntr = tdata.school;
            }
            else if (tdata.scope == "private"){
                adcode = $"Activity-{tdata.creatorId}";
                 blobcntr = tdata.creatorId;
            }
            ActivityData data = null;
            try
            {
                if (tdata.scope == "school")
                {
                    data = await client.GetContainer("TEAMModelOS", "School").ReadItemAsync<ActivityData>(adid, new Azure.Cosmos.PartitionKey($"{adcode}"));
                }
                else if (tdata.scope == "private")
                {
                    data = await client.GetContainer("TEAMModelOS", "Teacher").ReadItemAsync<ActivityData>(adid, new Azure.Cosmos.PartitionKey($"{adcode}"));
                }
            }
            catch { data = null; }
            Survey survey = await client.GetContainer("TEAMModelOS", "Common").ReadItemAsync<Survey>(input.Id, new Azure.Cosmos.PartitionKey($"{tdata.code}"));
            List<ChangeRecord> changeRecords = await _azureStorage.FindListByDict<ChangeRecord>(new Dictionary<string, object>() { { "RowKey", input.Id }, { "PartitionKey", survey.progress } });
            if (survey != null) {
                switch (survey.progress)
                {
                    case "pending":
                        var messageSurvey = new ServiceBusMessage(new { id = input.Id, progress = "going", code = tdata.code }.ToJsonString());
                        messageSurvey.ApplicationProperties.Add("name", "Survey");
                        if (changeRecords.Count > 0)
                        {
                            await _serviceBus.GetServiceBusClient().cancelMessage("active-task", changeRecords[0].sequenceNumber);
                            long start = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", messageSurvey, DateTimeOffset.FromUnixTimeMilliseconds(tdata.stime));
                            changeRecords[0].sequenceNumber = start;
                            await _azureStorage.SaveOrUpdate<ChangeRecord>(changeRecords[0]);
                        }
                        else
                        {
                            long start = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", messageSurvey, DateTimeOffset.FromUnixTimeMilliseconds(tdata.stime));
                            ChangeRecord changeRecord = new ChangeRecord
                            {
                                RowKey = input.Id,
                                PartitionKey = "pending",
                                sequenceNumber = start,
                                msgId = messageSurvey.MessageId
                            };
                            await _azureStorage.Save<ChangeRecord>(changeRecord);
                        }
                        break;
                    case "going":
                        if (survey.scope == "school")
                        {
                            data = new ActivityData
                            {
                                id = survey.id,
                                code = $"Activity-{survey.school}",
                                type = "survey",
                                name = survey.name,
                                startTime = survey.startTime,
                                endTime = survey.endTime,
                                scode = survey.code,
                                scope = survey.scope,
                                classes = survey.classes.IsNotEmpty() ? survey.classes : new List<string> { "" },
                                tmdids = survey.tmdids.IsNotEmpty() ? survey.tmdids : new List<string> { "" },
                                progress = "going",
                                subjects = new List<string> { "" }
                            };
                            await client.GetContainer("TEAMModelOS", "School").UpsertItemAsync<ActivityData>(data, new Azure.Cosmos.PartitionKey(data.code));
                        }
                        else if (survey.scope == "private")
                        {
                            data = new ActivityData
                            {
                                id = survey.id,
                                code = $"Activity-Common",
                                type = "survey",
                                name = survey.name,
                                startTime = survey.startTime,
                                endTime = survey.endTime,
                                scode = survey.code,
                                scope = survey.scope,
                                progress = "going",
                                classes = survey.classes.IsNotEmpty() ? survey.classes : new List<string> { "" },
                                tmdids = new List<string> { "" },
                                subjects = new List<string> { "" }
                            };
                            await client.GetContainer("TEAMModelOS", "Teacher").UpsertItemAsync<ActivityData>(data, new Azure.Cosmos.PartitionKey(data.code));
                        }
                        var messageSurveyEnd = new ServiceBusMessage(new { id = input.Id, progress = "finish", code = tdata.code }.ToJsonString());
                        messageSurveyEnd.ApplicationProperties.Add("name", "Survey");
                        if (changeRecords.Count > 0)
                        {
                            long end = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", messageSurveyEnd, DateTimeOffset.FromUnixTimeMilliseconds(tdata.etime));
                            await _serviceBus.GetServiceBusClient().cancelMessage("active-task", changeRecords[0].sequenceNumber);
                            changeRecords[0].sequenceNumber = end;
                            await _azureStorage.SaveOrUpdate<ChangeRecord>(changeRecords[0]);
                        }
                        else
                        {
                            long end = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", messageSurveyEnd, DateTimeOffset.FromUnixTimeMilliseconds(tdata.etime));
                            ChangeRecord changeRecord = new ChangeRecord
                            {
                                RowKey = input.Id,
                                PartitionKey = "going",
                                sequenceNumber = end,
                                msgId = messageSurveyEnd.MessageId
                            };
                            await _azureStorage.Save<ChangeRecord>(changeRecord);
                        }
                        await _dingDing.SendBotMsg($"问卷调查{tdata.id}将于:{tdata.etime}完成并结算！", GroupNames.成都开发測試群組);
                        break;
                    case "finish":
                        await _dingDing.SendBotMsg($"问卷调查{tdata.id}开始结算{tdata.etime}！", GroupNames.成都开发測試群組);
                        var records = await _azureRedis.GetRedisClient(8).HashGetAllAsync($"Survey:Record:{survey.id}");
                        var submits = await _azureRedis.GetRedisClient(8).SetMembersAsync($"Survey:Submit:{survey.id}");
                        List<dynamic> recs = new List<dynamic>();
                        foreach (var rcd in records)
                        {
                            var value = rcd.Value.ToString().ToObject<JsonElement>();
                            recs.Add(new { index = rcd.Name.ToString(), ans = value });
                        }

                        List<dynamic> userids = new List<dynamic>();
                        foreach (var submit in submits)
                        {
                            var value = submit.ToString();
                            userids.Add(value);
                        }
                        var cods = new { records = recs, userids };
                        //问卷整体情况
                        await _azureStorage.UploadFileByContainer(blobcntr, cods.ToJsonString(), "survey", $"{survey.id}/record.json");
                        //结算每道题的答题情况
                        var ContainerClient =  _azureStorage.GetBlobContainerClient(blobcntr);
                        List<Task<string>> tasks = new List<Task<string>>();
                        //获取
                        try {
                            List<string> items = await ContainerClient.List($"survey/{survey.id}/urecord");
                            List<SurveyRecord> surveyRecords = new List<SurveyRecord>();
                            foreach (string item in items)
                            {
                                var Download = await _azureStorage.GetBlobContainerClient(blobcntr).GetBlobClient(item).DownloadAsync();
                                var json = await JsonDocument.ParseAsync(Download.Value.Content);
                                var Record = json.RootElement.ToObject<SurveyRecord>();
                                surveyRecords.Add(Record);
                            }
                            await _dingDing.SendBotMsg($"问卷调查问题结算数据{surveyRecords.ToJsonString()}", GroupNames.成都开发測試群組);
                            for (int index = 0; index < survey.answers.Count; index++)
                            {
                                string url = $"{survey.id}/qrecord/{index}.json";
                                QuestionRecord question = new QuestionRecord() { index = index };
                                foreach (SurveyRecord record in surveyRecords)
                                {
                                    if (record.ans.Count == survey.answers.Count)
                                    {
                                        foreach (var an in record.ans[index])
                                        {
                                            //
                                            if (question.opt.ContainsKey(an))
                                            {
                                                if (question.opt[an] != null)
                                                {
                                                    question.opt[an].Add(record.userid);
                                                }
                                                else
                                                {
                                                    question.opt[an] = new HashSet<string>() { record.userid };
                                                }
                                            }
                                            else
                                            {
                                                if (survey.answers[index].Contains(an))
                                                {
                                                    //如果是客观题code
                                                    question.opt.Add(an, new HashSet<string> { record.userid });
                                                }
                                                else
                                                {
                                                    //如果不是客观code
                                                    question.other[record.userid] = an;
                                                }
                                            }
                                        }
                                    }
                                }
                                tasks.Add(  _azureStorage.UploadFileByContainer(blobcntr, question.ToJsonString(), "survey", url));
                            }
                            await Task.WhenAll(tasks);
                        } catch (Exception ex) {
                            await _dingDing.SendBotMsg($"问卷调查问题结算异常{ex.Message}\n{ex.StackTrace}", GroupNames.成都开发測試群組);
                        }
                        if (string.IsNullOrEmpty(survey.recordUrl))
                        {
                            survey.recordUrl = $"/survey/{survey.id}/record.json";
                            await client.GetContainer("TEAMModelOS", "Common").ReplaceItemAsync<Survey>(survey, survey.id, new Azure.Cosmos.PartitionKey(survey.code));
                        }
                        else {
                            _azureRedis.GetRedisClient(8).KeyDelete($"Survey:Record:{survey.id}");
                            _azureRedis.GetRedisClient(8).KeyDelete($"Survey:Submit:{survey.id}");
                            break;
                        }
                        
                        //更新结束状态
                        data.progress = "finish";
                        if (survey.scope == "school")
                        {
                            await client.GetContainer("TEAMModelOS", "School").ReplaceItemAsync<ActivityData>(data, data.id, new Azure.Cosmos.PartitionKey(data.code));
                        }
                        else if (survey.scope == "private")
                        {
                            await client.GetContainer("TEAMModelOS", "Teacher").ReplaceItemAsync<ActivityData>(data, data.id, new Azure.Cosmos.PartitionKey(data.code));
                        }
                        
                        break;
                }
            }
        }
    }
    /**
     * {survey.id}/qrecord/{index}.json
        {
            "opt": {
                "A": [
                    "userid1",
                    "userid2",
                    "userid3"
                ],
                "B": [
                    "userid1",
                    "userid2",
                    "userid3"
                ]
            },
            "other": {
                "userid1": "建议XXXX1",
                "userid2": "建议XXXX2"
            }
        }
     **/
    public class QuestionRecord { 
        public int index { get; set; }
        public Dictionary<string, HashSet<string>> opt { get; set; } = new Dictionary<string, HashSet<string>>();
        public Dictionary<string, string> other { get; set; } = new Dictionary<string, string>();
    }
}