/// <summary>
/// CookedRabbit RabbitConnectionPool factory.
/// </summary>
/// <param name="rabbitSeasoning"></param>
/// <returns><see cref="IRabbitConnectionPool"/></returns>
public static async Task <IRabbitConnectionPool> CreateRabbitConnectionPoolAsync(RabbitSeasoning rabbitSeasoning)
{
var rcp = new RabbitConnectionPool();
await rcp.Initialize(rabbitSeasoning).ConfigureAwait(false);
return(rcp);
}
/// <summary>
/// CookedRabbit RabbitConnectionPool factory.
/// </summary>
/// <param name="rabbitSeasoning"></param>
/// <returns><see cref="IRabbitConnectionPool"/></returns>
public static async Task <IRabbitConnectionPool> CreateRabbitConnectionPoolAsync(RabbitSeasoning rabbitSeasoning)
{
RabbitConnectionPool rcp = new RabbitConnectionPool();
await rcp.Initialize(rabbitSeasoning);
return(rcp);
}
public void should_cancel_running_get_operation()
{
const string ConString = "amqp://10.10.10.10/integration";
var bus = this.ConfigureBus(
"Test",
cfg =>
{
cfg.SetConnectionString(ConString);
cfg.Route("some.label");
});
var pool = new RabbitConnectionPool(bus);
var source = new CancellationTokenSource();
// ReSharper disable once MethodSupportsCancellation
var task = Task.Factory.StartNew(
() =>
{
var con = pool.Get(ConString, false, source.Token);
return(con);
});
task.Wait(5.Seconds());
source.Cancel();
Assert.Throws <AggregateException>(() => task.Wait(5.Seconds()));
}
public void should_cancel_running_get_operation()
{
const string ConString = "amqp://10.10.10.10/integration";
var bus = this.ConfigureBus(
"Test",
cfg =>
{
cfg.SetConnectionString(ConString);
cfg.Route("some.label");
});
var pool = new RabbitConnectionPool(bus);
var source = new CancellationTokenSource();
var token = source.Token;
var task = new Task <IConnection>(() => pool.Get(ConString, false, token), token);
task.Start();
// Wait for the connection pool to initialize the connection
task.Wait(TimeSpan.FromSeconds(10));
source.Cancel();
if (task.Wait(TimeSpan.FromMinutes(1)))
{
Assert.IsTrue(task.Result != null);
}
else
{
Assert.Fail("Operation has not been canceled");
}
}
public void should_remove_disposed_connections()
{
const int Count = 5;
var bus = this.ConfigureBus(
"Test",
cfg =>
{
cfg.On <BooMessage>("one")
.ReactWith(m => { })
.WithEndpoint(
builder => builder.ListenTo(builder.Topology.Declare(Queue.Named("one.queue"))));
});
var pool = new RabbitConnectionPool(bus);
var i = 0;
var conString = bus.Configuration.ConnectionString;
var source = new CancellationTokenSource();
var connections = new List <IConnection>();
while (i++ < Count)
{
var task = new Task <IConnection>(() => pool.Get(conString, false, source.Token));
task.Start();
if (!task.Wait(1.Minutes()))
{
Assert.Ignore("Failed to get a connection from a pool; unable to continue the test");
}
connections.Add(task.Result);
}
i = 0;
while (connections.Any())
{
var con = connections.First();
con.Dispose();
connections.Remove(con);
Assert.IsTrue(pool.Count == Count - ++i);
}
}
public void should_create_connections_after_drop()
{
var bus = this.ConfigureBus(
"Test",
cfg =>
{
cfg.On <BooMessage>("one")
.ReactWith(m => { })
.WithEndpoint(
builder => builder.ListenTo(builder.Topology.Declare(Queue.Named("one.queue"))));
});
var pool = new RabbitConnectionPool(bus);
var source = new CancellationTokenSource();
var conString = string.Empty;
pool.Get(conString, false, source.Token);
pool.Drop();
Assert.IsTrue(pool.Count == 0);
pool.Get(conString, false, source.Token);
Assert.IsTrue(pool.Count == 1);
}
public void should_remove_disposed_connections()
{
const int Count = 5;
var bus = this.ConfigureBus(
"Test",
cfg =>
{
cfg.On <BooMessage>("one")
.ReactWith(m => { })
.WithEndpoint(
builder => builder.ListenTo(builder.Topology.Declare(Queue.Named("one.queue"))));
});
var pool = new RabbitConnectionPool(bus);
var i = 0;
var conString = bus.Configuration.ConnectionString;
var source = new CancellationTokenSource();
var connections = new List <IConnection>();
while (i++ < Count)
{
var con = pool.Get(conString, false, source.Token);
connections.Add(con);
}
i = 0;
while (connections.Any())
{
var con = connections.First();
con.Dispose();
connections.Remove(con);
Assert.IsTrue(pool.Count == Count - ++i);
}
}
