public async Task BalancedStrategyLowTreshold()
{
using(ShimsContext.Create())
{
//create cluster
var config = new CqlConnectionStringBuilder {NewConnectionTreshold = 5};
var cluster = new Cluster(config);
//create nodes
var n = new Node(IPAddress.Parse("127.0.0.1"), cluster);
var n2 = new Node(IPAddress.Parse("127.0.0.2"), cluster);
var n3 = new Node(IPAddress.Parse("127.0.0.3"), cluster);
var n4 = new Node(IPAddress.Parse("127.0.0.4"), cluster);
var nodes = new Ring();
nodes.Update(new List<Node> {n, n2, n3, n4}, "RandomPartitioner",
new Logger(new NullLogger(), LogLevel.None));
ShimAllConnections();
var logger = cluster.LoggerManager.GetLogger("BalancedStrategyLowTresholdTest");
IConnectionStrategy strategy = new BalancedConnectionStrategy(nodes, config);
const int nr = 8;
for(int i = 0; i < nr; i++)
{
Connection connection;
using(logger.ThreadBinding())
{
connection = strategy.GetOrCreateConnection(ConnectionScope.Command, PartitionKey.None);
}
await
connection.SendRequestAsync(new QueryFrame("", CqlConsistency.Any, null), logger, 10,
CancellationToken.None);
}
Assert.AreEqual(nodes.Sum(nd => nd.ConnectionCount), nr);
Assert.IsTrue(nodes.All(nd => nd.ConnectionCount == nr/4));
}
}
public async Task BalancedStrategyManyRequests()
{
using (ShimsContext.Create())
{
//create cluster
var config = new ClusterConfig {
NewConnectionTreshold = 20
};
var cluster = new Cluster(config);
//create nodes
var n = new Node(IPAddress.Parse("127.0.0.1"), cluster);
var n2 = new Node(IPAddress.Parse("127.0.0.2"), cluster);
var n3 = new Node(IPAddress.Parse("127.0.0.3"), cluster);
var n4 = new Node(IPAddress.Parse("127.0.0.4"), cluster);
var nodes = new Ring(new List <Node> {
n, n2, n3, n4
}, "RandomPartitioner");
ShimAllConnections();
var logger = cluster.LoggerManager.GetLogger("BalancedStrategyManyRequestsTest");
IConnectionStrategy strategy = new BalancedConnectionStrategy(nodes, config);
const int nr = 80;
for (int i = 0; i < nr; i++)
{
Connection connection;
using (logger.ThreadBinding())
connection = strategy.GetOrCreateConnection(PartitionKey.None);
await connection.SendRequestAsync(new QueryFrame("", CqlConsistency.Any), logger, 10);
}
Assert.AreEqual(nodes.Sum(nd => nd.ConnectionCount), 8);
Assert.IsTrue(nodes.All(nd => nd.ConnectionCount == 2));
Assert.IsTrue(nodes.SelectMany(nd => nd).All(c => c.Load == (80 * 10) / 4 / 2));
}
}
public async Task BalancedStrategyManyRequestLowMaxConnections()
{
using (ShimsContext.Create())
{
//create cluster
var config = new ClusterConfig { NewConnectionTreshold = 5, MaxConnections = 6 };
var cluster = new Cluster(config);
//create nodes
var n1 = new Node(IPAddress.Parse("127.0.0.1"), cluster);
var n2 = new Node(IPAddress.Parse("127.0.0.2"), cluster);
var n3 = new Node(IPAddress.Parse("127.0.0.3"), cluster);
var n4 = new Node(IPAddress.Parse("127.0.0.4"), cluster);
var nodes = new Ring(new List<Node> { n1, n2, n3, n4 }, "RandomPartitioner");
ShimAllConnections();
var logger = cluster.LoggerManager.GetLogger("BalancedStrategyManyRequestLowMaxConnectionsTest");
IConnectionStrategy strategy = new BalancedConnectionStrategy(nodes, config);
const int nr = 80;
for (int i = 0; i < nr; i++)
{
Connection connection;
using (logger.ThreadBinding())
connection = strategy.GetOrCreateConnection(PartitionKey.None);
await connection.SendRequestAsync(new QueryFrame("", CqlConsistency.Any), logger, 10);
}
Assert.AreEqual(6, nodes.Sum(nd => nd.ConnectionCount));
Assert.IsTrue(nodes.All(n => n.Load == 80 * 10 / 4));
}
}
