Switched from std::RwLock to parking_lot

This commit is contained in:
Dreaded_X 2023-01-06 23:16:36 +01:00
parent 03282b81c5
commit 50ff5b162e
Signed by: Dreaded_X
GPG Key ID: 76BDEC4E165D8AD9
6 changed files with 15 additions and 9 deletions

1
Cargo.lock generated
View File

@ -44,6 +44,7 @@ dependencies = [
"dotenv", "dotenv",
"google-home", "google-home",
"impl_cast", "impl_cast",
"parking_lot",
"paste", "paste",
"pollster", "pollster",
"regex", "regex",

View File

@ -28,6 +28,7 @@ tracing-subscriber = { version = "0.3.16", features = ["env-filter"] }
bytes = "1.3.0" bytes = "1.3.0"
pollster = "0.2.5" pollster = "0.2.5"
regex = "1.7.0" regex = "1.7.0"
parking_lot = "0.12.1"
[profile.release] [profile.release]
lto=true lto=true

View File

@ -1,5 +1,6 @@
use std::sync::{Weak, RwLock}; use std::sync::Weak;
use parking_lot::RwLock;
use pollster::FutureExt as _; use pollster::FutureExt as _;
use rumqttc::{AsyncClient, matches}; use rumqttc::{AsyncClient, matches};
use tracing::{span, Level, log::{error, trace}, debug}; use tracing::{span, Level, log::{error, trace}, debug};
@ -34,7 +35,7 @@ impl LightSensor {
let _span = span!(Level::TRACE, "darkness_update").entered(); let _span = span!(Level::TRACE, "darkness_update").entered();
listeners.into_iter().for_each(|listener| { listeners.into_iter().for_each(|listener| {
if let Some(listener) = listener.upgrade() { if let Some(listener) = listener.upgrade() {
listener.write().unwrap().on_darkness(dark); listener.write().on_darkness(dark);
} }
}) })
} }

View File

@ -1,5 +1,6 @@
#![feature(async_closure)] #![feature(async_closure)]
use std::{time::Duration, sync::{Arc, RwLock}, process}; use std::{time::Duration, sync::Arc, process};
use parking_lot::RwLock;
use axum::{Router, Json, routing::post, http::StatusCode, extract::FromRef}; use axum::{Router, Json, routing::post, http::StatusCode, extract::FromRef};
@ -67,7 +68,7 @@ async fn main() {
device_config.into(identifier, &config, client.clone()) device_config.into(identifier, &config, client.clone())
}) })
.for_each(|device| { .for_each(|device| {
devices.write().unwrap().add_device(device); devices.write().add_device(device);
}); });
// Setup presence system // Setup presence system
@ -101,7 +102,7 @@ async fn main() {
// Handle request might block, so we need to spawn a blocking task // Handle request might block, so we need to spawn a blocking task
tokio::task::spawn_blocking(move || { tokio::task::spawn_blocking(move || {
let gc = GoogleHome::new(&user.preferred_username); let gc = GoogleHome::new(&user.preferred_username);
let result = gc.handle_request(payload, &mut devices.write().unwrap().as_google_home_devices()).unwrap(); let result = gc.handle_request(payload, &mut devices.write().as_google_home_devices()).unwrap();
return (StatusCode::OK, Json(result)); return (StatusCode::OK, Json(result));
}).await.unwrap() }).await.unwrap()

View File

@ -1,4 +1,5 @@
use std::sync::{Weak, RwLock}; use std::sync::Weak;
use parking_lot::RwLock;
use serde::{Serialize, Deserialize}; use serde::{Serialize, Deserialize};
use tracing::{error, debug, span, Level}; use tracing::{error, debug, span, Level};
@ -24,7 +25,7 @@ impl Mqtt {
let _span = span!(Level::TRACE, "mqtt_message").entered(); let _span = span!(Level::TRACE, "mqtt_message").entered();
listeners.into_iter().for_each(|listener| { listeners.into_iter().for_each(|listener| {
if let Some(listener) = listener.upgrade() { if let Some(listener) = listener.upgrade() {
listener.write().unwrap().on_mqtt(&message); listener.write().on_mqtt(&message);
} }
}) })
} }

View File

@ -1,5 +1,6 @@
use std::{sync::{Weak, RwLock}, collections::HashMap}; use std::{sync::Weak, collections::HashMap};
use parking_lot::RwLock;
use tracing::{debug, span, Level, error}; use tracing::{debug, span, Level, error};
use rumqttc::{AsyncClient, matches}; use rumqttc::{AsyncClient, matches};
use pollster::FutureExt as _; use pollster::FutureExt as _;
@ -32,7 +33,7 @@ impl Presence {
let _span = span!(Level::TRACE, "presence_update").entered(); let _span = span!(Level::TRACE, "presence_update").entered();
listeners.into_iter().for_each(|listener| { listeners.into_iter().for_each(|listener| {
if let Some(listener) = listener.upgrade() { if let Some(listener) = listener.upgrade() {
listener.write().unwrap().on_presence(presence); listener.write().on_presence(presence);
} }
}) })
} }