Add calls to flush_ui_queue for Hermes (#4579)

Rearranged jsi_init Refactored flush of UI queue for JSI
realm · Jun 9, 2022 · a781ddf · a781ddf
1 parent ea5d5ce
commit a781ddf
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 11 deletions.
diff --git a/src/jsi/jsi_function.hpp b/src/jsi/jsi_function.hpp
@@ -23,36 +23,43 @@
 namespace realm {
 namespace js {
 
+template <typename T>
+inline T flush_and_return(T&& val)
+{
+    flush_ui_queue();
+    return val;
+}
+
 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));
+    return flush_and_return(env(function->call(env, env.args(arguments, argc), argc)));
 }
 
 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));
+    return flush_and_return(env(function->callWithThis(env, this_object, env.args(arguments, argc), argc)));
 }
 
 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));
+    return flush_and_return(env(function->call(env, env.args(arguments, argc), argc)));
 }
 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));
+    return flush_and_return(env(function->callWithThis(env, this_object, env.args(arguments, argc), argc)));
 }
 
 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));
+    return flush_and_return(env(function->callAsConstructor(env, env.args(arguments, argc), argc).asObject(env)));
 }
 
 } // 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"