64 lines
1.9 KiB
Swift
64 lines
1.9 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This source file is part of the SwiftNIO open source project
|
|
//
|
|
// Copyright (c) 2019-2021 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 NIOCore
|
|
import NIOEmbedded
|
|
|
|
final class ChannelPipelineBenchmark: Benchmark {
|
|
private final class NoOpHandler: ChannelInboundHandler, RemovableChannelHandler {
|
|
typealias InboundIn = Any
|
|
}
|
|
private final class ConsumingHandler: ChannelInboundHandler, RemovableChannelHandler {
|
|
typealias InboundIn = Any
|
|
|
|
func channelReadComplete(context: ChannelHandlerContext) {
|
|
}
|
|
}
|
|
|
|
private let channel: EmbeddedChannel
|
|
private let runCount: Int
|
|
private let extraHandlers = 4
|
|
private var handlers: [RemovableChannelHandler] = []
|
|
|
|
init(runCount: Int) {
|
|
self.channel = EmbeddedChannel()
|
|
self.runCount = runCount
|
|
}
|
|
|
|
func setUp() throws {
|
|
for _ in 0..<self.extraHandlers {
|
|
let handler = NoOpHandler()
|
|
self.handlers.append(handler)
|
|
try self.channel.pipeline.addHandler(handler).wait()
|
|
}
|
|
let handler = ConsumingHandler()
|
|
self.handlers.append(handler)
|
|
try self.channel.pipeline.addHandler(handler).wait()
|
|
}
|
|
|
|
func tearDown() {
|
|
let handlersToRemove = self.handlers
|
|
self.handlers.removeAll()
|
|
try! handlersToRemove.forEach {
|
|
try self.channel.pipeline.removeHandler($0).wait()
|
|
}
|
|
}
|
|
|
|
func run() -> Int {
|
|
for _ in 0..<self.runCount {
|
|
self.channel.pipeline.fireChannelReadComplete()
|
|
}
|
|
return 1
|
|
}
|
|
}
|