mirror of
https://github.com/ruby/ruby.git
synced 2025-08-15 13:39:04 +02:00
d578684989
Introduce `rb_thread_lock_native_thread()` to allocate dedicated native thread to the current Ruby thread for M:N threads. This C API is similar to Go's `runtime.LockOSThread()`. Accepted at https://github.com/ruby/dev-meeting-log/blob/master/2023/DevMeeting-2023-08-24.md (and missed to implement on Ruby 3.3.0)
49 lines
935 B
C
49 lines
935 B
C
|
|
#include "ruby/ruby.h"
|
|
#include "ruby/thread.h"
|
|
|
|
#ifdef HAVE_PTHREAD_H
|
|
#include <pthread.h>
|
|
|
|
static pthread_key_t tls_key;
|
|
|
|
static VALUE
|
|
get_tls(VALUE self)
|
|
{
|
|
return (VALUE)pthread_getspecific(tls_key);
|
|
}
|
|
|
|
static VALUE
|
|
set_tls(VALUE self, VALUE vn)
|
|
{
|
|
pthread_setspecific(tls_key, (void *)vn);
|
|
return Qnil;
|
|
}
|
|
|
|
static VALUE
|
|
lock_native_thread(VALUE self)
|
|
{
|
|
return rb_thread_lock_native_thread() ? Qtrue : Qfalse;
|
|
}
|
|
|
|
void
|
|
Init_lock_native_thread(void)
|
|
{
|
|
int r;
|
|
|
|
if ((r = pthread_key_create(&tls_key, NULL)) != 0) {
|
|
rb_bug("pthread_key_create() returns %d", r);
|
|
}
|
|
pthread_setspecific(tls_key, NULL);
|
|
|
|
rb_define_method(rb_cThread, "lock_native_thread", lock_native_thread, 0);
|
|
rb_define_method(rb_cThread, "get_tls", get_tls, 0);
|
|
rb_define_method(rb_cThread, "set_tls", set_tls, 1);
|
|
}
|
|
|
|
#else // HAVE_PTHREAD_H
|
|
Init_lock_native_thread(void)
|
|
{
|
|
// do nothing
|
|
}
|
|
#endif // HAVE_PTHREAD_H
|