using System; using System.Collections.Generic; using System.Text.Json; using System.Threading.Tasks; using Azure.Cosmos; using Microsoft.Azure.WebJobs; using Microsoft.Azure.WebJobs.Host; using Microsoft.Extensions.Logging; using StackExchange.Redis; using TEAMModelOS.SDK.DI; using TEAMModelOS.SDK.Extension; using TEAMModelOS.SDK.Models; namespace TEAMModelFunction { public class MonitorServicesBus { private readonly AzureCosmosFactory _azureCosmos; private readonly DingDing _dingDing; private readonly AzureStorageFactory _azureStorage; private readonly AzureRedisFactory _azureRedis; public MonitorServicesBus(AzureCosmosFactory azureCosmos, DingDing dingDing, AzureStorageFactory azureStorage , AzureRedisFactory azureRedis ) { _azureCosmos = azureCosmos; _dingDing = dingDing; _azureStorage = azureStorage; _azureRedis = azureRedis; } [FunctionName("Exam")] public async Task Exam([ServiceBusTrigger("active-task", "exam", Connection = "Azure:ServiceBus:ConnectionString")] string msg) { try { var json = JsonDocument.Parse(msg); json.RootElement.TryGetProperty("id", out JsonElement id); json.RootElement.TryGetProperty("progress", out JsonElement progress); json.RootElement.TryGetProperty("code", out JsonElement code); //Dictionary keyValuePairs = mySbMsg.ToObject>(); var client = _azureCosmos.GetCosmosClient(); ExamInfo exam = await client.GetContainer("TEAMModelOS", "Common").ReadItemAsync(id.ToString(), new PartitionKey($"{code}")); exam.progress = progress.ToString(); await client.GetContainer("TEAMModelOS", "Common").ReplaceItemAsync(exam, id.ToString(), new PartitionKey($"{code}")); /* keyValuePairs.TryGetValue("id", out object id); keyValuePairs.TryGetValue("name", out object name); keyValuePairs.TryGetValue("code", out object code);*/ //keyValuePairs.TryGetValue("status", out object progress); /*if (name.ToString().Equals("Exam", StringComparison.OrdinalIgnoreCase)) { ExamInfo exam = await client.GetContainer("TEAMModelOS", "Common").ReadItemAsync(id.ToString(), new PartitionKey($"{code}")); if (DateTimeOffset.UtcNow.ToUnixTimeMilliseconds().CompareTo(exam.startTime) > 0 && DateTimeOffset.UtcNow.ToUnixTimeMilliseconds().CompareTo(exam.endTime) < 0) { exam.progress = "going"; } else if (DateTimeOffset.UtcNow.ToUnixTimeMilliseconds().CompareTo(exam.endTime) > 0) { exam.progress = "finish"; } await client.GetContainer("TEAMModelOS", "Common").ReplaceItemAsync(exam, id.ToString(), new PartitionKey($"{code}")); }*/ } catch (Exception ex) { await _dingDing.SendBotMsg($"ServiceBus,ExamBus()\n{ex.Message}", GroupNames.醍摩豆服務運維群組); } } [FunctionName("Vote")] public async Task Vote([ServiceBusTrigger("active-task", "vote", Connection = "Azure:ServiceBus:ConnectionString")] string msg) { try { var jsonMsg = JsonDocument.Parse(msg); jsonMsg.RootElement.TryGetProperty("id", out JsonElement id); jsonMsg.RootElement.TryGetProperty("progress", out JsonElement progress); jsonMsg.RootElement.TryGetProperty("code", out JsonElement code); //Dictionary keyValuePairs = mySbMsg.ToObject>(); var client = _azureCosmos.GetCosmosClient(); Vote vote = await client.GetContainer("TEAMModelOS", "Common").ReadItemAsync(id.ToString(), new PartitionKey($"{code}")); vote.progress = progress.ToString(); await client.GetContainer("TEAMModelOS", "Common").ReplaceItemAsync(vote, id.ToString(), new PartitionKey($"{code}")); /*keyValuePairs.TryGetValue("id", out object id); keyValuePairs.TryGetValue("name", out object name); keyValuePairs.TryGetValue("code", out object code);*/ //keyValuePairs.TryGetValue("status", out object progress); /*if (name.ToString().Equals("Vote", StringComparison.OrdinalIgnoreCase)) { Vote vote; var sresponse = await client.GetContainer("TEAMModelOS", "Common").ReadItemStreamAsync(id.ToString(), new PartitionKey($"{code}")); if (sresponse.Status == 200) { using var json = await JsonDocument.ParseAsync(sresponse.ContentStream); vote = json.ToObject(); if (DateTimeOffset.UtcNow.ToUnixTimeMilliseconds().CompareTo(vote.startTime) > 0 && DateTimeOffset.UtcNow.ToUnixTimeMilliseconds().CompareTo(vote.endTime) < 0) { vote.progress = "going"; } else if (DateTimeOffset.UtcNow.ToUnixTimeMilliseconds().CompareTo(vote.endTime) > 0) { vote.progress = "finish"; } await client.GetContainer("TEAMModelOS", "Common").ReplaceItemAsync(vote, id.ToString(), new PartitionKey($"{code}")); } }*/ } catch (Exception ex) { await _dingDing.SendBotMsg($"ServiceBus,VoteBus()\n{ex.Message}", GroupNames.醍摩豆服務運維群組); } } [FunctionName("Survey")] public async Task Survey([ServiceBusTrigger("active-task", "survey", Connection = "Azure:ServiceBus:ConnectionString")] string msg) { try { var jsonMsg = JsonDocument.Parse(msg); jsonMsg.RootElement.TryGetProperty("id", out JsonElement id); jsonMsg.RootElement.TryGetProperty("progress", out JsonElement progress); jsonMsg.RootElement.TryGetProperty("code", out JsonElement code); //Dictionary keyValuePairs = mySbMsg.ToObject>(); var client = _azureCosmos.GetCosmosClient(); Survey survey = await client.GetContainer("TEAMModelOS", "Common").ReadItemAsync(id.ToString(), new PartitionKey($"{code}")); survey.progress = progress.ToString(); await client.GetContainer("TEAMModelOS", "Common").ReplaceItemAsync(survey, id.ToString(), new PartitionKey($"{code}")); /*keyValuePairs.TryGetValue("id", out object id); keyValuePairs.TryGetValue("name", out object name); keyValuePairs.TryGetValue("code", out object code);*/ //keyValuePairs.TryGetValue("status", out object progress); /*if (name.ToString().Equals("Survey", StringComparison.OrdinalIgnoreCase)) { Survey survey = await client.GetContainer("TEAMModelOS", "Common").ReadItemAsync(id.ToString(), new PartitionKey($"{code}")); if (DateTimeOffset.UtcNow.ToUnixTimeMilliseconds().CompareTo(survey.startTime) > 0 && DateTimeOffset.UtcNow.ToUnixTimeMilliseconds().CompareTo(survey.endTime) < 0) { survey.progress = "going"; } else if (DateTimeOffset.UtcNow.ToUnixTimeMilliseconds().CompareTo(survey.endTime) > 0) { survey.progress = "finish"; } await client.GetContainer("TEAMModelOS", "Common").ReplaceItemAsync(survey, id.ToString(), new PartitionKey($"{code}")); }*/ } catch (Exception ex) { await _dingDing.SendBotMsg($"ServiceBus,VoteBus()\n{ex.Message}", GroupNames.醍摩豆服務運維群組); } } [FunctionName("Blob")] public async Task Blob([ServiceBusTrigger("active-task", "blob", Connection = "Azure:ServiceBus:ConnectionString")] string msg) { try { // await _dingDing.SendBotMsg($"ServiceBus,Blob(){msg}", GroupNames.醍摩豆服務運維群組); var jsonMsg = JsonDocument.Parse(msg); if(jsonMsg.RootElement.TryGetProperty("name", out JsonElement name)&& name.ValueKind==JsonValueKind.String) { var client = _azureStorage.GetBlobContainerClient($"{name}"); var size = await client.GetBlobsCatalogSize(); await _azureRedis.GetRedisClient(8).HashSetAsync($"Blob:Record", new RedisValue($"{name}"), new RedisValue($"{long.Parse($"{size.Item1}")}")); foreach (var key in size.Item2.Keys) { await _azureRedis.GetRedisClient(8).SortedSetRemoveAsync($"Blob:Catalog:{name}", key); await _azureRedis.GetRedisClient(8).SortedSetIncrementAsync($"Blob:Catalog:{name}", key, size.Item2[key].HasValue ? size.Item2[key].Value : 0); } await _dingDing.SendBotMsg($"ServiceBus,Blob() 容器:{name}使用:{size.Item1},文件分类:{size.Item2.ToJsonString()}", GroupNames.醍摩豆服務運維群組); } } catch (Exception ex) { await _dingDing.SendBotMsg($"ServiceBus,Blob()\n{ex.Message}{ex.StackTrace}", GroupNames.醍摩豆服務運維群組); } } } }