Skip to content

Commit

Permalink
Use UnsafeCell for fast constant thread locals
Browse files Browse the repository at this point in the history
  • Loading branch information
Zoxc committed Mar 16, 2024
1 parent c5b5713 commit 93562dc
Showing 1 changed file with 7 additions and 5 deletions.
12 changes: 7 additions & 5 deletions library/std/src/sys/thread_local/fast_local.rs
Original file line number Diff line number Diff line change
Expand Up @@ -22,13 +22,15 @@ pub macro thread_local_inner {
const INIT_EXPR: $t = $init;
// If the platform has support for `#[thread_local]`, use it.
#[thread_local]
static mut VAL: $t = INIT_EXPR;
// We use `UnsafeCell` here instead of `static mut` to ensure any generated TLS shims
// have a nonnull attribute on their return value.
static VAL: $crate::cell::UnsafeCell<$t> = $crate::cell::UnsafeCell::new(INIT_EXPR);

// If a dtor isn't needed we can do something "very raw" and
// just get going.
if !$crate::mem::needs_drop::<$t>() {
unsafe {
return $crate::option::Option::Some(&VAL)
return $crate::option::Option::Some(&*VAL.get())
}
}

Expand All @@ -55,15 +57,15 @@ pub macro thread_local_inner {
// so now.
0 => {
$crate::thread::local_impl::Key::<$t>::register_dtor(
$crate::ptr::addr_of_mut!(VAL) as *mut $crate::primitive::u8,
VAL.get() as *mut $crate::primitive::u8,
destroy,
);
STATE.set(1);
$crate::option::Option::Some(&VAL)
$crate::option::Option::Some(&*VAL.get())
}
// 1 == the destructor is registered and the value
// is valid, so return the pointer.
1 => $crate::option::Option::Some(&VAL),
1 => $crate::option::Option::Some(&*VAL.get()),
// otherwise the destructor has already run, so we
// can't give access.
_ => $crate::option::Option::None,
Expand Down

0 comments on commit 93562dc

Please sign in to comment.