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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
#include "opal/mca/threads/tsd.h"
static void _tracked_destructor(void *arg)
{
opal_tsd_list_item_t *tsd = NULL;
opal_tsd_tracked_key_t *key = NULL;
if (NULL == arg) {
return;
}
tsd = (opal_tsd_list_item_t *) arg;
key = tsd->tracked_key;
opal_mutex_lock(&key->mutex);
opal_list_remove_item(&key->tsd_list, &tsd->super);
opal_mutex_unlock(&key->mutex);
if (NULL != key->user_destructor) {
key->user_destructor(tsd->data);
}
OBJ_RELEASE(tsd);
}
void opal_tsd_tracked_key_constructor(opal_tsd_tracked_key_t *key)
{
OBJ_CONSTRUCT(&key->mutex, opal_mutex_t);
OBJ_CONSTRUCT(&key->tsd_list, opal_list_t);
key->user_destructor = NULL;
opal_tsd_key_create(&key->key, _tracked_destructor);
}
void opal_tsd_tracked_key_destructor(opal_tsd_tracked_key_t *key)
{
opal_tsd_list_item_t *tsd, *next;
opal_tsd_key_delete(key->key);
OPAL_LIST_FOREACH_SAFE (tsd, next, &key->tsd_list, opal_tsd_list_item_t) {
opal_list_remove_item(&key->tsd_list, &tsd->super);
if (NULL != key->user_destructor) {
key->user_destructor(tsd->data);
}
OBJ_RELEASE(tsd);
}
OBJ_DESTRUCT(&key->mutex);
OBJ_DESTRUCT(&key->tsd_list);
}
int opal_tsd_tracked_key_set(opal_tsd_tracked_key_t *key, void *p)
{
assert(NULL != key);
opal_tsd_list_item_t *tsd = NULL;
opal_tsd_get(key->key, (void **) &tsd);
if (NULL == tsd) {
tsd = OBJ_NEW(opal_tsd_list_item_t);
if (NULL == tsd) {
return OPAL_ERR_OUT_OF_RESOURCE;
}
opal_mutex_lock(&key->mutex);
opal_list_append(&key->tsd_list, &tsd->super);
opal_mutex_unlock(&key->mutex);
}
tsd->data = p;
tsd->tracked_key = key;
return opal_tsd_set(key->key, (void *) tsd);
}
void opal_tsd_tracked_key_set_destructor(opal_tsd_tracked_key_t *key,
opal_tsd_destructor_t destructor)
{
assert(NULL != key);
key->user_destructor = destructor;
}
OBJ_CLASS_INSTANCE(opal_tsd_list_item_t, opal_list_item_t, NULL, NULL);
OBJ_CLASS_INSTANCE(opal_tsd_tracked_key_t, opal_object_t, opal_tsd_tracked_key_constructor,
opal_tsd_tracked_key_destructor);
|
