File: Debouncer.swift

package info (click to toggle)
swiftlang 6.0.3-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 2,519,992 kB
  • sloc: cpp: 9,107,863; ansic: 2,040,022; asm: 1,135,751; python: 296,500; objc: 82,456; f90: 60,502; lisp: 34,951; pascal: 19,946; sh: 18,133; perl: 7,482; ml: 4,937; javascript: 4,117; makefile: 3,840; awk: 3,535; xml: 914; fortran: 619; cs: 573; ruby: 573
file content (84 lines) | stat: -rw-r--r-- 3,544 bytes parent folder | download
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 Swift.org open source project
//
// Copyright (c) 2014 - 2024 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

/// Debounces calls to a function/closure. If multiple calls to the closure are made, it allows aggregating the
/// parameters.
public actor Debouncer<Parameter> {
  /// How long to wait for further `scheduleCall` calls before committing to actually calling `makeCall`.
  private let debounceDuration: Duration

  /// When `scheduleCall` is called while another `scheduleCall` was waiting to commit its call, combines the parameters
  /// of those two calls.
  ///
  /// ### Example
  ///
  /// Two `scheduleCall` calls that are made within a time period shorter than `debounceDuration` like the following
  /// ```swift
  /// debouncer.scheduleCall(5)
  /// debouncer.scheduleCall(10)
  /// ```
  /// will call `combineParameters(5, 10)`
  private let combineParameters: (Parameter, Parameter) -> Parameter

  /// After the debounce duration has elapsed, commit the call.
  private let makeCall: (Parameter) async -> Void

  /// In the time between the call to `scheduleCall` and the call actually being committed (ie. in the time that the
  /// call can be debounced), the task that would commit the call (unless cancelled), the parameter with which this
  /// call should be made and the time at which the call should be made. Keeping track of the time ensures that we don't
  /// indefinitely debounce if a new `scheduleCall` is made every 0.4s but we debounce for 0.5s.
  private var inProgressData: (Parameter, ContinuousClock.Instant, Task<Void, Never>)?

  public init(
    debounceDuration: Duration,
    combineResults: @escaping (Parameter, Parameter) -> Parameter,
    _ makeCall: @Sendable @escaping (Parameter) async -> Void
  ) {
    self.debounceDuration = debounceDuration
    self.combineParameters = combineResults
    self.makeCall = makeCall
  }

  /// Schedule a debounced call. If `scheduleCall` is called within `debounceDuration`, the parameters of the two
  /// `scheduleCall` calls will be combined using `combineParameters` and the new debounced call will be scheduled
  /// `debounceDuration` after the second `scheduleCall` call.
  public func scheduleCall(_ parameter: Parameter) {
    var parameter = parameter
    var targetDate = ContinuousClock.now + debounceDuration
    if let (inProgressParameter, inProgressTargetDate, inProgressTask) = inProgressData {
      inProgressTask.cancel()
      parameter = combineParameters(inProgressParameter, parameter)
      targetDate = inProgressTargetDate
    }
    let task = Task {
      do {
        try await Task.sleep(until: targetDate)
        try Task.checkCancellation()
      } catch {
        return
      }
      inProgressData = nil
      await makeCall(parameter)
    }
    inProgressData = (parameter, ContinuousClock.now + debounceDuration, task)
  }
}

extension Debouncer<Void> {
  public init(debounceDuration: Duration, _ makeCall: @Sendable @escaping () async -> Void) {
    self.init(debounceDuration: debounceDuration, combineResults: { _, _ in }, makeCall)
  }

  public func scheduleCall() {
    self.scheduleCall(())
  }
}