public async Task QuantaBatchTest(KeyPair alphaKeyPair, ConnectionState state, Type excpectedException)
{
Global.AppState.State = ApplicationState.Ready;
var clientConnection = new AuditorWebSocketConnection(new FakeWebSocket(), null)
{
ConnectionState = state
};
var orderEnvelope = new QuantaBatch
{
Quanta = new List <MessageEnvelope>()
}.CreateEnvelope();
orderEnvelope.Sign(alphaKeyPair);
await AssertMessageHandling(clientConnection, orderEnvelope, excpectedException);
}
public async Task QuantaBatchTest(KeyPair alphaKeyPair, ConnectionState state, Type excpectedException)
{
context.AppState.State = ApplicationState.Ready;
var clientConnection = new AuditorWebSocketConnection(context, new FakeAuditorConnectionInfo(new FakeWebSocket()))
{
ConnectionState = state
};
var orderEnvelope = new QuantaBatch
{
Quanta = new List <MessageEnvelope>()
}.CreateEnvelope();
orderEnvelope.Sign(alphaKeyPair);
using var writer = new XdrBufferWriter();
var inMessage = orderEnvelope.ToIncomingMessage(writer);
await AssertMessageHandling(clientConnection, inMessage, excpectedException);
}
private async Task SendQuantums()
{
var batchSize = Context.Settings.SyncBatchSize;
while (!cancellationToken.IsCancellationRequested)
{
var apexDiff = Context.QuantumStorage.CurrentApex - CurrentApexCursor;
if (apexDiff < 0)
{
logger.Error($"Auditor {((KeyPair)auditor.ClientPubKey).AccountId} is above current constellation state.");
await auditor.CloseConnection(System.Net.WebSockets.WebSocketCloseStatus.ProtocolError, "Auditor is above all constellation.");
return;
}
if (auditor.ConnectionState != ConnectionState.Ready && apexDiff <= 100)
{
auditor.ConnectionState = ConnectionState.Ready;
}
else if (auditor.ConnectionState == ConnectionState.Ready && apexDiff >= 10_000) //auditor is too delayed
{
auditor.ConnectionState = ConnectionState.Validated;
return;
}
if (CurrentApexCursor == Context.QuantumStorage.CurrentApex)
{
Thread.Sleep(50);
continue;
}
try
{
List <MessageEnvelope> quanta = null;
if (!Context.QuantumStorage.GetQuantaBacth(CurrentApexCursor + 1, batchSize, out quanta))
{
quanta = await Context.PersistenceManager.GetQuantaAboveApex(CurrentApexCursor, batchSize); //quanta are not found in the in-memory storage
if (quanta.Count < 1)
{
throw new Exception("No quanta from database.");
}
}
if (quanta.Count < 1)
{
throw new Exception("No quanta from storage.");
}
var firstApex = ((Quantum)quanta.First().Message).Apex;
var lastApex = ((Quantum)quanta.Last().Message).Apex;
logger.Trace($"About to sent {quanta.Count} quanta. Apex from {firstApex} to {lastApex}");
var batchMessage = new QuantaBatch {
Quanta = quanta
};
await auditor.SendMessage(batchMessage);
CurrentApexCursor = lastApex;
}
catch (Exception exc)
{
if (exc is ObjectDisposedException ||
exc.GetBaseException() is ObjectDisposedException)
{
throw;
}
logger.Error(exc, $"Unable to get quanta. Cursor: {CurrentApexCursor}; CurrentApex: {Context.QuantumStorage.CurrentApex}");
}
}
}
