fix plug to the store

This commit is contained in:
2025-09-14 17:00:26 +07:00
parent caa0d9e1a6
commit 7809e45b28
3 changed files with 95 additions and 25 deletions

View File

@@ -123,7 +123,34 @@ public class TradingBotBase : ITradingBot
{
var grainKey = CandleHelpers.GetCandleStoreGrainKey(Account.Exchange, Config.Ticker, Config.Timeframe);
var grain = grainFactory.GetGrain<ICandleStoreGrain>(grainKey);
try
{
// Add a small delay to ensure grain is fully activated
await Task.Delay(100);
LastCandle = await grain.GetLastCandle();
}
catch (InvalidOperationException ex) when (ex.Message.Contains("invalid activation"))
{
Logger.LogWarning("Grain activation failed for {GrainKey}, retrying in 1 second...", grainKey);
// Wait a bit longer and retry once
await Task.Delay(1000);
try
{
LastCandle = await grain.GetLastCandle();
}
catch (Exception retryEx)
{
Logger.LogError(retryEx, "Failed to load last candle for {GrainKey} after retry", grainKey);
LastCandle = null;
}
}
catch (Exception ex)
{
Logger.LogError(ex, "Error loading last candle for {GrainKey}", grainKey);
LastCandle = null;
}
});
}

View File

@@ -33,6 +33,8 @@ public class CandleStoreGrain : Grain, ICandleStoreGrain, IAsyncObserver<Candle>
}
public override async Task OnActivateAsync(CancellationToken cancellationToken)
{
try
{
var grainKey = this.GetPrimaryKeyString();
_logger.LogInformation("CandleStoreGrain activated for key: {GrainKey}", grainKey);
@@ -51,6 +53,20 @@ public class CandleStoreGrain : Grain, ICandleStoreGrain, IAsyncObserver<Candle>
await base.OnActivateAsync(cancellationToken);
}
catch (Exception ex)
{
_logger.LogError(ex, "Error during CandleStoreGrain activation for key: {GrainKey}", this.GetPrimaryKeyString());
// Ensure state is initialized even if there's an error
if (_state.State.Candles == null)
{
_state.State.Candles = new List<Candle>();
await _state.WriteStateAsync();
}
throw; // Re-throw to let Orleans handle the activation failure
}
}
public override async Task OnDeactivateAsync(DeactivationReason reason, CancellationToken cancellationToken)
{
@@ -68,7 +84,14 @@ public class CandleStoreGrain : Grain, ICandleStoreGrain, IAsyncObserver<Candle>
{
try
{
return Task.FromResult(_state.State.Candles?.ToList() ?? new List<Candle>());
// Ensure state is initialized
if (_state.State.Candles == null)
{
_logger.LogWarning("State not initialized for grain {GrainKey}, returning empty list", this.GetPrimaryKeyString());
return Task.FromResult(new List<Candle>());
}
return Task.FromResult(_state.State.Candles.ToList());
}
catch (Exception ex)
{
@@ -135,14 +158,17 @@ public class CandleStoreGrain : Grain, ICandleStoreGrain, IAsyncObserver<Candle>
_logger.LogInformation("Loading initial candles for {Exchange}-{Ticker}-{Timeframe}",
exchange, ticker, timeframe);
// Ensure state is initialized first
if (_state.State.Candles == null)
{
_state.State.Candles = new List<Candle>();
}
// Load the last 500 candles from the database
var endDate = DateTime.UtcNow;
var startDate =
CandleHelpers
.GetBotPreloadSinceFromTimeframe(timeframe); // Look back 30 days to ensure we get enough data
var startDate = CandleHelpers.GetBotPreloadSinceFromTimeframe(timeframe);
var candles =
await _candleRepository.GetCandles(exchange, ticker, timeframe, startDate, endDate, MaxCandleCount);
var candles = await _candleRepository.GetCandles(exchange, ticker, timeframe, startDate, endDate, MaxCandleCount);
if (candles?.Any() == true)
{
@@ -195,7 +221,22 @@ public class CandleStoreGrain : Grain, ICandleStoreGrain, IAsyncObserver<Candle>
public Task<Candle> GetLastCandle()
{
return Task.FromResult(_state.State.Candles?.LastOrDefault());
try
{
// Ensure state is initialized
if (_state.State.Candles == null || _state.State.Candles.Count == 0)
{
_logger.LogDebug("No candles available for grain {GrainKey}", this.GetPrimaryKeyString());
return Task.FromResult<Candle>(null);
}
return Task.FromResult(_state.State.Candles.LastOrDefault());
}
catch (Exception ex)
{
_logger.LogError(ex, "Error retrieving last candle for grain {GrainKey}", this.GetPrimaryKeyString());
return Task.FromResult<Candle>(null);
}
}
}

View File

@@ -106,12 +106,12 @@ public class PriceFetcher15MinGrain : Grain, IPriceFetcher15MinGrain, IRemindabl
// Get the last candle date from database
var existingCandles = await _candleRepository.GetCandles(exchange, ticker, timeframe,
DateTime.UtcNow.AddDays(-30), 1);
DateTime.UtcNow.AddDays(-30), DateTime.UtcNow.AddDays(1), 20);
var isFirstCall = !existingCandles.Any();
var startDate = !isFirstCall
? existingCandles.Max(c => c.Date).AddMinutes(GetTimeframeMinutes(timeframe))
? existingCandles.Last().Date
: new DateTime(2017, 1, 1);
// Fetch new candles from external API
@@ -125,7 +125,9 @@ public class PriceFetcher15MinGrain : Grain, IPriceFetcher15MinGrain, IRemindabl
newCandles.Count, streamKey);
// Process all new candles
var processedCandles = newCandles.OrderBy(c => c.Date).ToList();
var processedCandles = newCandles.OrderBy(c => c.Date)
.Where(c => c.Date <= DateTime.UtcNow) // Avoid duplicates
.ToList();
foreach (var candle in processedCandles)
{