std: reduce code size of set_current

rust-lang · Apr 2, 2024 · b5ccebf · b5ccebf
1 parent e2cf2cb
commit b5ccebf
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/library/std/src/thread/mod.rs b/library/std/src/thread/mod.rs
@@ -684,9 +684,12 @@ thread_local! {
 
 /// Sets the thread handle for the current thread.
 ///
-/// Panics if the handle has been set already or when called from a TLS destructor.
+/// Aborts if the handle has been set already to reduce code size.
 pub(crate) fn set_current(thread: Thread) {
-    CURRENT.with(|current| current.set(thread).unwrap());
+    CURRENT.with(|current| {
+        rtassert!(current.get().is_none());
+        current.set(thread).unwrap();
+    });
 }
 
 /// Gets a handle to the thread that invokes it.