Commit before refact factory into service
This commit is contained in:
@@ -11,30 +11,21 @@ namespace Managing.Application.ManageBot;
|
||||
|
||||
public class LoadBackupBotCommandHandler : IRequestHandler<LoadBackupBotCommand, string>
|
||||
{
|
||||
private readonly IBotFactory _botFactory;
|
||||
private readonly ITaskCache _taskCache;
|
||||
private readonly IMoneyManagementService _moneyManagementService;
|
||||
private readonly IBotRepository _botRepository;
|
||||
private readonly IBotService _botService;
|
||||
private readonly ILogger<LoadBackupBotCommandHandler> _logger;
|
||||
|
||||
public LoadBackupBotCommandHandler(
|
||||
IBotFactory botFactory,
|
||||
ITaskCache taskCache,
|
||||
IMoneyManagementService moneyManagementService,
|
||||
IBotRepository botRepository,
|
||||
ILogger<LoadBackupBotCommandHandler> logger)
|
||||
ILogger<LoadBackupBotCommandHandler> logger, IBotService botService)
|
||||
{
|
||||
_botFactory = botFactory;
|
||||
_taskCache = taskCache;
|
||||
_moneyManagementService = moneyManagementService;
|
||||
_botRepository = botRepository;
|
||||
_logger = logger;
|
||||
_botService = botService;
|
||||
}
|
||||
|
||||
public Task<string> Handle(LoadBackupBotCommand request, CancellationToken cancellationToken)
|
||||
{
|
||||
BotStatus botStatus = BotStatus.Down;
|
||||
var backupBots = _botRepository.GetBots();
|
||||
var backupBots = _botService.GetSavedBots();
|
||||
var activeBots = _botService.GetActiveBots();
|
||||
var result = new Dictionary<string, BotStatus>();
|
||||
|
||||
_logger.LogInformation($"Loading {backupBots.Count()} backup bots");
|
||||
@@ -47,12 +38,11 @@ public class LoadBackupBotCommandHandler : IRequestHandler<LoadBackupBotCommand,
|
||||
switch (backupBot.BotType)
|
||||
{
|
||||
case BotType.SimpleBot:
|
||||
var simpleBot = _taskCache.Get<IBot>(backupBot.Name);
|
||||
var simpleBot = activeBots.FirstOrDefault(b => b.GetName() == backupBot.Name);
|
||||
if (simpleBot == null)
|
||||
{
|
||||
_logger.LogInformation($"Starting backup bot {backupBot.Name}");
|
||||
StartBot(backupBot);
|
||||
simpleBot.LoadBackup(backupBot);
|
||||
_botService.StartBot(backupBot);
|
||||
result.Add(simpleBot.GetName(), BotStatus.Backup);
|
||||
}
|
||||
else
|
||||
@@ -63,19 +53,18 @@ public class LoadBackupBotCommandHandler : IRequestHandler<LoadBackupBotCommand,
|
||||
break;
|
||||
case BotType.ScalpingBot:
|
||||
case BotType.FlippingBot:
|
||||
var scalpingBot = _taskCache.Get<ITradingBot>(backupBot.Name);
|
||||
var scalpingBot = activeBots.FirstOrDefault(b => b.GetName() == backupBot.Name);
|
||||
if (scalpingBot == null)
|
||||
{
|
||||
_logger.LogInformation($"Starting backup bot {backupBot.Name}");
|
||||
StartBot(backupBot);
|
||||
scalpingBot = _taskCache.Get<ITradingBot>(backupBot.Name);
|
||||
scalpingBot.LoadBackup(backupBot);
|
||||
result.Add(scalpingBot.GetName(), BotStatus.Backup);
|
||||
_botService.StartBot(backupBot);
|
||||
}
|
||||
else
|
||||
{
|
||||
result.Add(scalpingBot.GetName(), MiscExtensions.ParseEnum<BotStatus>(scalpingBot.GetStatus()));
|
||||
result.Add(scalpingBot.GetName(),
|
||||
MiscExtensions.ParseEnum<BotStatus>(scalpingBot.GetStatus()));
|
||||
}
|
||||
|
||||
break;
|
||||
default:
|
||||
result.Add(backupBot.Name, BotStatus.Down);
|
||||
@@ -91,47 +80,6 @@ public class LoadBackupBotCommandHandler : IRequestHandler<LoadBackupBotCommand,
|
||||
|
||||
return Task.FromResult(botStatus.ToString());
|
||||
}
|
||||
|
||||
private void StartBot(BotBackup backupBot)
|
||||
{
|
||||
switch (backupBot.BotType)
|
||||
{
|
||||
case BotType.SimpleBot:
|
||||
Func<Task<IBot>> simpleBot = () => Task.FromResult(_botFactory.CreateSimpleBot(backupBot.Name, null));
|
||||
var bot1 = _taskCache.AddOrGetExisting(backupBot.Name, simpleBot).Result;
|
||||
bot1.LoadBackup(backupBot);
|
||||
bot1.Start();
|
||||
break;
|
||||
case BotType.ScalpingBot:
|
||||
var data = JsonConvert.DeserializeObject<TradingBotBackup>(backupBot.Data);
|
||||
Func<Task<ITradingBot>> scalpingBot = () => Task.FromResult(_botFactory.CreateScalpingBot(
|
||||
data.AccountName,
|
||||
data.MoneyManagement,
|
||||
data.Name,
|
||||
data.Ticker,
|
||||
data.Scenario,
|
||||
data.Timeframe,
|
||||
data.IsForWatchingOnly));
|
||||
var bot2 = _taskCache.AddOrGetExisting(backupBot.Name, scalpingBot).Result;
|
||||
bot2.LoadBackup(backupBot);
|
||||
bot2.Start();
|
||||
break;
|
||||
case BotType.FlippingBot:
|
||||
var dataFlippingBot = JsonConvert.DeserializeObject<TradingBotBackup>(backupBot.Data);
|
||||
Func<Task<ITradingBot>> flippingBot = () => Task.FromResult(_botFactory.CreateFlippingBot(
|
||||
dataFlippingBot.AccountName,
|
||||
dataFlippingBot.MoneyManagement,
|
||||
dataFlippingBot.Name,
|
||||
dataFlippingBot.Ticker,
|
||||
dataFlippingBot.Scenario,
|
||||
dataFlippingBot.Timeframe,
|
||||
dataFlippingBot.IsForWatchingOnly));
|
||||
var bot3 = _taskCache.AddOrGetExisting(backupBot.Name, flippingBot).Result;
|
||||
bot3.LoadBackup(backupBot);
|
||||
bot3.Start();
|
||||
break;
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
public class LoadBackupBotCommand : IRequest<string>
|
||||
|
||||
Reference in New Issue
Block a user