如何在Rust的Map中自增计数器并获取当前值？能否单表达式实现？

普通HashMap场景 #

你的现有代码可以合并为一个块表达式，避免额外的get(&key).unwrap()调用（毕竟通过entry操作后，key必然存在）：

use std::collections::HashMap;

let mut map: HashMap<&str, usize> = HashMap::new();
let key = "test";

// 单个块表达式完成自增并获取当前计数
let count = {
    let counter = map.entry(key).or_default();
    *counter += 1;
    *counter
};

这个写法利用了Rust的块表达式特性——整个块会返回最后一行表达式的值，因此可以将自增和取值逻辑封装在一个表达式内。

原子操作/并发安全场景 #

如果需要线程安全的原子操作，推荐结合AtomicUsize和并发安全的Map实现：

标准库方案（Mutex + HashMap + AtomicUsize） #

use std::collections::HashMap;
use std::sync::atomic::{AtomicUsize, Ordering};
use std::sync::Mutex;

let map = Mutex::new(HashMap::new());
let key = "test";

// 原子自增并返回当前值
let count = {
    let mut locked_map = map.lock().unwrap();
    let atomic_counter = locked_map.entry(key)
        .or_insert_with(|| AtomicUsize::new(0));
    // fetch_add返回自增前的旧值，加1得到当前计数
    atomic_counter.fetch_add(1, Ordering::Relaxed) + 1
};

第三方库方案（DashMap） #

DashMap是无锁并发Map，用法更简洁（需在Cargo.toml添加dashmap = "5.4"依赖）：

use dashmap::DashMap;
use std::sync::atomic::{AtomicUsize, Ordering};

let map = DashMap::new();
let key = "test";

// 单链式表达式完成原子自增并取值
let count = map.entry(key)
    .or_insert_with(|| AtomicUsize::new(0))
    .fetch_add(1, Ordering::Relaxed) + 1;

注意：Ordering::Relaxed适用于普通计数场景，若有更严格的内存顺序需求，可根据实际情况替换为Ordering::Acquire/Ordering::Release等。

内容的提问来源于stack exchange，提问作者fadedbee