1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
#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
void
Init_lock_native_thread(void)
{
// do nothing
}
#endif // HAVE_PTHREAD_H
|