1*6dbdd20aSAndroid Build Coastguard Worker /*
2*6dbdd20aSAndroid Build Coastguard Worker * Copyright (C) 2024 The Android Open Source Project
3*6dbdd20aSAndroid Build Coastguard Worker *
4*6dbdd20aSAndroid Build Coastguard Worker * Licensed under the Apache License, Version 2.0 (the "License");
5*6dbdd20aSAndroid Build Coastguard Worker * you may not use this file except in compliance with the License.
6*6dbdd20aSAndroid Build Coastguard Worker * You may obtain a copy of the License at
7*6dbdd20aSAndroid Build Coastguard Worker *
8*6dbdd20aSAndroid Build Coastguard Worker * http://www.apache.org/licenses/LICENSE-2.0
9*6dbdd20aSAndroid Build Coastguard Worker *
10*6dbdd20aSAndroid Build Coastguard Worker * Unless required by applicable law or agreed to in writing, software
11*6dbdd20aSAndroid Build Coastguard Worker * distributed under the License is distributed on an "AS IS" BASIS,
12*6dbdd20aSAndroid Build Coastguard Worker * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*6dbdd20aSAndroid Build Coastguard Worker * See the License for the specific language governing permissions and
14*6dbdd20aSAndroid Build Coastguard Worker * limitations under the License.
15*6dbdd20aSAndroid Build Coastguard Worker */
16*6dbdd20aSAndroid Build Coastguard Worker
17*6dbdd20aSAndroid Build Coastguard Worker #include "perfetto/ext/base/weak_runner.h"
18*6dbdd20aSAndroid Build Coastguard Worker
19*6dbdd20aSAndroid Build Coastguard Worker #include "perfetto/base/task_runner.h"
20*6dbdd20aSAndroid Build Coastguard Worker
21*6dbdd20aSAndroid Build Coastguard Worker namespace perfetto::base {
22*6dbdd20aSAndroid Build Coastguard Worker
WeakRunner(base::TaskRunner * task_runner)23*6dbdd20aSAndroid Build Coastguard Worker WeakRunner::WeakRunner(base::TaskRunner* task_runner)
24*6dbdd20aSAndroid Build Coastguard Worker : task_runner_(task_runner), destroyed_(std::make_shared<bool>(false)) {}
25*6dbdd20aSAndroid Build Coastguard Worker
~WeakRunner()26*6dbdd20aSAndroid Build Coastguard Worker WeakRunner::~WeakRunner() {
27*6dbdd20aSAndroid Build Coastguard Worker *destroyed_ = true;
28*6dbdd20aSAndroid Build Coastguard Worker }
29*6dbdd20aSAndroid Build Coastguard Worker
PostTask(std::function<void ()> f) const30*6dbdd20aSAndroid Build Coastguard Worker void WeakRunner::PostTask(std::function<void()> f) const {
31*6dbdd20aSAndroid Build Coastguard Worker task_runner_->PostTask([destroyed = destroyed_, f = std::move(f)]() {
32*6dbdd20aSAndroid Build Coastguard Worker if (*destroyed) {
33*6dbdd20aSAndroid Build Coastguard Worker return;
34*6dbdd20aSAndroid Build Coastguard Worker }
35*6dbdd20aSAndroid Build Coastguard Worker f();
36*6dbdd20aSAndroid Build Coastguard Worker });
37*6dbdd20aSAndroid Build Coastguard Worker }
38*6dbdd20aSAndroid Build Coastguard Worker
PostDelayedTask(std::function<void ()> f,uint32_t delay_ms) const39*6dbdd20aSAndroid Build Coastguard Worker void WeakRunner::PostDelayedTask(std::function<void()> f,
40*6dbdd20aSAndroid Build Coastguard Worker uint32_t delay_ms) const {
41*6dbdd20aSAndroid Build Coastguard Worker task_runner_->PostDelayedTask(
42*6dbdd20aSAndroid Build Coastguard Worker [destroyed = destroyed_, f = std::move(f)]() {
43*6dbdd20aSAndroid Build Coastguard Worker if (*destroyed) {
44*6dbdd20aSAndroid Build Coastguard Worker return;
45*6dbdd20aSAndroid Build Coastguard Worker }
46*6dbdd20aSAndroid Build Coastguard Worker f();
47*6dbdd20aSAndroid Build Coastguard Worker },
48*6dbdd20aSAndroid Build Coastguard Worker delay_ms);
49*6dbdd20aSAndroid Build Coastguard Worker }
50*6dbdd20aSAndroid Build Coastguard Worker
51*6dbdd20aSAndroid Build Coastguard Worker } // namespace perfetto::base
52