File: coalesced_tasks.h

package info (click to toggle)
chromium 139.0.7258.127-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 6,122,156 kB
  • sloc: cpp: 35,100,771; ansic: 7,163,530; javascript: 4,103,002; python: 1,436,920; asm: 946,517; xml: 746,709; pascal: 187,653; perl: 88,691; sh: 88,436; objc: 79,953; sql: 51,488; cs: 44,583; fortran: 24,137; makefile: 22,147; tcl: 15,277; php: 13,980; yacc: 8,984; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (84 lines) | stat: -rw-r--r-- 3,325 bytes parent folder | download | duplicates (5)
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
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef THIRD_PARTY_WEBRTC_OVERRIDES_COALESCED_TASKS_H_
#define THIRD_PARTY_WEBRTC_OVERRIDES_COALESCED_TASKS_H_

#include <cstdint>
#include <map>
#include <optional>
#include <set>

#include "base/functional/callback.h"
#include "base/synchronization/lock.h"
#include "base/thread_annotations.h"
#include "base/time/time.h"
#include "third_party/abseil-cpp/absl/functional/any_invocable.h"
#include "third_party/webrtc/rtc_base/system/rtc_export.h"

namespace blink {

// A thread-safe class for storing queued tasks until they are scheduled to run.
// This is useful for implementing metronome-like task queues where tasks are
// coalesced onto metronome ticks but the tasks still need to execute in order.
class RTC_EXPORT CoalescedTasks {
 public:
  typedef base::RepeatingCallback<std::optional<base::TimeTicks>()>
      PrepareRunTaskCallback;
  typedef base::RepeatingCallback<void(std::optional<base::TimeTicks>)>
      FinalizeRunTaskCallback;

  ~CoalescedTasks();

  // Queue a delayed tasks that can later be run with RunScheduledTasks().
  // `task_time` is the original run time of `task` and `scheduled_time` is the
  // `task_time` but snapped to the next time that scheduled tasks may run.
  // If true is returned, this is the first task that has been queued to run on
  // `scheduled_time`. In this case, the caller is responsible for scheduling a
  // call to RunScheduledTasks() at `scheduled_time`.
  bool QueueDelayedTask(base::TimeTicks task_time,
                        absl::AnyInvocable<void() &&> task,
                        base::TimeTicks scheduled_time);
  // Run all queued tasks up to and including `scheduled_time`. If multiple
  // tasks were queued onto the same `scheduled_time` they will execute in order
  // of their `task_time`. Optionally, the prepare/finalize callbacks can be
  // used to sample task run times by being called before/after each task.
  void RunScheduledTasks(base::TimeTicks scheduled_time,
                         PrepareRunTaskCallback prepare_run_task_callback =
                             PrepareRunTaskCallback(),
                         FinalizeRunTaskCallback finalize_run_task_callback =
                             FinalizeRunTaskCallback());
  // Clear the queue, deleting all tasks on the calling sequence without running
  // them.
  void Clear();

  // Returns true if there are no stored tasks.
  bool Empty() const {
    base::AutoLock lock(lock_);
    return delayed_tasks_.empty();
  }

 private:
  // The (time_ticks, unique_id) pair allows multiple tasks to be scheduled on
  // the same `time_ticks`.
  struct RTC_EXPORT UniqueTimeTicks {
    UniqueTimeTicks(base::TimeTicks time_ticks, uint64_t unique_id);

    // Used for std::map<> ordering.
    bool operator<(const UniqueTimeTicks& other) const;

    base::TimeTicks time_ticks;
    uint64_t unique_id;
  };

  mutable base::Lock lock_;
  std::set<base::TimeTicks> scheduled_ticks_ GUARDED_BY(lock_);
  uint64_t next_unique_id_ GUARDED_BY(lock_) = 0;
  std::map<UniqueTimeTicks, absl::AnyInvocable<void() &&>> delayed_tasks_
      GUARDED_BY(lock_);
};

}  // namespace blink

#endif  // THIRD_PARTY_WEBRTC_OVERRIDES_COALESCED_TASKS_H_