Update Agent balance tracking

src/Managing.Api/Controllers/DataController.cs

@@ -416,7 +416,8 @@ public class DataController : ControllerBase
             userStrategies.Select(strategy =>
                 ServiceScopeHelpers.WithScopedService<ITradingService, UserStrategyDetailsViewModel>(
                     _serviceScopeFactory,
-                    async tradingService => await MapStrategyToViewModelAsync(strategy, agentBalanceHistory, tradingService)))
+                    async tradingService =>
+                        await MapStrategyToViewModelAsync(strategy, agentBalanceHistory, tradingService)))
         );
 
         return Ok(result);
@@ -487,7 +488,7 @@ public class DataController : ControllerBase
 
         // Convert agent balance history to wallet balances dictionary
         var walletBalances = agentBalanceHistory?.AgentBalances?
-            .ToDictionary(b => b.Time, b => b.TotalValue) ?? new Dictionary<DateTime, decimal>();
+            .ToDictionary(b => b.Time, b => b.TotalBalanceValue) ?? new Dictionary<DateTime, decimal>();
 
         return new UserStrategyDetailsViewModel
         {

src/Managing.Application/Abstractions/IBotService.cs

@@ -22,6 +22,7 @@ public interface IBotService
     Task<Position> OpenPositionManuallyAsync(Guid identifier, TradeDirection direction);
     Task<Position> ClosePositionAsync(Guid identifier, Guid positionId);
     Task<TradingBotConfig> GetBotConfig(Guid identifier);
+    Task<IEnumerable<TradingBotConfig>> GetBotConfigsByIdsAsync(IEnumerable<Guid> botIds);
     Task<bool> UpdateBotStatisticsAsync(Guid identifier);
     Task<bool> SaveBotStatisticsAsync(Bot bot);
     

src/Managing.Application/Bots/Grains/AgentGrain.cs

@@ -1,6 +1,7 @@
 #nullable enable
 using Managing.Application.Abstractions;
 using Managing.Application.Abstractions.Grains;
+using Managing.Application.Abstractions.Repositories;
 using Managing.Application.Abstractions.Services;
 using Managing.Application.Bots.Models;
 using Managing.Application.Orleans;
@@ -31,6 +32,7 @@ public class AgentGrain : Grain, IAgentGrain
     private readonly IUserService _userService;
     private readonly IAccountService _accountService;
     private readonly ITradingService _tradingService;
+    private readonly IAgentBalanceRepository _agentBalanceRepository;
     private readonly IServiceScopeFactory _scopeFactory;
 
     public AgentGrain(
@@ -43,6 +45,7 @@ public class AgentGrain : Grain, IAgentGrain
         IUserService userService,
         IAccountService accountService,
         ITradingService tradingService,
+        IAgentBalanceRepository agentBalanceRepository,
         IServiceScopeFactory scopeFactory)
     {
         _state = state;
@@ -53,6 +56,7 @@ public class AgentGrain : Grain, IAgentGrain
         _userService = userService;
         _accountService = accountService;
         _tradingService = tradingService;
+        _agentBalanceRepository = agentBalanceRepository;
         _scopeFactory = scopeFactory;
     }
 
@@ -127,7 +131,8 @@ public class AgentGrain : Grain, IAgentGrain
     {
         try
         {
-            _logger.LogInformation("Position closed event received for user {UserId}, position: {PositionId}, PnL: {PnL}",
+            _logger.LogInformation(
+                "Position closed event received for user {UserId}, position: {PositionId}, PnL: {PnL}",
                 this.GetPrimaryKeyLong(), evt.PositionIdentifier, evt.RealizedPnL);
 
             await UpdateSummary();
@@ -220,9 +225,8 @@ public class AgentGrain : Grain, IAgentGrain
                 totalBalance = 0; // Set to 0 if calculation fails
             }
 
-             var bots = await ServiceScopeHelpers.WithScopedService<IBotService, IEnumerable<Bot>> (_scopeFactory, async (botService) => {
+            var bots = await ServiceScopeHelpers.WithScopedService<IBotService, IEnumerable<Bot>>(_scopeFactory,
-                return await botService.GetBotsByUser((int)this.GetPrimaryKeyLong());
+                async (botService) => { return await botService.GetBotsByUser((int)this.GetPrimaryKeyLong()); });
-            });
 
             // Calculate Runtime based on the earliest position date
             DateTime? runtime = null;
@@ -233,6 +237,18 @@ public class AgentGrain : Grain, IAgentGrain
 
             var activeStrategiesCount = bots.Count(b => b.Status == BotStatus.Running);
 
+            // Calculate bots allocation USD value from bot configurations
+            var botsAllocationUsdValue = 0m;
+            if (bots.Any())
+            {
+                var botIds = bots.Select(b => b.Identifier);
+                var botConfigs =
+                    await ServiceScopeHelpers.WithScopedService<IBotService, IEnumerable<TradingBotConfig>>(
+                        _scopeFactory,
+                        async (botService) => { return await botService.GetBotConfigsByIdsAsync(botIds); });
+                botsAllocationUsdValue = botConfigs.Sum(config => config.BotTradingBalance);
+            }
+
             var summary = new AgentSummary
             {
                 UserId = (int)this.GetPrimaryKeyLong(),
@@ -252,7 +268,11 @@ public class AgentGrain : Grain, IAgentGrain
             // Save summary to database
             await _agentService.SaveOrUpdateAgentSummary(summary);
 
-            _logger.LogDebug("Updated agent summary from position data for user {UserId}: NetPnL={NetPnL}, TotalPnL={TotalPnL}, Fees={Fees}, Volume={Volume}, Wins={Wins}, Losses={Losses}",
+            // Insert balance tracking data
+            InsertBalanceTrackingData(totalBalance, botsAllocationUsdValue, netPnL);
+
+            _logger.LogDebug(
+                "Updated agent summary from position data for user {UserId}: NetPnL={NetPnL}, TotalPnL={TotalPnL}, Fees={Fees}, Volume={Volume}, Wins={Wins}, Losses={Losses}",
                 this.GetPrimaryKeyLong(), netPnL, totalPnL, totalFees, totalVolume, totalWins, totalLosses);
         }
         catch (Exception ex)
@@ -522,7 +542,8 @@ public class AgentGrain : Grain, IAgentGrain
                 }
                 catch (Exception ex)
                 {
-                    _logger.LogWarning(ex, "Error checking open positions for bot {BotId}, skipping", botEntry.Identifier);
+                    _logger.LogWarning(ex, "Error checking open positions for bot {BotId}, skipping",
+                        botEntry.Identifier);
                     // Continue checking other bots even if one fails
                 }
             }
@@ -532,7 +553,8 @@ public class AgentGrain : Grain, IAgentGrain
         }
         catch (Exception ex)
         {
-            _logger.LogError(ex, "Error checking for open positions across all bots for user {UserId}", this.GetPrimaryKeyLong());
+            _logger.LogError(ex, "Error checking for open positions across all bots for user {UserId}",
+                this.GetPrimaryKeyLong());
             return false; // Default to false on error to avoid blocking autoswap
         }
     }
@@ -600,4 +622,33 @@ public class AgentGrain : Grain, IAgentGrain
             return null;
         }
     }
+
+    /// <summary>
+    /// Inserts balance tracking data into the AgentBalanceRepository
+    /// </summary>
+    private void InsertBalanceTrackingData(decimal totalAccountUsdValue, decimal botsAllocationUsdValue, decimal pnl)
+    {
+        try
+        {
+            var agentBalance = new AgentBalance
+            {
+                AgentName = _state.State.AgentName,
+                TotalBalanceValue = totalAccountUsdValue,
+                BotsAllocationUsdValue = botsAllocationUsdValue,
+                PnL = pnl,
+                Time = DateTime.UtcNow
+            };
+
+            _agentBalanceRepository.InsertAgentBalance(agentBalance);
+
+            _logger.LogDebug(
+                "Inserted balance tracking data for agent {AgentName}: TotalBalanceValue={TotalBalanceValue}, BotsAllocationUsdValue={BotsAllocationUsdValue}, PnL={PnL}",
+                agentBalance.AgentName, agentBalance.TotalBalanceValue, agentBalance.BotsAllocationUsdValue,
+                agentBalance.PnL);
+        }
+        catch (Exception ex)
+        {
+            _logger.LogError(ex, "Error inserting balance tracking data for agent {AgentName}", _state.State.AgentName);
+        }
+    }
 }

src/Managing.Application/ManageBot/BotService.cs

@@ -208,6 +208,28 @@ namespace Managing.Application.ManageBot
             return await grain.GetConfiguration();
         }
 
+        public async Task<IEnumerable<TradingBotConfig>> GetBotConfigsByIdsAsync(IEnumerable<Guid> botIds)
+        {
+            var configs = new List<TradingBotConfig>();
+            
+            foreach (var botId in botIds)
+            {
+                try
+                {
+                    var grain = _grainFactory.GetGrain<ILiveTradingBotGrain>(botId);
+                    var config = await grain.GetConfiguration();
+                    configs.Add(config);
+                }
+                catch (Exception ex)
+                {
+                    _tradingBotLogger.LogWarning(ex, "Failed to get configuration for bot {BotId}", botId);
+                    // Continue with other bots even if one fails
+                }
+            }
+            
+            return configs;
+        }
+
         public async Task<IEnumerable<string>> GetActiveBotsNamesAsync()
         {
             var bots = await _botRepository.GetBotsByStatusAsync(BotStatus.Running);

src/Managing.Application/Workers/BalanceTrackingWorker.cs

@@ -1,171 +0,0 @@
-using Managing.Application.Abstractions.Repositories;
-using Managing.Application.Abstractions.Services;
-using Managing.Application.ManageBot.Commands;
-using Managing.Domain.Bots;
-using Managing.Domain.Statistics;
-using MediatR;
-using Microsoft.Extensions.Logging;
-using static Managing.Common.Enums;
-
-namespace Managing.Application.Workers;
-
-public class BalanceTrackingWorker : BaseWorker<BalanceTrackingWorker>
-{
-    private readonly IMediator _mediator;
-    private readonly IAccountService _accountService;
-    private readonly IAgentBalanceRepository _agentBalanceRepository;
-    private bool _isInitialized;
-
-    public BalanceTrackingWorker(
-        ILogger<BalanceTrackingWorker> logger,
-        IServiceProvider serviceProvider,
-        IMediator mediator,
-        IAccountService accountService,
-        IAgentBalanceRepository agentBalanceRepository)
-        : base(
-            WorkerType.BalanceTracking,
-            logger,
-            TimeSpan.FromHours(1),
-            serviceProvider)
-    {
-        _mediator = mediator;
-        _accountService = accountService;
-        _agentBalanceRepository = agentBalanceRepository;
-        _isInitialized = false;
-    }
-
-    protected override async Task Run(CancellationToken cancellationToken)
-    {
-        if (!_isInitialized)
-        {
-            _logger.LogInformation("Waiting 5 minutes for bots to initialize before starting balance tracking...");
-            await Task.Delay(TimeSpan.FromMinutes(3), cancellationToken);
-            _isInitialized = true;
-        }
-
-        _logger.LogInformation("Starting balance tracking...");
-
-        // Get all active bots
-        var bots = await _mediator.Send(new GetBotsByStatusCommand(BotStatus.Running));
-
-        var botCount = bots.Count();
-        if (botCount == 0)
-        {
-            _logger.LogWarning("No active bots found. Skipping balance tracking.");
-            return;
-        }
-
-        _logger.LogInformation($"Found {botCount} active bots. Proceeding with balance tracking.");
-        await TrackBalances(bots);
-        _logger.LogInformation("Completed balance tracking");
-    }
-
-    private async Task TrackBalances(IEnumerable<Bot> bots)
-    {
-        // Group bots by agent/user
-        var botsByAgent = bots
-            .Where(b => b.User != null)
-            .GroupBy(b => b.User.AgentName)
-            .ToDictionary(g => g.Key, g => g.ToList());
-
-        foreach (var agentEntry in botsByAgent)
-        {
-            try
-            {
-                var agentName = agentEntry.Key;
-                var agentBots = agentEntry.Value;
-
-                // Check if we need to update this agent's balance
-                var lastBalance = (await _agentBalanceRepository.GetAgentBalances(
-                    agentName,
-                    DateTime.UtcNow.AddDays(-1),
-                    DateTime.UtcNow)).OrderByDescending(b => b.Time).FirstOrDefault();
-
-                if (lastBalance != null && DateTime.UtcNow.Subtract(lastBalance.Time).TotalHours < 24)
-                {
-                    _logger.LogInformation(
-                        $"Skipping agent {agentName} - Last balance update was {lastBalance.Time:g} UTC");
-                    continue;
-                }
-
-                decimal totalAgentValue = 0;
-                decimal totalBotAllocatedBalance = 0;
-                decimal totalAccountUsdValue = 0;
-                decimal botsAllocationUsdValue = 0;
-                decimal totalPnL = 0;
-
-                _logger.LogInformation($"Processing agent: {agentName} with {agentBots.Count} bots");
-
-                // Calculate total allocated balance for all bots
-                foreach (var bot in agentBots)
-                {
-                    totalBotAllocatedBalance += bot.Volume;
-                    _logger.LogInformation(
-                        $"Bot {bot.Name} allocated balance: {bot.Volume} USD");
-                }
-
-                // Get account balances for this agent (only once per agent)
-                var agent = agentBots.First().User; // Get the user object from the first bot
-                var accountBalances = _accountService.GetAccountsBalancesByUser(agent, true);
-                foreach (var accountBalance in accountBalances)
-                {
-                    if (accountBalance.Balances != null)
-                    {
-                        var accountTotalValue = accountBalance.Balances.Sum(b => b.Value);
-
-                        // If this is the account that holds the bot balances (USDC), subtract the allocated amounts
-                        var usdcBalance = accountBalance.Balances.FirstOrDefault(b => b.TokenName == "USDC");
-                        if (usdcBalance != null)
-                        {
-                            _logger.LogInformation(
-                                $"Account {accountBalance.Name} USDC balance before bot allocation: {usdcBalance.Value} USD");
-                            usdcBalance.Value -= totalBotAllocatedBalance;
-                            _logger.LogInformation(
-                                $"Account {accountBalance.Name} USDC balance after bot allocation: {usdcBalance.Value} USD");
-                        }
-
-                        totalAccountUsdValue += accountTotalValue;
-
-                        _logger.LogInformation(
-                            $"Account {accountBalance.Name} total value: {accountTotalValue} USD");
-
-                        // Log individual token balances for debugging
-                        foreach (var balance in accountBalance.Balances)
-                        {
-                            _logger.LogInformation(
-                                $"  - {balance.TokenName}: {balance.Amount} (Value: {balance.Value} USD)");
-                        }
-                    }
-                }
-
-                // Process all bots in a single iteration
-                foreach (var bot in agentBots)
-                {
-                    totalPnL += bot.Pnl;
-                }
-
-                totalAgentValue = totalAccountUsdValue + botsAllocationUsdValue;
-
-                _logger.LogInformation(
-                    $"Agent {agentName} total aggregated value: {totalAgentValue} USD (Account: {totalAccountUsdValue} USD, Bot Wallet: {botsAllocationUsdValue} USD)");
-
-                // Create and save the agent balance
-                var agentBalance = new AgentBalance
-                {
-                    AgentName = agentName,
-                    TotalValue = totalAgentValue,
-                    TotalAccountUsdValue = totalAccountUsdValue,
-                    BotsAllocationUsdValue = botsAllocationUsdValue,
-                    PnL = totalPnL,
-                    Time = DateTime.UtcNow
-                };
-
-                _agentBalanceRepository.InsertAgentBalance(agentBalance);
-            }
-            catch (Exception ex)
-            {
-                _logger.LogError(ex, $"Error processing agent {agentEntry.Key}");
-            }
-        }
-    }
-}

src/Managing.Bootstrap/ApiBootstrap.cs

@@ -443,12 +443,6 @@ public static class ApiBootstrap
 
     private static IServiceCollection AddWorkers(this IServiceCollection services, IConfiguration configuration)
     {
-        // Balance Workers
-        if (configuration.GetValue<bool>("WorkerBalancesTracking", false))
-        {
-            services.AddHostedService<BalanceTrackingWorker>();
-        }
-
         if (configuration.GetValue<bool>("WorkerNotifyBundleBacktest", false))
         {
             services.AddHostedService<NotifyBundleBacktestWorker>();

src/Managing.Domain.Statistics/AgentBalance.cs

@@ -1,8 +1,7 @@
 public class AgentBalance
 {
     public string AgentName { get; set; }
-    public decimal TotalValue { get; set; }
+    public decimal TotalBalanceValue { get; set; }
-    public decimal TotalAccountUsdValue { get; set; }
     public decimal BotsAllocationUsdValue { get; set; }
     public decimal PnL { get; set; }
     public DateTime Time { get; set; }

src/Managing.Domain/Statistics/AgentBalance.cs

@@ -3,8 +3,7 @@ namespace Managing.Domain.Statistics;
 public class AgentBalance
 {
     public string AgentName { get; set; }
-    public decimal TotalValue { get; set; }
+    public decimal TotalBalanceValue { get; set; }
-    public decimal TotalAccountUsdValue { get; set; }
     public decimal BotsAllocationUsdValue { get; set; }
     public decimal PnL { get; set; }
     public DateTime Time { get; set; }

src/Managing.Infrastructure.Database/InfluxDb/AgentBalanceRepository.cs

@@ -5,7 +5,7 @@ using Managing.Infrastructure.Databases.InfluxDb.Abstractions;
 using Managing.Infrastructure.Databases.InfluxDb.Models;
 using Microsoft.Extensions.Logging;
 
-namespace Managing.Infrastructure.Databases;
+namespace Managing.Infrastructure.Databases.InfluxDb;
 
 public class AgentBalanceRepository : IAgentBalanceRepository
 {
@@ -26,8 +26,7 @@ public class AgentBalanceRepository : IAgentBalanceRepository
             var balanceDto = new AgentBalanceDto
             {
                 AgentName = balance.AgentName,
-                TotalValue = balance.TotalValue,
+                TotalBalanceValue = balance.TotalBalanceValue,
-                TotalAccountUsdValue = balance.TotalAccountUsdValue,
                 BotsAllocationUsdValue = balance.BotsAllocationUsdValue,
                 PnL = balance.PnL,
                 Time = balance.Time
@@ -57,8 +56,7 @@ public class AgentBalanceRepository : IAgentBalanceRepository
             return result.Select(balance => new AgentBalance
             {
                 AgentName = balance.AgentName,
-                TotalValue = balance.TotalValue,
+                TotalBalanceValue = balance.TotalBalanceValue,
-                TotalAccountUsdValue = balance.TotalAccountUsdValue,
                 BotsAllocationUsdValue = balance.BotsAllocationUsdValue,
                 PnL = balance.PnL,
                 Time = balance.Time
@@ -92,8 +90,7 @@ public class AgentBalanceRepository : IAgentBalanceRepository
                     AgentBalances =                     g.Select(b => new AgentBalance
                     {
                         AgentName = b.AgentName,
-                        TotalValue = b.TotalValue,
+                        TotalBalanceValue = b.TotalBalanceValue,
-                        TotalAccountUsdValue = b.TotalAccountUsdValue,
                         BotsAllocationUsdValue = b.BotsAllocationUsdValue,
                         PnL = b.PnL,
                         Time = b.Time

src/Managing.Infrastructure.Database/InfluxDb/Models/AgentBalanceDto.cs

@@ -7,9 +7,7 @@ public class AgentBalanceDto
 {
     [Column("agent_name", IsTag = true)] public string AgentName { get; set; }
 
-    [Column("total_value")] public decimal TotalValue { get; set; }
+    [Column("total_balance_value")] public decimal TotalBalanceValue { get; set; }
-
-    [Column("total_account_usd_value")] public decimal TotalAccountUsdValue { get; set; }
 
     [Column("bots_allocation_usd_value")] public decimal BotsAllocationUsdValue { get; set; }