public override async Task RunAsync()
{
var pool = new EventHubClientPool(() => CreateEventHubClient());
try
{
int count = 0;
var elapsed = await TimeFunctionAsync(async() =>
{
var cancellationTokenSource = new CancellationTokenSource(_samplingTime);
var threads = (from i in Enumerable.Range(0, _threadCount)
select OneThreadAsync(pool, cancellationTokenSource.Token)).ToArray();
await Task.WhenAll(threads);
var counts = from t in threads
select t.Result;
count = counts.Sum();
});
Console.WriteLine($"Total Events: {count}");
Console.WriteLine($"Duration: {elapsed}");
Console.WriteLine($"Throughput: {(double)count / elapsed.TotalSeconds}");
Console.WriteLine($"Connection Pool size: {pool.PoolSize}");
}
finally
{
await pool.CloseAsync();
}
}
public UnsafeBufferBatchEventHubClient(EventHubClientPool pool, int batchSize)
{
_clientPool = pool ?? throw new ArgumentNullException(nameof(pool));
if (batchSize <= 0)
{
throw new ArgumentOutOfRangeException(nameof(batchSize), "Must be greater than zero");
}
_batchSize = batchSize;
}
private async Task <int> OneThreadAsync(EventHubClientPool pool, CancellationToken cancellationToken)
{
int eventCount = 0;
while (!cancellationToken.IsCancellationRequested)
{
var client = pool.AcquireClient();
var sendTask = client.SendAsync(GetDummyEventObject());
if (_earlyConnectionRelease)
{
pool.ReleaseClient(client);
await sendTask;
}
else
{
await sendTask;
pool.ReleaseClient(client);
}
++eventCount;
}
return(eventCount);
}
