performance/src/benchmarks/micro/libraries/System.IO/BinaryWriterExtendedTests.cs at c0cca99d2dba0f42f97f0cdd8d36993665431cea · dotnet/performance · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

using System.Text;
using BenchmarkDotNet.Attributes;
using MicroBenchmarks;

namespace System.IO.Tests
{
    [BenchmarkCategory(Categories.Libraries)]
    public class BinaryWriterExtendedTests
    {
        private string _asciiInput;
        private char[] _asciiInputAsChars;
        private string _nonAsciiInput;
        private char[] _nonAsciiInputAsChars;
        private BinaryWriter _bw;

        [Params(32, 8_000, 2_000_000)]
        public int StringLengthInChars;

        [GlobalSetup]
        public void Setup()
        {
            _bw = new BinaryWriter(new NullWriteStream());

            _asciiInput = new string('x', StringLengthInChars);
            _asciiInputAsChars = _asciiInput.ToCharArray();
            // '\u00E0' (à) encodes to 2 bytes in UTF-8, exercising multi-byte encoding paths
            _nonAsciiInput = new string('\u00E0', StringLengthInChars);
            _nonAsciiInputAsChars = _nonAsciiInput.ToCharArray();
        }

        [Benchmark]
        [MemoryRandomization]
        public void WriteAsciiCharArray()
        {
            _bw.Write(_asciiInputAsChars);
        }

        [Benchmark]
        [MemoryRandomization]
        public void WriteAsciiString()
        {
            _bw.Write(_asciiInput);
        }

        [Benchmark]
        [MemoryRandomization]
        public void WriteNonAsciiCharArray()
        {
            _bw.Write(_nonAsciiInputAsChars);
        }

        [Benchmark]
        [MemoryRandomization]
        public void WriteNonAsciiString()
        {
            _bw.Write(_nonAsciiInput);
        }
    }

    /// <summary>
    /// Benchmarks for BinaryWriter with a non-UTF-8 encoding, exercising the
    /// _useFastUtf8 = false code path.
    /// </summary>
    [BenchmarkCategory(Categories.Libraries)]
    public class BinaryWriterUnicodeEncodingTests
    {
        private string _asciiInput;
        private char[] _asciiInputAsChars;
        private string _nonAsciiInput;
        private char[] _nonAsciiInputAsChars;
        private BinaryWriter _bw;

        [Params(32, 8_000, 2_000_000)]
        public int StringLengthInChars;

        [GlobalSetup]
        public void Setup()
        {
            _bw = new BinaryWriter(new NullWriteStream(), Encoding.Unicode);

            _asciiInput = new string('x', StringLengthInChars);
            _asciiInputAsChars = _asciiInput.ToCharArray();
            // '\u00E0' (à) encodes to 2 bytes in UTF-8, exercising multi-byte encoding paths
            _nonAsciiInput = new string('\u00E0', StringLengthInChars);
            _nonAsciiInputAsChars = _nonAsciiInput.ToCharArray();
        }

        [Benchmark]
        [MemoryRandomization]
        public void WriteAsciiCharArray()
        {
            _bw.Write(_asciiInputAsChars);
        }

        [Benchmark]
        [MemoryRandomization]
        public void WriteAsciiString()
        {
            _bw.Write(_asciiInput);
        }

        [Benchmark]
        [MemoryRandomization]
        public void WriteNonAsciiCharArray()
        {
            _bw.Write(_nonAsciiInputAsChars);
        }

        [Benchmark]
        [MemoryRandomization]
        public void WriteNonAsciiString()
        {
            _bw.Write(_nonAsciiInput);
        }
    }
}