nero/tsuki
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases Packages Wiki Activity Actions

fix: atomic increment

Browse source
This commit is contained in:
Timo Kösters 2021-12-20 10:16:22 +01:00
parent ee3d2db8e0
commit b1d9ec3efc
No known key found for this signature in database
GPG key ID: 356E705610F626D5
3 changed files with 21 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/database/abstraction/watchers.rs
View file

@ -1,6 +1,6 @@
use parking_lot::RwLock;
use std::{
collections::{hash_map, HashMap},
sync::RwLock,
future::Future,
pin::Pin,
};
@ -16,7 +16,7 @@ impl Watchers {
&'a self,
prefix: &[u8],
) -> Pin<Box<dyn Future<Output = ()> + Send + 'a>> {
let mut rx = match self.watchers.write().entry(prefix.to_vec()) {
let mut rx = match self.watchers.write().unwrap().entry(prefix.to_vec()) {
hash_map::Entry::Occupied(o) => o.get().1.clone(),
hash_map::Entry::Vacant(v) => {
let (tx, rx) = tokio::sync::watch::channel(());
@ -31,7 +31,7 @@ impl Watchers {
})
}
pub(super) fn wake(&self, key: &[u8]) {
let watchers = self.watchers.read();
let watchers = self.watchers.read().unwrap();
let mut triggered = Vec::new();
for length in 0..=key.len() {
@ -43,7 +43,7 @@ impl Watchers {
drop(watchers);
if !triggered.is_empty() {
let mut watchers = self.watchers.write();
let mut watchers = self.watchers.write().unwrap();
for prefix in triggered {
if let Some(tx) = watchers.remove(prefix) {
let _ = tx.0.send(());