未验证 提交 d2a958f4 编写于 作者: M Mazdak Farrokhzad 提交者: GitHub

Rollup merge of #59291 - SimonSapin:nonzero-thread-id, r=alexcrichton

Make Option<ThreadId> no larger than ThreadId, with NonZeroU64
......@@ -163,6 +163,7 @@
use crate::fmt;
use crate::io;
use crate::mem;
use crate::num::NonZeroU64;
use crate::panic;
use crate::panicking;
use crate::str;
......@@ -1036,7 +1037,7 @@ pub fn park_timeout(dur: Duration) {
/// [`Thread`]: ../../std/thread/struct.Thread.html
#[stable(feature = "thread_id", since = "1.19.0")]
#[derive(Eq, PartialEq, Clone, Copy, Hash, Debug)]
pub struct ThreadId(u64);
pub struct ThreadId(NonZeroU64);
impl ThreadId {
// Generate a new unique thread ID.
......@@ -1044,7 +1045,7 @@ fn new() -> ThreadId {
// We never call `GUARD.init()`, so it is UB to attempt to
// acquire this mutex reentrantly!
static GUARD: mutex::Mutex = mutex::Mutex::new();
static mut COUNTER: u64 = 0;
static mut COUNTER: u64 = 1;
unsafe {
let _guard = GUARD.lock();
......@@ -1058,7 +1059,7 @@ fn new() -> ThreadId {
let id = COUNTER;
COUNTER += 1;
ThreadId(id)
ThreadId(NonZeroU64::new(id).unwrap())
}
}
}
......@@ -1484,9 +1485,10 @@ fn _assert_both<T: Send + Sync>() {}
mod tests {
use super::Builder;
use crate::any::Any;
use crate::mem;
use crate::sync::mpsc::{channel, Sender};
use crate::result;
use crate::thread;
use crate::thread::{self, ThreadId};
use crate::time::Duration;
use crate::u32;
......@@ -1716,6 +1718,11 @@ fn sleep_ms_smoke() {
thread::sleep(Duration::from_millis(2));
}
#[test]
fn test_size_of_option_thread_id() {
assert_eq!(mem::size_of::<Option<ThreadId>>(), mem::size_of::<ThreadId>());
}
#[test]
fn test_thread_id_equal() {
assert!(thread::current().id() == thread::current().id());
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册