Struct local_sync::OnceCell

source ·

pub struct OnceCell<T> { /* private fields */ }

Expand description

A thread-safe cell that can be written to only once.

A OnceCell is typically used for global variables that need to be initialized once on first use, but need no further changes. The OnceCell in Tokio allows the initialization procedure to be asynchronous.

Examples

use local_sync::OnceCell;

async fn some_computation() -> u32 {
    1 + 1
}

thread_local! {
    static ONCE: OnceCell<u32> = OnceCell::new();
}

#[monoio::main]
async fn main() {
    let once = ONCE.with(|once| unsafe {
        std::ptr::NonNull::new_unchecked(once as *const _ as *mut OnceCell<u32>).as_ref()
    });
    let result = once.get_or_init(some_computation).await;
    assert_eq!(*result, 2);
}

It is often useful to write a wrapper method for accessing the value.

use local_sync::OnceCell;

thread_local! {
    static ONCE: OnceCell<u32> = OnceCell::new();
}

async fn get_global_integer() -> &'static u32 {
    let once = ONCE.with(|once| unsafe {
        std::ptr::NonNull::new_unchecked(once as *const _ as *mut OnceCell<u32>).as_ref()
    });
    once.get_or_init(|| async {
        1 + 1
    }).await
}

#[monoio::main]
async fn main() {
    let result = get_global_integer().await;
    assert_eq!(*result, 2);
}

Implementations§

source §

impl<T> OnceCell<T>

source

pub const fn new() -> Self

Creates a new empty OnceCell instance.

source

pub fn new_with(value: Option<T>) -> Self

Creates a new OnceCell that contains the provided value, if any.

If the Option is None, this is equivalent to OnceCell::new.

source

pub fn initialized(&self) -> bool

Returns true if the OnceCell currently contains a value, and false otherwise.

source

pub fn get(&self) -> Option<&T>

Returns a reference to the value currently stored in the OnceCell, or None if the OnceCell is empty.

source

pub fn get_mut(&mut self) -> Option<&mut T>

Returns a mutable reference to the value currently stored in the OnceCell, or None if the OnceCell is empty.

Since this call borrows the OnceCell mutably, it is safe to mutate the value inside the OnceCell — the mutable borrow statically guarantees no other references exist.

source

pub fn set(&self, value: T) -> Result<(), SetError<T>>

Set the value of the OnceCell to the given value if the OnceCell is empty.

If the OnceCell already has a value, this call will fail with an SetError::AlreadyInitializedError.

If the OnceCell is empty, but some other task is currently trying to set the value, this call will fail with SetError::InitializingError.

source

pub async fn get_or_init<F, Fut>(&self, f: F) -> &Twhere F: FnOnce() -> Fut, Fut: Future<Output = T>,

Get the value currently in the OnceCell, or initialize it with the given asynchronous operation.

If some other task is currently working on initializing the OnceCell, this call will wait for that other task to finish, then return the value that the other task produced.

If the provided operation is cancelled or panics, the initialization attempt is cancelled. If there are other tasks waiting for the value to be initialized, one of them will start another attempt at initializing the value.

This will deadlock if f tries to initialize the cell recursively.

source

pub async fn get_or_try_init<E, F, Fut>(&self, f: F) -> Result<&T, E>where F: FnOnce() -> Fut, Fut: Future<Output = Result<T, E>>,

Get the value currently in the OnceCell, or initialize it with the given asynchronous operation.

If some other task is currently working on initializing the OnceCell, this call will wait for that other task to finish, then return the value that the other task produced.

If the provided operation returns an error, is cancelled or panics, the initialization attempt is cancelled. If there are other tasks waiting for the value to be initialized, one of them will start another attempt at initializing the value.

This will deadlock if f tries to initialize the cell recursively.

source