-
Notifications
You must be signed in to change notification settings - Fork 625
/
NIOAsyncSequenceProducerStrategies.swift
47 lines (43 loc) · 2.06 KB
/
NIOAsyncSequenceProducerStrategies.swift
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
//===----------------------------------------------------------------------===//
//
// This source file is part of the SwiftNIO open source project
//
// Copyright (c) 2022 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
//
//===----------------------------------------------------------------------===//
public enum NIOAsyncSequenceProducerBackPressureStrategies {
/// A high-low watermarked back-pressure strategy for a ``NIOAsyncSequenceProducer``.
///
/// This strategy does the following:
/// - On yield it keeps on demanding more elements from the producer as long as the number of buffered elements hasn't reached the `highWatermark`.
/// - On consume it starts to demand again from the producer once the number of buffered elements reach the `lowWatermark`.
@available(macOS 10.15, iOS 13, tvOS 13, watchOS 6, *)
public struct HighLowWatermark: NIOAsyncSequenceProducerBackPressureStrategy {
private let lowWatermark: Int
private let highWatermark: Int
/// Initializes a new ``NIOAsyncSequenceProducerBackPressureStrategies/HighLowWatermark``.
///
/// - Parameters:
/// - lowWatermark: The low watermark where demand should start.
/// - highWatermark: The high watermark where demand should be stopped.
public init(lowWatermark: Int, highWatermark: Int) {
precondition(lowWatermark <= highWatermark)
self.lowWatermark = lowWatermark
self.highWatermark = highWatermark
}
public mutating func didYield(bufferDepth: Int) -> Bool {
// We are demanding more until we reach the high watermark
bufferDepth < self.highWatermark
}
public mutating func didConsume(bufferDepth: Int) -> Bool {
// We start demanding again once we are below the low watermark
bufferDepth < self.lowWatermark
}
}
}