Fix update agentName

This commit is contained in:
2025-10-03 02:15:55 +07:00
parent 7da0143e91
commit ceb52a52d7

View File

@@ -308,25 +308,82 @@ public class AgentSummaryRepository : IAgentSummaryRepository
public async Task UpdateAgentNameAsync(int userId, string agentName)
{
var entity = await _context.AgentSummaries
.AsNoTracking()
.FirstOrDefaultAsync(a => a.UserId == userId);
if (entity != null)
try
{
entity.AgentName = agentName;
entity.UpdatedAt = DateTime.UtcNow;
// First, check if there's already a tracked entity with this key
var trackedEntity = _context.ChangeTracker.Entries<AgentSummaryEntity>()
.FirstOrDefault(e => e.Entity.UserId == userId);
_context.AgentSummaries.Update(entity);
await _context.SaveChangesAsync();
AgentSummaryEntity? entityToUpdate = null;
bool wasTracked = false;
_logger.LogInformation("Agent name updated for user {UserId} to {AgentName}", userId, agentName);
if (trackedEntity != null)
{
// Entity is already tracked, update it directly
entityToUpdate = trackedEntity.Entity;
wasTracked = true;
_logger.LogInformation("Found tracked entity for user {UserId}. Current agent name: {CurrentAgentName}, New agent name: {NewAgentName}",
userId, entityToUpdate.AgentName, agentName);
}
else
{
// Entity is not tracked, fetch it normally
entityToUpdate = await _context.AgentSummaries
.FirstOrDefaultAsync(a => a.UserId == userId);
if (entityToUpdate == null)
{
_logger.LogWarning(
"No AgentSummary found for user {UserId} when trying to update agent name to {AgentName}",
userId, agentName);
return;
}
_logger.LogInformation("Fetched entity for user {UserId}. Current agent name: {CurrentAgentName}, New agent name: {NewAgentName}",
userId, entityToUpdate.AgentName, agentName);
}
// Check if the agent name is actually different
if (entityToUpdate.AgentName == agentName)
{
_logger.LogInformation("Agent name for user {UserId} is already {AgentName}, no update needed", userId, agentName);
return;
}
// Update the entity
var oldAgentName = entityToUpdate.AgentName;
entityToUpdate.AgentName = agentName;
entityToUpdate.UpdatedAt = DateTime.UtcNow;
// If it wasn't tracked, explicitly mark it as modified
if (!wasTracked)
{
_context.Entry(entityToUpdate).State = EntityState.Modified;
}
// Log the change tracker state before saving
var modifiedEntries = _context.ChangeTracker.Entries()
.Where(e => e.State == EntityState.Modified)
.ToList();
_logger.LogInformation("Change tracker has {Count} modified entries before save", modifiedEntries.Count);
// Save changes
var changesSaved = await _context.SaveChangesAsync();
_logger.LogInformation("Agent name updated for user {UserId} from '{OldAgentName}' to '{NewAgentName}'. Changes saved: {ChangesSaved}",
userId, oldAgentName, agentName, changesSaved);
if (changesSaved == 0)
{
_logger.LogWarning("No changes were saved for user {UserId}. This might indicate a tracking issue.", userId);
}
}
else
catch (Exception ex)
{
_logger.LogWarning(
"No AgentSummary found for user {UserId} when trying to update agent name to {AgentName}",
userId, agentName);
_logger.LogError(ex, "Error updating agent name for user {UserId} to {AgentName}", userId, agentName);
throw;
}
}
@@ -337,26 +394,75 @@ public class AgentSummaryRepository : IAgentSummaryRepository
public async Task IncrementBacktestCountAsync(int userId)
{
var entity = await _context.AgentSummaries
.AsNoTracking()
.FirstOrDefaultAsync(a => a.UserId == userId);
if (entity != null)
try
{
var newCount = entity.BacktestCount + 1;
entity.BacktestCount = newCount;
entity.UpdatedAt = DateTime.UtcNow;
// First, check if there's already a tracked entity with this key
var trackedEntity = _context.ChangeTracker.Entries<AgentSummaryEntity>()
.FirstOrDefault(e => e.Entity.UserId == userId);
_context.AgentSummaries.Update(entity);
await _context.SaveChangesAsync();
AgentSummaryEntity? entityToUpdate = null;
bool wasTracked = false;
_logger.LogInformation("Backtest count incremented for user {UserId} to {BacktestCount}",
userId, entity.BacktestCount);
if (trackedEntity != null)
{
// Entity is already tracked, update it directly
entityToUpdate = trackedEntity.Entity;
wasTracked = true;
_logger.LogInformation("Found tracked entity for user {UserId}. Current backtest count: {CurrentCount}",
userId, entityToUpdate.BacktestCount);
}
else
{
// Entity is not tracked, fetch it normally
entityToUpdate = await _context.AgentSummaries
.FirstOrDefaultAsync(a => a.UserId == userId);
if (entityToUpdate == null)
{
_logger.LogWarning("No AgentSummary found for user {UserId} when trying to increment backtest count",
userId);
return;
}
_logger.LogInformation("Fetched entity for user {UserId}. Current backtest count: {CurrentCount}",
userId, entityToUpdate.BacktestCount);
}
// Update the entity
var oldCount = entityToUpdate.BacktestCount;
var newCount = oldCount + 1;
entityToUpdate.BacktestCount = newCount;
entityToUpdate.UpdatedAt = DateTime.UtcNow;
// If it wasn't tracked, explicitly mark it as modified
if (!wasTracked)
{
_context.Entry(entityToUpdate).State = EntityState.Modified;
}
// Log the change tracker state before saving
var modifiedEntries = _context.ChangeTracker.Entries()
.Where(e => e.State == EntityState.Modified)
.ToList();
_logger.LogInformation("Change tracker has {Count} modified entries before save", modifiedEntries.Count);
// Save changes
var changesSaved = await _context.SaveChangesAsync();
_logger.LogInformation("Backtest count incremented for user {UserId} from {OldCount} to {NewCount}. Changes saved: {ChangesSaved}",
userId, oldCount, newCount, changesSaved);
if (changesSaved == 0)
{
_logger.LogWarning("No changes were saved for user {UserId}. This might indicate a tracking issue.", userId);
}
}
else
catch (Exception ex)
{
_logger.LogWarning("No AgentSummary found for user {UserId} when trying to increment backtest count",
userId);
_logger.LogError(ex, "Error incrementing backtest count for user {UserId}", userId);
throw;
}
}
}