119 lines
4.6 KiB
C#
119 lines
4.6 KiB
C#
using Managing.Application.Abstractions;
|
|
using Managing.Core;
|
|
using MediatR;
|
|
using Microsoft.Extensions.Logging;
|
|
using static Managing.Common.Enums;
|
|
|
|
namespace Managing.Application.ManageBot;
|
|
|
|
public class LoadBackupBotCommandHandler : IRequestHandler<LoadBackupBotCommand, string>
|
|
{
|
|
private readonly IBotService _botService;
|
|
private readonly ILogger<LoadBackupBotCommandHandler> _logger;
|
|
|
|
public LoadBackupBotCommandHandler(
|
|
ILogger<LoadBackupBotCommandHandler> logger, IBotService botService)
|
|
{
|
|
_logger = logger;
|
|
_botService = botService;
|
|
}
|
|
|
|
public Task<string> Handle(LoadBackupBotCommand request, CancellationToken cancellationToken)
|
|
{
|
|
var backupBots = _botService.GetSavedBots().ToList();
|
|
_logger.LogInformation("Loading {Count} backup bots.", backupBots.Count);
|
|
|
|
var result = new Dictionary<string, BotStatus>();
|
|
bool anyBackupStarted = false;
|
|
bool anyBotActive = false;
|
|
|
|
foreach (var backupBot in backupBots)
|
|
{
|
|
try
|
|
{
|
|
if (backupBot.LastStatus == BotStatus.Down)
|
|
{
|
|
_logger.LogInformation("Skipping backup bot {Identifier} as it is marked as Down.",
|
|
backupBot.Identifier);
|
|
continue;
|
|
}
|
|
|
|
var activeBot = _botService.GetActiveBots().FirstOrDefault(b => b.Identifier == backupBot.Identifier);
|
|
|
|
if (activeBot == null)
|
|
{
|
|
_logger.LogInformation("No active instance found for bot {Identifier}. Starting backup...",
|
|
backupBot.Identifier);
|
|
|
|
// Start the bot from backup
|
|
_botService.StartBotFromBackup(backupBot);
|
|
|
|
// Wait a short time to allow the bot to initialize
|
|
Thread.Sleep(1000);
|
|
|
|
// Try to get the active bot multiple times to ensure it's properly started
|
|
int attempts = 0;
|
|
const int maxAttempts = 5;
|
|
|
|
while (attempts < maxAttempts)
|
|
{
|
|
activeBot = _botService.GetActiveBots().FirstOrDefault(b => b.Identifier == backupBot.Identifier);
|
|
if (activeBot != null)
|
|
{
|
|
result[activeBot.Identifier] = BotStatus.Up;
|
|
anyBackupStarted = true;
|
|
_logger.LogInformation("Backup bot {Identifier} started successfully.", backupBot.Identifier);
|
|
break;
|
|
}
|
|
|
|
attempts++;
|
|
if (attempts < maxAttempts)
|
|
{
|
|
Thread.Sleep(1000); // Wait another second before next attempt
|
|
}
|
|
}
|
|
|
|
if (activeBot == null)
|
|
{
|
|
result[backupBot.Identifier] = BotStatus.Down;
|
|
_logger.LogWarning("Backup bot {Identifier} failed to start after {MaxAttempts} attempts.",
|
|
backupBot.Identifier, maxAttempts);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
var status = MiscExtensions.ParseEnum<BotStatus>(activeBot.GetStatus());
|
|
result[activeBot.Identifier] = status;
|
|
anyBotActive = true;
|
|
_logger.LogInformation("Bot {Identifier} is already active with status {Status}.",
|
|
activeBot.Identifier,
|
|
status);
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error loading bot {Identifier}. Deleting its backup.", backupBot.Identifier);
|
|
_botService.DeleteBotBackup(backupBot.Identifier);
|
|
result[backupBot.Identifier] = BotStatus.Down;
|
|
}
|
|
}
|
|
|
|
var summary = string.Join(", ", result.Select(b => $"{b.Key}: {b.Value}"));
|
|
_logger.LogInformation("Bot loading completed. Summary: {Summary}", summary);
|
|
|
|
// Determine final status
|
|
BotStatus finalStatus = anyBackupStarted
|
|
? BotStatus.Backup
|
|
: anyBotActive
|
|
? BotStatus.Up
|
|
: BotStatus.Down;
|
|
|
|
_logger.LogInformation("Final aggregate bot status: {FinalStatus}", finalStatus);
|
|
|
|
return Task.FromResult(finalStatus.ToString());
|
|
}
|
|
}
|
|
|
|
public class LoadBackupBotCommand : IRequest<string>
|
|
{
|
|
} |