xref: /aosp_15_r20/external/cronet/base/android/task_scheduler/task_runner_android.h (revision 6777b5387eb2ff775bb5750e3f5d96f37fb7352b)
1 // Copyright 2018 The Chromium Authors
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4 
5 #ifndef BASE_ANDROID_TASK_SCHEDULER_TASK_RUNNER_ANDROID_H_
6 #define BASE_ANDROID_TASK_SCHEDULER_TASK_RUNNER_ANDROID_H_
7 
8 #include <memory>
9 
10 #include "base/android/jni_weak_ref.h"
11 #include "base/android/task_scheduler/task_traits_android.h"
12 #include "base/base_export.h"
13 #include "base/functional/callback_forward.h"
14 #include "base/task/single_thread_task_runner.h"
15 
16 namespace base {
17 
18 // GENERATED_JAVA_ENUM_PACKAGE: org.chromium.base.task
19 enum class TaskRunnerType { BASE, SEQUENCED, SINGLE_THREAD };
20 
21 // Native implementation backing TaskRunnerImpl.java which posts java tasks onto
22 // a C++ TaskRunner.
23 class BASE_EXPORT TaskRunnerAndroid {
24  public:
25   explicit TaskRunnerAndroid(scoped_refptr<TaskRunner> task_runner,
26                              TaskRunnerType type);
27 
28   TaskRunnerAndroid(const TaskRunnerAndroid&) = delete;
29   TaskRunnerAndroid& operator=(const TaskRunnerAndroid&) = delete;
30 
31   ~TaskRunnerAndroid();
32 
33   void Destroy(JNIEnv* env);
34 
35   void PostDelayedTask(JNIEnv* env,
36                        const base::android::JavaRef<jobject>& task,
37                        jlong delay,
38                        std::string& runnable_class_name);
39 
40   bool BelongsToCurrentThread(JNIEnv* env);
41 
42   static std::unique_ptr<TaskRunnerAndroid> Create(jint task_runner_type,
43                                                    jint j_task_traits);
44 
45   using UiThreadTaskRunnerCallback =
46       RepeatingCallback<scoped_refptr<base::SingleThreadTaskRunner>(
47           ::TaskTraits)>;
48 
49   static void SetUiThreadTaskRunnerCallback(
50       UiThreadTaskRunnerCallback callback);
51 
52  private:
53   const scoped_refptr<TaskRunner> task_runner_;
54   const TaskRunnerType type_;
55 };
56 
57 }  // namespace base
58 
59 #endif  // BASE_ANDROID_TASK_SCHEDULER_TASK_RUNNER_ANDROID_H_
60