fix: improve dashmap reference management

Pumpkin-MC · Mar 1, 2025 · 1cbaad0 · 1cbaad0
1 parent 1a2b492
commit 1cbaad0
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/pumpkin-world/src/level.rs b/pumpkin-world/src/level.rs
@@ -383,21 +383,23 @@ impl Level {
             let mut recv = recv;
             while let Some((pos, data)) = recv.recv().await {
                 if let Some(data) = data {
-                    let entry = loaded_chunks
+                    let value = loaded_chunks
                         .entry(pos)
-                        .or_insert_with(|| Arc::new(RwLock::new(data)));
-                    let value = entry.clone();
+                        .or_insert_with(|| Arc::new(RwLock::new(data)))
+                        .value()
+                        .clone();
                     send_chunk(false, value).await;
                 } else {
                     let loaded_chunks = loaded_chunks.clone();
                     let world_gen = world_gen.clone();
                     let gen_channel = gen_channel.clone();
                     rayon::spawn(move || {
                         let data = world_gen.generate_chunk(pos);
-                        let entry = loaded_chunks
+                        let value = loaded_chunks
                             .entry(pos)
-                            .or_insert_with(|| Arc::new(RwLock::new(data)));
-                        let value = entry.clone();
+                            .or_insert_with(|| Arc::new(RwLock::new(data)))
+                            .value()
+                            .clone();
                         gen_channel
                             .blocking_send((value, true))
                             .expect("Failed to send chunk from generation thread!");