components/util/tid.rs


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

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

use std::sync::atomic::{AtomicUint, ATOMIC_UINT_INIT, Ordering};
use std::rc::Rc;
use std::cell::RefCell;

static mut next_tid: AtomicUint = ATOMIC_UINT_INIT;

thread_local!(static TASK_LOCAL_TID: Rc<RefCell<Option<uint>>> = Rc::new(RefCell::new(None)));

/// Every task gets one, that's unique.
pub fn tid() -> uint {
    TASK_LOCAL_TID.with(|ref k| {
        let ret =
            match *k.borrow() {
                None => unsafe { next_tid.fetch_add(1, Ordering::SeqCst) },
                Some(x) => x,
            };

        *k.borrow_mut() = Some(ret);

        ret
    })
}