TEAMModelOS/TEAMModelFunction/TriggerSurvey.cs

using Azure.Cosmos;
using Azure.Messaging.ServiceBus;
using Microsoft.Azure.Documents;
using System;
using System.Collections.Generic;
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;

namespace TEAMModelFunction
{
   public class TriggerSurvey
    {
        public static async void Trigger(AzureCosmosFactory _azureCosmos, AzureServiceBusFactory _serviceBus, AzureStorageFactory _azureStorage, DingDing _dingDing,
               CosmosClient client, Document input, string code, long stime, long etime, string school,AzureRedisFactory _azureRedis)
        {
            Survey survey = await client.GetContainer("TEAMModelOS", "Common").ReadItemAsync<Survey>(input.Id, new Azure.Cosmos.PartitionKey($"{code}"));
            List<ChangeRecord> changeRecords = await _azureStorage.FindListByDict<ChangeRecord>(new Dictionary<string, object>() { { "RowKey", input.Id }, { "PartitionKey", survey.progress } });
            if (survey.ttl >= 1)
            {
                //TODO  处理TTL删除业务
                return;
            }
            switch (survey.progress)
            {
                case "pending":
                    var messageSurvey = new ServiceBusMessage(new { id = input.Id, progress = "going", code = 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(stime));
                        changeRecords[0].sequenceNumber = start;
                        await _azureStorage.SaveOrUpdate<ChangeRecord>(changeRecords[0]);
                    }
                    else
                    {
                        long start = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", messageSurvey, DateTimeOffset.FromUnixTimeMilliseconds(stime));
                        ChangeRecord changeRecord = new ChangeRecord
                        {
                            RowKey = input.Id,
                            PartitionKey = "pending",
                            sequenceNumber = start,
                            msgId = messageSurvey.MessageId
                        };
                        await _azureStorage.Save<ChangeRecord>(changeRecord);
                    }
                    break;
                case "going":
                    ActivityData data;
                    if (survey.scope == "school")
                    {
                        data = new ActivityData
                        {
                            id = survey.id,
                            code = $"Activity-{survey.owner}",
                            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",
                            owner = survey.owner,
                            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> { "" },
                            owner = survey.owner,
                            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 = code }.ToJsonString());
                    messageSurveyEnd.ApplicationProperties.Add("name", "Survey");
                    if (changeRecords.Count > 0)
                    {
                        long end = await _serviceBus.GetServiceBusClient().SendScheduleMessageAsync("active-task", messageSurveyEnd, DateTimeOffset.FromUnixTimeMilliseconds(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(etime));
                        ChangeRecord changeRecord                                                                                                                                                                                                                                                   = new ChangeRecord
                        {
                            RowKey = input.Id,
                            PartitionKey = "going",
                            sequenceNumber = end,
                            msgId = messageSurveyEnd.MessageId
                        };
                        await _azureStorage.Save<ChangeRecord>(changeRecord);
                    }
                    break;
                case "finish":
                    var records =await  _azureRedis.GetRedisClient(8).HashGetAllAsync($"Survey:Record:{survey.id}_{survey.code}");
                    List<dynamic> recs = new List<dynamic>();
                    foreach (var rcd in records) {
                        //var key =int.Parse(rcd.Name.ToString());
                        var value = rcd.Value.ToString().ToObject<JsonElement>();
                        recs.Add(new { index = rcd.Name.ToString(), count = value });
                    }
                    await  _azureStorage.UploadFileByContainer(survey.owner, recs.ToJsonString(), "vote", $"{survey.id}/record.json");
                    break;
            }
        }
    }
}