Fix agent count

This commit is contained in:
2025-09-28 23:34:56 +07:00
parent 57b3603302
commit 014a3ed7e5
5 changed files with 6 additions and 13 deletions

View File

@@ -30,7 +30,7 @@ public interface IPlatformSummaryGrain : IGrainWithStringKey
/// Increments the total agent count when a new agent is activated
/// </summary>
[OneWay]
Task IncrementAgentCountAsync();
Task RefreshAgentCountAsync();
[OneWay]
Task OnPositionClosedAsync(PositionClosedEvent evt);

View File

@@ -92,7 +92,7 @@ public class AgentGrain : Grain, IAgentGrain
await ServiceScopeHelpers.WithScopedService<IGrainFactory>(_scopeFactory, async grainFactory =>
{
var platformGrain = grainFactory.GetGrain<IPlatformSummaryGrain>("platform-summary");
await platformGrain.IncrementAgentCountAsync();
await platformGrain.RefreshAgentCountAsync();
_logger.LogDebug("Notified platform summary about new agent activation for user {UserId}", userId);
});
}

View File

@@ -81,7 +81,6 @@ public class PlatformSummaryGrain : Grain, IPlatformSummaryGrain, IRemindable
_logger.LogInformation("Created initial empty daily snapshot for {Date}", today);
}
_state.State.TotalAgents = await _agentService.GetTotalAgentCount();
await RefreshDataAsync();
}
@@ -219,16 +218,12 @@ public class PlatformSummaryGrain : Grain, IPlatformSummaryGrain, IRemindable
}
}
public async Task IncrementAgentCountAsync()
public async Task RefreshAgentCountAsync()
{
try
{
_logger.LogInformation("Incrementing agent count from {CurrentCount} to {NewCount}",
_state.State.TotalAgents, _state.State.TotalAgents + 1);
_state.State.TotalAgents++;
_state.State.TotalAgents = await _agentService.GetTotalAgentCount();
await _state.WriteStateAsync();
_logger.LogInformation("Agent count incremented to: {NewCount}", _state.State.TotalAgents);
}
catch (Exception ex)

View File

@@ -116,8 +116,6 @@ public class UserService : IUserService
var agentGrain = _grainFactory.GetGrain<IAgentGrain>(user.Id);
await agentGrain.InitializeAsync(user.Id, string.Empty);
_logger.LogInformation("AgentGrain initialized for new user {UserId}", user.Id);
}
catch (Exception ex)
{