Rust:原子操作 AtomicBool

在 Rust 中,你可以使用 std::sync::atomic 模块来进行原子操作。原子操作在多线程环境中特别有用,因为它们可以确保操作的原子性和可见性,从而避免数据竞争和其他并发问题。

为了读取和设置布尔值,你可以使用 AtomicBool 类型。以下是一个简单的示例,展示了如何使用 AtomicBool 来读取和设置布尔值:

rust

use std::sync::atomic::{AtomicBool, Ordering};

fn main() {

// 创建一个新的 AtomicBool,初始值为 false

let atomic_bool = AtomicBool::new(false);

复制代码
// 设置 AtomicBool 的值为 true
atomic_bool.store(true, Ordering::Relaxed);

// 读取 AtomicBool 的值
let value = atomic_bool.load(Ordering::Relaxed);
println!("The value of atomic_bool is: {}", value);

// 使用 compare_and_swap 方法,如果当前值为 true,则将其设置为 false,并返回旧值
let old_value = atomic_bool.compare_and_swap(true, false, Ordering::Relaxed);
println!("The old value of atomic_bool was: {}", old_value);

// 再次读取值以确认更改
let new_value = atomic_bool.load(Ordering::Relaxed);
println!("The new value of atomic_bool is: {}", new_value);

// 使用 swap 方法,将值设置为 true 并返回旧值
let swapped_value = atomic_bool.swap(true, Ordering::Relaxed);
println!("The swapped value of atomic_bool was: {}", swapped_value);

}

解释

创建 AtomicBool:

rust

let atomic_bool = AtomicBool::new(false);

这创建了一个新的 AtomicBool 实例,初始值为 false。

设置值:

rust

atomic_bool.store(true, Ordering::Relaxed);

使用 store 方法将 AtomicBool 的值设置为 true。Ordering::Relaxed 表示内存排序约束较弱,适用于不需要严格排序的场景。

读取值:

rust

let value = atomic_bool.load(Ordering::Relaxed);

使用 load 方法读取 AtomicBool 的当前值。

比较并交换:

rust

let old_value = atomic_bool.compare_and_swap(true, false, Ordering::Relaxed);

compare_and_swap 方法会检查当前值是否为 true,如果是,则将其设置为 false 并返回旧值。

再次读取值:

rust

let new_value = atomic_bool.load(Ordering::Relaxed);

再次读取值以确认更改。

交换值:

rust

let swapped_value = atomic_bool.swap(true, Ordering::Relaxed);

swap 方法将值设置为 true 并返回旧值。

内存排序 (Ordering)

在原子操作中,内存排序 (Ordering) 决定了操作的内存可见性和顺序。常见的选项包括:

Ordering::Relaxed:最弱的排序约束,适用于不需要严格排序的场景。

Ordering::Acquire:在读取操作之前获取所有权。

Ordering::Release:在写入操作之后释放所有权。

Ordering::SeqCst:顺序一致性,最强的排序约束,确保所有线程以相同的顺序看到操作。

选择适当的内存排序约束可以提高性能,但也要确保满足你的并发需求。

相关推荐
聚客AI1 天前
🙋‍♀️Transformer训练与推理全流程:从输入处理到输出生成
人工智能·算法·llm
大怪v1 天前
前端:人工智能?我也会啊!来个花活,😎😎😎“自动驾驶”整起!
前端·javascript·算法
惯导马工1 天前
【论文导读】ORB-SLAM3:An Accurate Open-Source Library for Visual, Visual-Inertial and
深度学习·算法
骑自行车的码农1 天前
【React用到的一些算法】游标和栈
算法·react.js
博笙困了1 天前
AcWing学习——双指针算法
c++·算法
moonlifesudo1 天前
322:零钱兑换(三种方法)
算法
该用户已不存在2 天前
Mojo vs Python vs Rust: 2025年搞AI,该学哪个?
后端·python·rust
NAGNIP2 天前
大模型框架性能优化策略:延迟、吞吐量与成本权衡
算法
大卫小东(Sheldon)2 天前
写了一个BBP算法的实现库,欢迎讨论
数学·rust
美团技术团队2 天前
LongCat-Flash:如何使用 SGLang 部署美团 Agentic 模型
人工智能·算法