Adding more streaming benchmark scenarios (#10049)
This commit is contained in:
parent
48d2f40192
commit
e3bf01eec3
|
|
@ -3,6 +3,7 @@
|
||||||
|
|
||||||
using System;
|
using System;
|
||||||
using System.Buffers;
|
using System.Buffers;
|
||||||
|
using System.Collections.Generic;
|
||||||
using System.IO;
|
using System.IO;
|
||||||
using System.IO.Pipelines;
|
using System.IO.Pipelines;
|
||||||
using System.Reactive.Linq;
|
using System.Reactive.Linq;
|
||||||
|
|
@ -168,6 +169,26 @@ namespace Microsoft.AspNetCore.SignalR.Microbenchmarks
|
||||||
return channel.Reader;
|
return channel.Reader;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public async IAsyncEnumerable<int> StreamIAsyncEnumerableCount(int count)
|
||||||
|
{
|
||||||
|
await Task.Yield();
|
||||||
|
|
||||||
|
for (var i = 0; i < count; i++)
|
||||||
|
{
|
||||||
|
yield return i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public async IAsyncEnumerable<int> StreamIAsyncEnumerableCountCompletedTask(int count)
|
||||||
|
{
|
||||||
|
await Task.CompletedTask;
|
||||||
|
|
||||||
|
for (var i = 0; i < count; i++)
|
||||||
|
{
|
||||||
|
yield return i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
public async Task UploadStream(ChannelReader<string> channelReader)
|
public async Task UploadStream(ChannelReader<string> channelReader)
|
||||||
{
|
{
|
||||||
while (await channelReader.WaitToReadAsync())
|
while (await channelReader.WaitToReadAsync())
|
||||||
|
|
@ -177,6 +198,13 @@ namespace Microsoft.AspNetCore.SignalR.Microbenchmarks
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public async Task UploadStreamIAsynEnumerable(IAsyncEnumerable<string> stream)
|
||||||
|
{
|
||||||
|
await foreach (var item in stream)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
[Benchmark]
|
[Benchmark]
|
||||||
|
|
@ -236,6 +264,24 @@ namespace Microsoft.AspNetCore.SignalR.Microbenchmarks
|
||||||
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
[Benchmark]
|
||||||
|
public async Task StreamIAsyncEnumerableCount_Zero()
|
||||||
|
{
|
||||||
|
await _dispatcher.DispatchMessageAsync(_connectionContext, new StreamInvocationMessage("123", "StreamIAsyncEnumerableCount", new object[] { 0 }));
|
||||||
|
|
||||||
|
await (_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted.Task;
|
||||||
|
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
||||||
|
}
|
||||||
|
|
||||||
|
[Benchmark]
|
||||||
|
public async Task StreamIAsyncEnumerableCompletedTaskCount_Zero()
|
||||||
|
{
|
||||||
|
await _dispatcher.DispatchMessageAsync(_connectionContext, new StreamInvocationMessage("123", "StreamIAsyncEnumerableCountCompletedTask", new object[] { 0 }));
|
||||||
|
|
||||||
|
await (_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted.Task;
|
||||||
|
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
||||||
|
}
|
||||||
|
|
||||||
[Benchmark]
|
[Benchmark]
|
||||||
public async Task StreamChannelReaderCount_One()
|
public async Task StreamChannelReaderCount_One()
|
||||||
{
|
{
|
||||||
|
|
@ -245,6 +291,24 @@ namespace Microsoft.AspNetCore.SignalR.Microbenchmarks
|
||||||
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
[Benchmark]
|
||||||
|
public async Task StreamIAsyncEnumerableCount_One()
|
||||||
|
{
|
||||||
|
await _dispatcher.DispatchMessageAsync(_connectionContext, new StreamInvocationMessage("123", "StreamIAsyncEnumerableCount", new object[] { 1 }));
|
||||||
|
|
||||||
|
await (_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted.Task;
|
||||||
|
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
||||||
|
}
|
||||||
|
|
||||||
|
[Benchmark]
|
||||||
|
public async Task StreamIAsyncEnumerableCompletedTaskCount_One()
|
||||||
|
{
|
||||||
|
await _dispatcher.DispatchMessageAsync(_connectionContext, new StreamInvocationMessage("123", "StreamIAsyncEnumerableCountCompletedTask", new object[] { 1 }));
|
||||||
|
|
||||||
|
await (_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted.Task;
|
||||||
|
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
||||||
|
}
|
||||||
|
|
||||||
[Benchmark]
|
[Benchmark]
|
||||||
public async Task StreamChannelReaderCount_Thousand()
|
public async Task StreamChannelReaderCount_Thousand()
|
||||||
{
|
{
|
||||||
|
|
@ -254,6 +318,24 @@ namespace Microsoft.AspNetCore.SignalR.Microbenchmarks
|
||||||
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
[Benchmark]
|
||||||
|
public async Task StreamIAsyncEnumerableCount_Thousand()
|
||||||
|
{
|
||||||
|
await _dispatcher.DispatchMessageAsync(_connectionContext, new StreamInvocationMessage("123", "StreamIAsyncEnumerableCount", new object[] { 1000 }));
|
||||||
|
|
||||||
|
await (_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted.Task;
|
||||||
|
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
||||||
|
}
|
||||||
|
|
||||||
|
[Benchmark]
|
||||||
|
public async Task StreamIAsyncEnumerableCompletedTaskCount_Thousand()
|
||||||
|
{
|
||||||
|
await _dispatcher.DispatchMessageAsync(_connectionContext, new StreamInvocationMessage("123", "StreamIAsyncEnumerableCountCompletedTask", new object[] { 1000 }));
|
||||||
|
|
||||||
|
await (_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted.Task;
|
||||||
|
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
||||||
|
}
|
||||||
|
|
||||||
[Benchmark]
|
[Benchmark]
|
||||||
public async Task UploadStream_One()
|
public async Task UploadStream_One()
|
||||||
{
|
{
|
||||||
|
|
@ -265,6 +347,17 @@ namespace Microsoft.AspNetCore.SignalR.Microbenchmarks
|
||||||
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
[Benchmark]
|
||||||
|
public async Task UploadStreamIAsyncEnumerable_One()
|
||||||
|
{
|
||||||
|
await _dispatcher.DispatchMessageAsync(_connectionContext, new InvocationMessage("123", nameof(TestHub.UploadStreamIAsynEnumerable), Array.Empty<object>(), streamIds: new string[] { "1" }));
|
||||||
|
await _dispatcher.DispatchMessageAsync(_connectionContext, new StreamItemMessage("1", "test"));
|
||||||
|
await _dispatcher.DispatchMessageAsync(_connectionContext, CompletionMessage.Empty("1"));
|
||||||
|
|
||||||
|
await (_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted.Task;
|
||||||
|
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
||||||
|
}
|
||||||
|
|
||||||
[Benchmark]
|
[Benchmark]
|
||||||
public async Task UploadStream_Thousand()
|
public async Task UploadStream_Thousand()
|
||||||
{
|
{
|
||||||
|
|
@ -278,5 +371,19 @@ namespace Microsoft.AspNetCore.SignalR.Microbenchmarks
|
||||||
await (_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted.Task;
|
await (_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted.Task;
|
||||||
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
[Benchmark]
|
||||||
|
public async Task UploadStreamIAsyncEnumerable_Thousand()
|
||||||
|
{
|
||||||
|
await _dispatcher.DispatchMessageAsync(_connectionContext, new InvocationMessage("123", nameof(TestHub.UploadStreamIAsynEnumerable), Array.Empty<object>(), streamIds: new string[] { "1" }));
|
||||||
|
for (var i = 0; i < 1000; ++i)
|
||||||
|
{
|
||||||
|
await _dispatcher.DispatchMessageAsync(_connectionContext, new StreamItemMessage("1", "test"));
|
||||||
|
}
|
||||||
|
await _dispatcher.DispatchMessageAsync(_connectionContext, CompletionMessage.Empty("1"));
|
||||||
|
|
||||||
|
await (_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted.Task;
|
||||||
|
(_connectionContext as NoErrorHubConnectionContext).ReceivedCompleted = new TaskCompletionSource<object>();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue