Add calls to flush_ui_queue for Hermes (#4579)

Rearranged jsi_init
realm · Jun 8, 2022 · 0c4dff1 · 0c4dff1
1 parent 62a5b21
commit 0c4dff1
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 11 deletions.
diff --git a/src/jsi/jsi_function.hpp b/src/jsi/jsi_function.hpp
@@ -26,33 +26,43 @@ namespace js {
 template <>
 inline JsiVal realmjsi::Function::call(JsiEnv env, const JsiFunc& function, size_t argc, const JsiVal arguments[])
 {
-    return env(function->call(env, env.args(arguments, argc), argc));
+    auto result = env(function->call(env, env.args(arguments, argc), argc));
+    flush_ui_queue();
+    return result;
 }
 
 template <>
 inline JsiVal realmjsi::Function::call(JsiEnv env, const JsiFunc& function, const JsiObj& this_object, size_t argc,
                                        const JsiVal arguments[])
 {
-    return env(function->callWithThis(env, this_object, env.args(arguments, argc), argc));
+    auto result = env(function->callWithThis(env, this_object, env.args(arguments, argc), argc));
+    flush_ui_queue();
+    return result;
 }
 
 template <>
 inline JsiVal realmjsi::Function::callback(JsiEnv env, const JsiFunc& function, size_t argc, const JsiVal arguments[])
 {
-    return env(function->call(env, env.args(arguments, argc), argc));
+    auto result = env(function->call(env, env.args(arguments, argc), argc));
+    flush_ui_queue();
+    return result;
 }
 template <>
 inline JsiVal realmjsi::Function::callback(JsiEnv env, const JsiFunc& function, const JsiObj& this_object,
                                            size_t argc, const JsiVal arguments[])
 {
-    return env(function->callWithThis(env, this_object, env.args(arguments, argc), argc));
+    auto result = env(function->callWithThis(env, this_object, env.args(arguments, argc), argc));
+    flush_ui_queue();
+    return result;
 }
 
 template <>
 inline JsiObj realmjsi::Function::construct(JsiEnv env, const JsiFunc& function, size_t argc,
                                             const JsiVal arguments[])
 {
-    return env(function->callAsConstructor(env, env.args(arguments, argc), argc).asObject(env));
+    auto result = env(function->callAsConstructor(env, env.args(arguments, argc), argc).asObject(env));
+    flush_ui_queue();
+    return result;
 }
 
 } // namespace js

diff --git a/src/jsi/jsi_init.cpp b/src/jsi/jsi_init.cpp
@@ -19,12 +19,6 @@
 
 #include "jsi_init.hpp"
 
-namespace realm {
-namespace js {
-std::function<void()> flush_ui_queue;
-} // namespace js
-} // namespace realm
-
 #if !REALM_ENABLE_SYNC
 #pragma comment(lib, "ws2_32.lib")
 #pragma comment(lib, "crypt32")

diff --git a/src/jsi/jsi_init.hpp b/src/jsi/jsi_init.hpp
@@ -18,6 +18,14 @@
 
 #pragma once
 
+#include <functional>
+
+namespace realm {
+namespace js {
+std::function<void()> flush_ui_queue;
+} // namespace js
+} // namespace realm
+
 #include "jsi_string.hpp"
 #include "jsi_protected.hpp"
 #include "jsi_function.hpp"