78 lines
2.2 KiB
C#
78 lines
2.2 KiB
C#
// Copyright (c) .NET Foundation. All rights reserved.
|
|
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
|
|
|
|
using System;
|
|
using System.Buffers;
|
|
using System.IO;
|
|
using BenchmarkDotNet.Attributes;
|
|
using Microsoft.AspNetCore.Internal;
|
|
|
|
namespace Microsoft.AspNetCore.SignalR.Microbenchmarks
|
|
{
|
|
public class MessageParserBenchmark
|
|
{
|
|
private static readonly Random Random = new Random();
|
|
private byte[] _binaryInput;
|
|
private byte[] _textInput;
|
|
|
|
[Params(32, 64)]
|
|
public int ChunkSize { get; set; }
|
|
|
|
[Params(64, 128)]
|
|
public int MessageLength { get; set; }
|
|
|
|
[IterationSetup]
|
|
public void Setup()
|
|
{
|
|
var buffer = new byte[MessageLength];
|
|
Random.NextBytes(buffer);
|
|
var writer = MemoryBufferWriter.Get();
|
|
try
|
|
{
|
|
BinaryMessageFormatter.WriteLengthPrefix(buffer.Length, writer);
|
|
writer.Write(buffer);
|
|
_binaryInput = writer.ToArray();
|
|
}
|
|
finally
|
|
{
|
|
MemoryBufferWriter.Return(writer);
|
|
}
|
|
|
|
buffer = new byte[MessageLength];
|
|
Random.NextBytes(buffer);
|
|
writer = MemoryBufferWriter.Get();
|
|
try
|
|
{
|
|
writer.Write(buffer);
|
|
TextMessageFormatter.WriteRecordSeparator(writer);
|
|
|
|
_textInput = writer.ToArray();
|
|
}
|
|
finally
|
|
{
|
|
MemoryBufferWriter.Return(writer);
|
|
}
|
|
}
|
|
|
|
[Benchmark]
|
|
public void SingleBinaryMessage()
|
|
{
|
|
var data = new ReadOnlySequence<byte>(_binaryInput);
|
|
if (!BinaryMessageParser.TryParseMessage(ref data, out _))
|
|
{
|
|
throw new InvalidOperationException("Failed to parse");
|
|
}
|
|
}
|
|
|
|
[Benchmark]
|
|
public void SingleTextMessage()
|
|
{
|
|
var data = new ReadOnlySequence<byte>(_textInput);
|
|
if (!TextMessageParser.TryParseMessage(ref data, out _))
|
|
{
|
|
throw new InvalidOperationException("Failed to parse");
|
|
}
|
|
}
|
|
}
|
|
}
|