Oda/managing-apps
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add bot worker

Browse Source
This commit is contained in:
cryptooda
2024-06-28 17:26:28 +07:00
parent 7a6d656335
commit d96b2a4c3b
24 changed files with 120 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
src/Managing.Application/ManageBot/BackupBotCommandHandler.cs
View File

@@ -1,63 +0,0 @@
using Managing.Domain.Bots;
using MediatR;
using static Managing.Common.Enums;
using Managing.Application.Abstractions;
namespace Managing.Application.ManageBot
{
public class BackupBotCommandHandler : IRequestHandler<BackupBotCommand, bool>
{
private readonly ITaskCache _taskCache;
private readonly IBotService _botService;
public BackupBotCommandHandler(ITaskCache taskCache, IBotService botService)
{
_taskCache = taskCache;
_botService = botService;
}
public Task<bool> Handle(BackupBotCommand request, CancellationToken cancellationToken)
{
var botBackup = new BotBackup
{
Name = request.Name,
BotType = request.BotType,
Data = ""
};
switch (request.BotType)
{
case BotType.SimpleBot:
var simpleBot = _taskCache.Get<IBot>(request.Name);
botBackup.Data = simpleBot.GetBackup();
break;
case BotType.ScalpingBot:
var scalpingBot = _taskCache.Get<ITradingBot>(request.Name);
botBackup.Data = scalpingBot.GetBackup();
break;
case BotType.FlippingBot:
var flippingBot = _taskCache.Get<ITradingBot>(request.Name);
botBackup.Data = flippingBot.GetBackup();
break;
default:
return Task.FromResult(false);
}
_botService.SaveBotBackup(botBackup);
return Task.FromResult(true);
}
}
public class BackupBotCommand : IRequest<bool>
{
public string Name { get; }
public BotType BotType { get; }
public BackupBotCommand(BotType botType, string name)
{
BotType = botType;
Name = name;
}
}
}

32
src/Managing.Application/ManageBot/BotService.cs
View File

@@ -1,35 +1,33 @@
using Managing.Application.Abstractions;
using Managing.Application.Bots;
using Managing.Common;
namespace Managing.Application.ManageBot
{
public class BotService : IBotService
{
private readonly IBotFactory _botFactory;
private readonly IBotRepository _botRepository;
public BotService(IBotFactory botFactory, IBotRepository botRepository)
public BotService(IBotRepository botRepository)
{
_botFactory = botFactory;
_botRepository = botRepository;
}
// public void CreateBot()
// {
// // Use the factory to create a new bot
// return _botFactory.CreateBot();
// }
// public void LoadBotBackup(BotBackup botBackup)
// {
// // Deserialize the JSON into a Bot object
// var bot = JsonConvert.DeserializeObject<Bot>(json);
// return bot;
// }
public async void SaveBotBackup(BotBackup botBackup)
{
await _botRepository.InsertBotAsync(botBackup);
}
public void SaveBotBackup(string name, Enums.BotType botType, string data)
{
var botBackup = new BotBackup
{
Name = name,
BotType = botType,
Data = data
};
_botRepository.InsertBotAsync(botBackup);
}
}
}

28
src/Managing.Application/ManageBot/LoadBackupBotCommandHandler.cs
View File

@@ -3,9 +3,9 @@ using MediatR;
using static Managing.Common.Enums;
using Managing.Application.Abstractions;
using Managing.Core;
using Managing.Domain.MoneyManagements;
using Newtonsoft.Json;
using Managing.Application.Bots;
using Microsoft.Extensions.Logging;
namespace Managing.Application.ManageBot
{
@@ -15,20 +15,18 @@ namespace Managing.Application.ManageBot
private readonly ITaskCache _taskCache;
private readonly IMoneyManagementService _moneyManagementService;
private readonly IBotRepository _botRepository;
private readonly IMediator _mediator;
private readonly ILogger<LoadBackupBotCommandHandler> _logger;
public LoadBackupBotCommandHandler(
IBotFactory botFactory,
ITaskCache taskCache,
IMoneyManagementService moneyManagementService,
IBotRepository botRepository,
IMediator mediator)
IBotRepository botRepository)
{
_botFactory = botFactory;
_taskCache = taskCache;
_moneyManagementService = moneyManagementService;
_botRepository = botRepository;
_mediator = mediator;
}
public Task<string> Handle(LoadBackupBotCommand request, CancellationToken cancellationToken)
@@ -37,6 +35,8 @@ namespace Managing.Application.ManageBot
var backupBots = _botRepository.GetBots();
var result = new Dictionary<string, BotStatus>();
_logger.LogInformation($"Loading {backupBots.Count()} backup bots");
foreach (var backupBot in backupBots)
{
// Check if bot is existing in cache
@@ -46,7 +46,8 @@ namespace Managing.Application.ManageBot
var simpleBot = _taskCache.Get<IBot>(backupBot.Name);
if (simpleBot == null)
{
StartBot(request, backupBot);
_logger.LogInformation($"Starting backup bot {backupBot.Name}");
StartBot(backupBot);
simpleBot.LoadBackup(backupBot);
result.Add(simpleBot.GetName(), BotStatus.Backup);
}
@@ -61,7 +62,8 @@ namespace Managing.Application.ManageBot
var scalpingBot = _taskCache.Get<ITradingBot>(backupBot.Name);
if (scalpingBot == null)
{
StartBot(request, backupBot);
_logger.LogInformation($"Starting backup bot {backupBot.Name}");
StartBot(backupBot);
scalpingBot.LoadBackup(backupBot);
result.Add(scalpingBot.GetName(), BotStatus.Backup);
}
@@ -80,13 +82,13 @@ namespace Managing.Application.ManageBot
return Task.FromResult(botStatus.ToString());
}
private void StartBot(LoadBackupBotCommand request, BotBackup backupBot)
private void StartBot(BotBackup backupBot)
{
switch (backupBot.BotType)
{
case BotType.SimpleBot:
Func<Task<IBot>> simpleBot = () => Task.FromResult(_botFactory.CreateSimpleBot(request.Name, null));
var bot1 = _taskCache.AddOrGetExisting(request.Name, simpleBot).Result;
Func<Task<IBot>> simpleBot = () => Task.FromResult(_botFactory.CreateSimpleBot(backupBot.Name, null));
var bot1 = _taskCache.AddOrGetExisting(backupBot.Name, simpleBot).Result;
bot1.LoadBackup(backupBot);
break;
case BotType.ScalpingBot:
@@ -99,7 +101,7 @@ namespace Managing.Application.ManageBot
data.Scenario,
data.Timeframe,
data.IsForWatchingOnly));
var bot2 = _taskCache.AddOrGetExisting(request.Name, scalpingBot).Result;
var bot2 = _taskCache.AddOrGetExisting(backupBot.Name, scalpingBot).Result;
bot2.LoadBackup(backupBot);
break;
case BotType.FlippingBot:
@@ -112,7 +114,7 @@ namespace Managing.Application.ManageBot
dataFlippingBot.Scenario,
dataFlippingBot.Timeframe,
dataFlippingBot.IsForWatchingOnly));
var bot3 = _taskCache.AddOrGetExisting(request.Name, flippingBot).Result;
var bot3 = _taskCache.AddOrGetExisting(backupBot.Name, flippingBot).Result;
bot3.LoadBackup(backupBot);
break;
};
@@ -121,7 +123,5 @@ namespace Managing.Application.ManageBot
public class LoadBackupBotCommand : IRequest<string>
{
public string Name { get; internal set; }
public string AccountName { get; internal set; }
}
}