TEAMModelOS/TEAMModelFunction/MonitorCosmosDB.cs

using System;
using System.Collections.Generic;
using System.Net.Http;
using System.Text.Json;
using System.Threading.Tasks;
using Azure.Cosmos;
using Microsoft.Azure.Documents;
using Microsoft.Azure.WebJobs;
using Microsoft.Extensions.Logging;
using TEAMModelOS.SDK.DI;
using TEAMModelOS.SDK.Extension;

namespace TEAMModelFunction
{
    public class MonitorCosmosDB
    {
        private readonly IHttpClientFactory _clientFactory;
        private readonly AzureCosmosFactory _azureCosmos;

        public MonitorCosmosDB(IHttpClientFactory clientFactory, AzureCosmosFactory azureCosmos)
        {
            _clientFactory = clientFactory;
            _azureCosmos = azureCosmos;
        }

        [FunctionName("School")]
        public async Task School([CosmosDBTrigger(
            databaseName: "TEAMModelOS",
            collectionName: "Common",
            ConnectionStringSetting = "CosmosConnection",
            LeaseCollectionName = "leases")]IReadOnlyList<Document> input, ILogger log)
        {
            if (input != null && input.Count > 0)
            {
                log.LogInformation("Documents modified " + input.Count);
                log.LogInformation("First document Id " + input[0].Id);
            }
            //input[0]
            var client = _azureCosmos.GetCosmosClient();
            List<ExamInfo> exams = new List<ExamInfo>();
            string pk = input[0].GetPropertyValue<string>("pk");
            if (!string.IsNullOrEmpty(pk) && pk.Equals("Exam"))
            {
                string code = input[0].GetPropertyValue<string>("code");
                await foreach (var item in client.GetContainer("TEAMModelOS", "Common").GetItemQueryStreamIterator(queryText: $"select value(c) from c where c.id = '{input[0].Id}'", requestOptions: new QueryRequestOptions() { PartitionKey = new Azure.Cosmos.PartitionKey($"{code}") }))
                {
                    using var json = await JsonDocument.ParseAsync(item.ContentStream);
                    if (json.RootElement.TryGetProperty("_count", out JsonElement count) && count.GetUInt16() > 0)
                    {
                        foreach (var obj in json.RootElement.GetProperty("Documents").EnumerateArray())
                        {

                            exams.Add(obj.ToObject<ExamInfo>());
                        }
                    }
                }
                for (int i = 0; i < exams.Count; i++)
                {
                    List<ExamClassResult> examClassResults = new List<ExamClassResult>();
                    await foreach (var item in client.GetContainer("TEAMModelOS", "Common").GetItemQueryStreamIterator(queryText: $"select value(c) from c where c.examId = '{exams[i].id}'", requestOptions: new QueryRequestOptions() { PartitionKey = new Azure.Cosmos.PartitionKey($"{code}") }))
                    {
                        using var json = await JsonDocument.ParseAsync(item.ContentStream);
                        if (json.RootElement.TryGetProperty("_count", out JsonElement count) && count.GetUInt16() > 0)
                        {
                            foreach (var obj in json.RootElement.GetProperty("Documents").EnumerateArray())
                            {
                                examClassResults.Add(obj.ToObject<ExamClassResult>());
                            }
                        }
                    }
                    /*if (exams[i].startTime.CompareTo(DateTimeOffset.UtcNow.ToUnixTimeMilliseconds()) < 0 
                        && exams[i].endTime.CompareTo(DateTimeOffset.UtcNow.ToUnixTimeMilliseconds()) > 0 )
                    {*/
                    if (examClassResults.Count < 0)
                    {
                        if (exams[i].progress.Equals("going"))
                        {

                            //await client.GetContainer("TEAMModelOS", "Common").ReplaceItemAsync(exams[i], exams[i].id.ToString(), new Azure.Cosmos.PartitionKey($"{exams[i].code}"));
                            for (int j = 0; j < exams[i].subjects.Count; j++)
                            {
                                for (int k = 0; k < exams[i].targetClassIds.Count; k++)
                                {
                                    ExamClassResult result = new ExamClassResult();
                                    result.code = "ExamClassResult-" + exams[i].school;
                                    result.examId = exams[i].id;
                                    result.id = Guid.NewGuid().ToString();
                                    result.subjectId = exams[i].subjects[j].id;
                                    result.year = exams[i].year;
                                    result.ttl = -1;
                                    result.scope = exams[i].scope;
                                    result.pk = typeof(ExamClassResult).Name;
                                    result.info.id = exams[i].targetClassIds[k];
                                    var sresponse = await client.GetContainer("TEAMModelOS", "School").ReadItemStreamAsync(exams[i].targetClassIds[k], new Azure.Cosmos.PartitionKey($"Class-{exams[i].school}"));
                                    if (sresponse.Status == 200)
                                    {
                                        using var json = await JsonDocument.ParseAsync(sresponse.ContentStream);
                                        Classroom classroom = json.ToObject<Classroom>();
                                        result.info.name = classroom.name;
                                        List<List<string>> ans = new List<List<string>>();
                                        List<double> ansPoint = new List<double>();
                                        foreach (double p in exams[i].papers[j].point)
                                        {
                                            ans.Add(new List<string>());
                                            ansPoint.Add(0);
                                        }
                                        foreach (StudentSimple stu in classroom.students)
                                        {
                                            result.studentIds.Add(stu.id);
                                            result.studentAnswers.Add(ans);
                                            result.studentScores.Add(ansPoint);
                                        }
                                    }
                                    result.progress = exams[i].progress;
                                    result.school = exams[i].school;
                                    await client.GetContainer("TEAMModelOS", "Common").CreateItemAsync(result, new Azure.Cosmos.PartitionKey($"{result.code}"));

                                }
                            }
                        }
                    }
                    if (exams[i].progress.Equals("finish"))
                    {
                        for (int j = 0; j < exams[i].subjects.Count; j++)
                        {
                            ExamResult result = new ExamResult();
                            result.ttl = -1;
                            result.pk = typeof(ExamResult).Name;
                            result.code = "ExamResult-" + exams[i].school;
                            result.school = exams[i].school;
                            result.id = Guid.NewGuid().ToString();
                            result.examId = exams[i].id;
                            result.subjectId = exams[i].subjects[j].id;
                            result.year = exams[i].year;
                            result.paper = exams[i].papers[j];
                            result.point = exams[i].papers[j].point;
                            result.scope = exams[i].scope;
                            result.name = exams[i].name;
                            //result.time

                            //人数总和
                            int Count = 0;
                            int m = 0;
                            List<ClassRange> classRanges = new List<ClassRange>();
                            foreach (ExamClassResult classResult in examClassResults)
                            {
                                //处理班级信息
                                ClassRange range = new ClassRange();
                                range.id = classResult.info.id;
                                range.name = classResult.info.name;
                                List<int> ran = new List<int>();
                                int stuCount = classResult.studentIds.Count;
                                Count += stuCount;
                                if (m == 0)
                                {
                                    ran.Add(0);
                                    ran.Add(stuCount - 1);
                                }
                                else
                                {
                                    ran.Add(Count - stuCount);
                                    ran.Add(Count - 1);
                                }
                                m++;
                                range.range = ran;
                                classRanges.Add(range);
                                //处理学生ID
                                foreach (string id in classResult.studentIds)
                                {
                                    result.studentIds.Add(id);
                                }
                                foreach (List<double> scores in classResult.studentScores)
                                {
                                    result.studentScores.Add(scores);
                                }
                            }
                            result.classes = classRanges;
                            await _azureCosmos.GetCosmosClient().GetContainer("TEAMModelOS", "Common").CreateItemAsync(result, new Azure.Cosmos.PartitionKey($"ExamResult-{result.school}"));
                        }
                    }
                }
            }
        }
    }
}