-
Notifications
You must be signed in to change notification settings - Fork 4
Implementation of threadsafe functions #220
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: main
Are you sure you want to change the base?
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Pull Request Overview
This pull request implements complete thread-safe function support for Node-API, enabling cross-thread communication between native code and JavaScript. The implementation provides all core Node-API threadsafe functions with proper lifecycle management, queueing semantics, and reference counting.
Key changes:
- Added ThreadSafeFunction class with full Node-API compatibility including creation, calling, acquire/release, and ref/unref operations
- Implemented all seven Node-API threadsafe functions in RuntimeNodeApiAsync.cpp with proper error handling and state management
- Added comprehensive test suite based on Node.js official tests to validate functionality across different threading scenarios
Reviewed Changes
Copilot reviewed 12 out of 12 changed files in this pull request and generated 4 comments.
Show a summary per file
File | Description |
---|---|
packages/host/cpp/ThreadsafeFunction.hpp | Header defining the ThreadSafeFunction class with thread-safe queue management and lifecycle controls |
packages/host/cpp/ThreadsafeFunction.cpp | Complete implementation of thread-safe function with global registry, queueing, and finalization logic |
packages/host/cpp/RuntimeNodeApiAsync.cpp | Node-API function implementations delegating to ThreadSafeFunction class methods |
packages/host/scripts/generate-weak-node-api-injector.ts | Added threadsafe function names to implemented functions list |
packages/node-addon-examples/tests/threadsafe_function/* | Test suite files including C++ addon and JavaScript test harness |
packages/host/android/CMakeLists.txt | Build configuration updates to include new ThreadsafeFunction source files |
Tip: Customize your code reviews with copilot-instructions.md. Create the file or learn how to get started.
// Global registry to map unique IDs to ThreadSafeFunction instances. | ||
// We use IDs instead of raw pointers to avoid any use-after-free issues. | ||
static std::unordered_map<std::uintptr_t, | ||
std::shared_ptr<callstack::nodeapihost::ThreadSafeFunction>> |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The static global registry introduces potential memory management complexity. Consider using a singleton pattern or RAII wrapper to ensure proper cleanup on module unload.
Copilot uses AI. Check for mistakes.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Cleanup is ensured programatically.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Could / should this map be owned by the napi_env
?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
My understanding is that it would change anything:
- Functions are called and referenced using handle
napi_threadsafe_function
, not usingnapi_env
. Threadsafe function will always use the correct env, the one the function was created for. - I am not aware of any way to detect and cleanup the map when
napi_env
is destroyed - no change in this aspect. - There will be no memory leaks since smart pointers are used anyway.
I just cannot see benefits from adding this "ownership" layer, but I may be missing something. Could you see any?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
that it would change anything
That it would or that it would not?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Should be would not
packages/node-addon-examples/tests/threadsafe_function/CMakeLists.txt
Outdated
Show resolved
Hide resolved
job->state = AsyncJob::State::Cancelled; | ||
return napi_ok; | ||
} | ||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I wonder if the name of this file needs an update now that it's not "just async work related"? Or of this should go into as separate file? 🤔
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I was also thinking about that, finally I decided to put it there since the threadsafe functions in Node-API are called asynchronously and other implementation often group them together. But I am open to change that I move it to the separate file.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I don't have a strong preference either way 👍
If it's a conscious choice, I'm all for that. Just didn't seem that related to me.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just did a pair-programming session with Kræn and came up with one important change for thread-hopping!
// Hop to JS thread; we drain one item per hop to keep latency predictable | ||
// and avoid long monopolization of the JS queue. | ||
invoker->invokeAsync([self = shared_from_this()] { self->processQueue(); }); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This one-line change was enough to get thread-hopping in NativeScript Node-API working (where before, it was crashing):
// Hop to JS thread; we drain one item per hop to keep latency predictable | |
// and avoid long monopolization of the JS queue. | |
invoker->invokeAsync([self = shared_from_this()] { self->processQueue(); }); | |
// Invoke from the current thread. Libraries like NativeScript can wrap a | |
// JS function in an Objective-C block to be dispatched onto another thread | |
// (e.g. the main thread, with the intention of accessing the UI), and so we | |
// should run the JS function on the same thread the Objective-C block was | |
// dispatched to. | |
invoker->invokeSync([self = shared_from_this()] { self->processQueue(); }); |
I was able to do intricate moves back and forth between the JS and main threads, and use JS timers (see screenshot below). It all worked perfectly.

There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just came back from the long vacation and saw that, sweet!
I tried that with my reproduction of the problem but it didn't help - looks like my repro conditions were not correct.
I have applied the changes 😊
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Unfortunately this change introduced problems to the solution, such as deadlocks, some of them are already fixed, but I will have to revisit the entire PR in the future.
3239f2b
to
b9ebf1f
Compare
This pull request introduces a complete implementation of Node-API's thread-safe function support for the host runtime, including its registration, lifecycle management, and queueing semantics.
Thread-safe function implementation:
napi_create_threadsafe_function
,napi_call_threadsafe_function
,napi_acquire_threadsafe_function
,napi_release_threadsafe_function
,napi_unref_threadsafe_function
,napi_ref_threadsafe_function
,napi_get_threadsafe_function_context
) inRuntimeNodeApiAsync.cpp
ThreadSafeFunction
class , providing a full implementation of Node-API's thread-safe function, including queueing, reference management, context retrieval, and finalization logic. This includes a global registry for safe handle management.Testing and examples:
Note
Implements Node-API thread-safe functions with a new
ThreadSafeFunction
core, wires them into the runtime/injector, and adds test coverage with example integration.ThreadSafeFunction
implementation (packages/host/cpp/ThreadsafeFunction.{hpp,cpp}
) with queueing, ref/release, finalize, and context retrieval.napi_*
TSFN APIs inRuntimeNodeApiAsync.{hpp,cpp}
:napi_create_threadsafe_function
,napi_get_threadsafe_function_context
,napi_call_threadsafe_function
,napi_acquire_threadsafe_function
,napi_release_threadsafe_function
,napi_unref_threadsafe_function
,napi_ref_threadsafe_function
.packages/host/android/CMakeLists.txt
).packages/host/scripts/generate-weak-node-api-injector.ts
).packages/node-addon-examples/tests/threadsafe_function/*
) and wires into examples index.apps/test-app/App.tsx
).react-native-node-api
.Written by Cursor Bugbot for commit 87a78f2. This will update automatically on new commits. Configure here.