feat(dom): set dom thread priority

Tencent · Sep 21, 2023 · 2fbb558 · 2fbb558
1 parent 1e22e89
commit 2fbb558
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 1 deletion.
diff --git a/framework/android/connector/dom/src/main/cpp/src/dom_jni.cc b/framework/android/connector/dom/src/main/cpp/src/dom_jni.cc
@@ -31,6 +31,8 @@
 #include "footstone/worker_impl.h"
 #include "jni/jni_register.h"
 #include "jni/data_holder.h"
+#include "jni/jni_env.h"
+#include "jni/scoped_java_ref.h"
 
 namespace hippy {
 inline namespace framework {
@@ -130,11 +132,32 @@ void SetDomManager(JNIEnv* j_env,
   root_node->SetDomManager(dom_manager_object);
 }
 
-jint CreateDomManager(__unused JNIEnv* j_env, __unused jobject j_obj) {
+static void SetThreadPriority(jobject j_object) {
+  auto j_env = JNIEnvironment::GetInstance()->AttachCurrentThread();
+  auto j_class = j_env->GetObjectClass(j_object);
+  if (!j_class) {
+    FOOTSTONE_LOG(ERROR) << "CallJavaMethod j_class error";
+    return;
+  }
+  auto j_cb_id = j_env->GetMethodID(j_class, "setThreadPrority", "()V");
+  if (!j_cb_id) {
+    FOOTSTONE_LOG(ERROR) << "CallJavaMethod j_cb_id error";
+    return;
+  }
+  j_env->CallVoidMethod(j_object, j_cb_id);
+  JNIEnvironment::ClearJEnvException(j_env);
+  j_env->DeleteLocalRef(j_class);
+}
+
+jint CreateDomManager(JNIEnv* j_env, jobject j_obj) {
   auto dom_manager = std::make_shared<DomManager>();
   auto dom_id = hippy::global_data_holder_key.fetch_add(1);
   hippy::global_data_holder.Insert(dom_id, dom_manager);
   auto worker = std::make_shared<WorkerImpl>(kDomWorkerName, false);
+  auto callback = std::make_shared<JavaRef>(j_env, j_obj);
+  worker->BeforeStart([callback]() {
+    if (callback->GetObj()) SetThreadPriority(callback->GetObj());
+  });
   worker->Start();
   auto runner = std::make_shared<TaskRunner>(kDomRunnerName);
   runner->SetWorker(worker);

diff --git a/framework/android/connector/dom/src/main/java/com/openhippy/connector/DomManager.java b/framework/android/connector/dom/src/main/java/com/openhippy/connector/DomManager.java
@@ -18,6 +18,7 @@
 
 import android.view.View;
 import androidx.annotation.NonNull;
+import android.os.Process;
 
 @SuppressWarnings("JavaJniMissingFunction")
 public class DomManager implements Connector {
@@ -56,6 +57,11 @@ public void attachToRoot(View root) {
         setDomManager(root.getId(), mInstanceId);
     }
 
+    public void setThreadPrority() {
+        int tid = Process.myTid();
+        Process.setThreadPriority(tid, Thread.MAX_PRIORITY);
+    }
+
     /**
      * Attach to renderer with specified id.
      *

diff --git a/modules/footstone/include/footstone/worker.h b/modules/footstone/include/footstone/worker.h
@@ -77,6 +77,7 @@ class Worker {
   }
   static bool IsTaskRunning();
   bool RunTask();
+  void BeforeStart(std::function<void()> before_start) { before_start_ = before_start; }
   void Start(bool in_new_thread = true);
 
   virtual void SetName(const std::string& name) = 0;
@@ -108,6 +109,7 @@ class Worker {
   void UpdateSpecific(uint32_t task_runner_id, std::array<void *, kWorkerKeysMax> array);
 
   std::thread thread_;
+  std::function<void()> before_start_;
   std::string name_;
   std::list<std::vector<std::shared_ptr<TaskRunner>>> running_group_list_;
   std::list<std::vector<std::shared_ptr<TaskRunner>>> pending_group_list_;

diff --git a/modules/footstone/src/worker.cc b/modules/footstone/src/worker.cc
@@ -128,6 +128,7 @@ void Worker::Start(bool in_new_thread) {
   if (in_new_thread) {
     thread_ = std::thread([this]() -> void {
       SetName(name_);
+      if (before_start_) before_start_();
       driver_->Start();
     });
   } else {