public Task SendMessage(Immutable <Message> message, RmqSerializer serializer)
{
switch (serializer)
{
case RmqSerializer.ProtoBuf:
return(SendMessage(message, Globals.StreamProviderNameProtoBuf, Globals.StreamNameSpaceProtoBuf));
default:
return(SendMessage(message, Globals.StreamProviderNameDefault, Globals.StreamNameSpaceDefault));
}
}
public static async Task TestRmqStreamProviderOnFly(this TestCluster cluster, Action <Connection> setupProxy, int nMessages, int itersToWait, RmqSerializer serializer = RmqSerializer.Default)
{
var rand = new Random();
var messages = Enumerable.Range(1, nMessages).Select(id => new Message(id, rand.Next(1, 5) * 100)).ToArray();
var aggregator = cluster.GrainFactory.GetGrain <IAggregatorGrain>(Guid.Empty);
await aggregator.CleanUp(); // has to be done here, because the sender is also accessing the aggregator
using (var connection = GetConnectionIfNeeded(setupProxy))
{
setupProxy?.Invoke(connection);
var sender = cluster.GrainFactory.GetGrain <ISenderGrain>(Guid.Empty);
await Task.WhenAll(messages.Select(msg => sender.SendMessage(msg.AsImmutable(), serializer)));
int iters = 0;
while (!await AllMessagesSentAndDelivered(aggregator, messages) && iters < itersToWait)
{
iters++;
await Task.Delay(TimeSpan.FromSeconds(10));
}
Assert.IsTrue(await AllMessagesSentAndDelivered(aggregator, messages), await PrintError(aggregator, messages));
Assert.AreEqual(cluster.Silos.Count(), await aggregator.GetProcessingSilosCount(), "Silo count mismatch!");
}
}
public static async Task TestRmqStreamProviderWithPrefilledQueue(this TestCluster cluster, Action <Connection> setupProxyForSender, Action <Connection> setupProxyForReceiver, int nMessages, int itersToWait, RmqSerializer serializer = RmqSerializer.Default)
{
await cluster.StopPullingAgents();
var rand = new Random();
var messages = Enumerable.Range(1, nMessages).Select(id => new Message(id, rand.Next(1, 5) * 100)).ToArray();
var aggregator = cluster.GrainFactory.GetGrain <IAggregatorGrain>(Guid.Empty);
await aggregator.CleanUp(); // has to be done here, because the sender is also accessing the aggregator
using (var connection = new Connection(resetAllToxicsAndProxiesOnClose: true))
{
setupProxyForSender(connection);
var sender = cluster.GrainFactory.GetGrain <ISenderGrain>(Guid.Empty);
await Task.WhenAll(messages.Select(msg => sender.SendMessage(msg.AsImmutable(), serializer)));
}
using (var connection = new Connection(resetAllToxicsAndProxiesOnClose: true))
{
setupProxyForReceiver(connection);
await cluster.StartPullingAgents();
int iters = 0;
while (!await AllMessagesSentAndDelivered(aggregator, messages) && iters < itersToWait)
{
iters++;
await Task.Delay(TimeSpan.FromSeconds(10));
}
Assert.IsTrue(await AllMessagesSentAndDelivered(aggregator, messages));
Assert.AreEqual(cluster.Silos.Count(), await aggregator.GetProcessingSilosCount());
}
}