swift-nio-extras/Sources/NIOExtrasPerformanceTester/RollingPCAPPerformanceTest.swift

//===----------------------------------------------------------------------===//
//
// This source file is part of the SwiftNIO open source project
//
// Copyright (c) 2020-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
import NIOExtras

final class RollingPCAPPerformanceTest: Benchmark {
    let numberOfRepeats: Int

    let byteBuffer = ByteBuffer(repeating: 0x65, count: 1000)

    init(numberOfRepeats: Int) {
        self.numberOfRepeats = numberOfRepeats
    }

    func setUp() throws {
    }

    func tearDown() {
    }

    func run() throws -> Int {
        let channel = EmbeddedChannel()
        defer {
            _ = try! channel.finish()
        }

        let pcapRingBuffer = NIOPCAPRingBuffer(
            maximumFragments: 25,
            maximumBytes: 1_000_000
        )
        let pcapHandler = NIOWritePCAPHandler(
            mode: .client,
            fileSink: pcapRingBuffer.addFragment
        )
        try channel.pipeline.syncOperations.addHandler(pcapHandler, position: .first)

        for _ in 0..<self.numberOfRepeats {
            channel.writeAndFlush(self.byteBuffer, promise: nil)
            _ = try channel.readOutbound(as: ByteBuffer.self)
        }
        return self.numberOfRepeats
    }
}