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
|
//===----------------------------------------------------------------------===//
//
// This source file is part of the SwiftNIO open source project
//
// Copyright (c) 2019 Apple Inc. and the SwiftNIO project authors
// Licensed under Apache License v2.0
//
// See LICENSE.txt for license information
// See CONTRIBUTORS.txt for the list of SwiftNIO project authors
//
// SPDX-License-Identifier: Apache-2.0
//
//===----------------------------------------------------------------------===//
import NIO
import NIOWebSocket
final class WebSocketFrameDecoderBenchmark {
private let channel: EmbeddedChannel
private let runCount: Int
private let dataSize: Int
private let maskingKey: WebSocketMaskingKey?
private var data: ByteBuffer!
init(dataSize: Int, runCount: Int, maskingKey: WebSocketMaskingKey? = nil) {
self.channel = EmbeddedChannel()
self.dataSize = dataSize
self.maskingKey = maskingKey
self.runCount = runCount
}
}
extension WebSocketFrameDecoderBenchmark: Benchmark {
func setUp() throws {
self.data = ByteBufferAllocator().webSocketFrame(size: dataSize, maskingKey: maskingKey)
try self.channel.pipeline.addHandler(ByteToMessageHandler(WebSocketFrameDecoder(maxFrameSize: dataSize))).wait()
}
func tearDown() {
_ = try! self.channel.finish()
}
func run() throws -> Int {
for _ in 0..<self.runCount {
try self.channel.writeInbound(self.data)
let _: WebSocketFrame? = try self.channel.readInbound()
}
return 1
}
}
extension ByteBufferAllocator {
fileprivate func webSocketFrame(size: Int, maskingKey: WebSocketMaskingKey?) -> ByteBuffer {
var data = self.buffer(capacity: size)
// Calculate some information about the mask.
let maskBitMask: UInt8 = maskingKey != nil ? 0x80 : 0x00
// Time to add the extra bytes. To avoid checking this twice, we also start writing stuff out here.
switch size {
case 0...125:
data.writeInteger(UInt8(0x81))
data.writeInteger(UInt8(size) | maskBitMask)
case 126...Int(UInt16.max):
data.writeInteger(UInt8(0x81))
data.writeInteger(UInt8(126) | maskBitMask)
data.writeInteger(UInt16(size))
default:
data.writeInteger(UInt8(0x81))
data.writeInteger(UInt8(127) | maskBitMask)
data.writeInteger(UInt64(size))
}
if let maskingKey = maskingKey {
data.writeBytes(maskingKey)
}
data.writeBytes(repeatElement(0, count: size))
return data
}
}
|
