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
|
//
// RetryStrategy.swift
// MullvadREST
//
// Created by pronebird on 09/12/2021.
// Copyright © 2021 Mullvad VPN AB. All rights reserved.
//
import Foundation
import MullvadTypes
extension REST {
public struct RetryStrategy {
public var maxRetryCount: Int
public var delay: RetryDelay
public var applyJitter: Bool
public init(maxRetryCount: Int, delay: RetryDelay, applyJitter: Bool) {
self.maxRetryCount = maxRetryCount
self.delay = delay
self.applyJitter = applyJitter
}
public func makeDelayIterator() -> AnyIterator<Duration> {
let inner = delay.makeIterator()
if applyJitter {
return AnyIterator(Jittered(inner))
} else {
return AnyIterator(inner)
}
}
/// Strategy configured to never retry.
public static var noRetry = RetryStrategy(
maxRetryCount: 0,
delay: .never,
applyJitter: false
)
/// Strategy configured with 2 retry attempts and exponential backoff.
public static var `default` = RetryStrategy(
maxRetryCount: 2,
delay: defaultRetryDelay,
applyJitter: true
)
/// Strategy configured with 10 retry attempts and exponential backoff.
public static var aggressive = RetryStrategy(
maxRetryCount: 10,
delay: defaultRetryDelay,
applyJitter: true
)
/// Default retry delay.
public static var defaultRetryDelay: RetryDelay = .exponentialBackoff(
initial: .seconds(2),
multiplier: 2,
maxDelay: .seconds(8)
)
}
public enum RetryDelay: Equatable {
/// Never wait to retry.
case never
/// Constant delay.
case constant(Duration)
/// Exponential backoff.
case exponentialBackoff(initial: Duration, multiplier: UInt64, maxDelay: Duration?)
func makeIterator() -> AnyIterator<Duration> {
switch self {
case .never:
return AnyIterator {
nil
}
case let .constant(duration):
return AnyIterator {
duration
}
case let .exponentialBackoff(initial, multiplier, maxDelay):
return AnyIterator(ExponentialBackoff(
initial: initial,
multiplier: multiplier,
maxDelay: maxDelay
))
}
}
}
}
|
