Заготовки под подтверждение статуса проблемы и рассылку нотификации при её появлении

This commit is contained in:
ftc
2024-01-20 19:22:00 +03:00
parent 15bf517434
commit 070bc3fbf0
3 changed files with 43 additions and 13 deletions
+1
View File
@@ -16,5 +16,6 @@ namespace ZaoBot.Problems
public string label; public string label;
public string name; public string name;
public Status status; public Status status;
public bool notified = false;
} }
} }
+30 -2
View File
@@ -1,7 +1,10 @@
using System; using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.Text; using System.Text;
using System.Text.Json;
using System.Timers; using System.Timers;
using Telegram.Bot;
using ZaoBot.Config;
namespace ZaoBot.Problems namespace ZaoBot.Problems
{ {
@@ -10,15 +13,19 @@ namespace ZaoBot.Problems
const int DEFAULT_INTERVAL = 2000; const int DEFAULT_INTERVAL = 2000;
private OpenHab.OpenHabClient openhab; private OpenHab.OpenHabClient openhab;
private ITelegramBotClient bot;
private JsonDocument config;
private Timer timer; private Timer timer;
private Dictionary<string, Item> statusItems = new Dictionary<string, Item>(); private Dictionary<string, Item> statusItems = new Dictionary<string, Item>();
public IEnumerable<Item> StatusItems => statusItems.Values; public IEnumerable<Item> StatusItems => statusItems.Values;
public StatusManager(OpenHab.OpenHabClient openhab) public StatusManager(OpenHab.OpenHabClient openhab, ITelegramBotClient bot, IConfig config)
{ {
this.bot = bot;
this.openhab = openhab; this.openhab = openhab;
this.config = config.Data;
UpdateStatus(); UpdateStatus();
timer = new Timer(DEFAULT_INTERVAL); timer = new Timer(DEFAULT_INTERVAL);
timer.Elapsed += OnTimer; timer.Elapsed += OnTimer;
@@ -39,6 +46,8 @@ namespace ZaoBot.Problems
{ {
if (!item.name.StartsWith("Problem_")) continue; if (!item.name.StartsWith("Problem_")) continue;
Item statusItem = null; Item statusItem = null;
lock (statusItems)
{
if (statusItems.ContainsKey(item.name)) statusItem = statusItems[item.name]; if (statusItems.ContainsKey(item.name)) statusItem = statusItems[item.name];
else else
{ {
@@ -50,17 +59,36 @@ namespace ZaoBot.Problems
}; };
statusItems[item.name] = statusItem; statusItems[item.name] = statusItem;
} }
}
var state = await openhab.GetItemState(item.name); var state = await openhab.GetItemState(item.name);
lock (statusItems)
{
if (state.state == "OFF") statusItem.status = Status.OK; if (state.state == "OFF") statusItem.status = Status.OK;
else if (state.state == "ON") statusItem.status = Status.PROBLEM; else if (state.state == "ON") statusItem.status = Status.PROBLEM;
else statusItem.status = Status.UNKNOWN; else statusItem.status = Status.UNKNOWN;
if (statusItem.status == Status.PROBLEM && statusItem.notified == false)
{
Notify(statusItem);
statusItem.notified = true;
}
}
} }
} }
catch (Exception ex) catch (Exception ex)
{ {
Console.WriteLine("Exception on getting status: " + ex.ToString()); Console.WriteLine("Exception on getting status: " + ex.ToString());
} }
} }
private async void Notify(Item item)
{
foreach (var element in config.RootElement.GetProperty("notify_user_ids").EnumerateArray())
{
await bot.SendTextMessageAsync(
chatId: element.GetString(),
text: "Problem appeared: " + item.label
);
}
}
} }
} }
+1
View File
@@ -41,6 +41,7 @@ namespace ZaoBot
var receiver = provider.GetRequiredService<NotificationReceiver>(); var receiver = provider.GetRequiredService<NotificationReceiver>();
var notificationThread = new Thread(new ThreadStart(receiver.Run)); var notificationThread = new Thread(new ThreadStart(receiver.Run));
notificationThread.Start(); notificationThread.Start();
var statusManager = provider.GetRequiredService<StatusManager>();
while (true) while (true)
{ {
try try