1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
use std::str::FromStr;
use std::net::IpAddr;
use std::sync::Arc;
use axum::extract::{ws::WebSocket};
use axum::extract::ws::Message;
use dashmap::DashMap;
use time::{Duration, Instant};
use tokio::sync::broadcast::{Sender};
use tracing::{debug, error, trace};
use crate::AppState;
use crate::config::SETTINGS;
pub type PingMap = DashMap<String, PingValue>;
#[derive(Debug, Clone)]
pub struct PingValue {
pub ip: String,
pub online: bool
}
pub async fn spawn(tx: Sender<BroadcastCommands>, ip: String, uuid: String, ping_map: &PingMap) {
let timer = Instant::now();
let payload = [0; 8];
let ping_ip = IpAddr::from_str(&ip).expect("bad ip");
let mut msg: Option<BroadcastCommands> = None;
while msg.is_none() {
let ping = surge_ping::ping(
ping_ip,
&payload
).await;
if let Err(ping) = ping {
let ping_timeout = matches!(ping, surge_ping::SurgeError::Timeout { .. });
if !ping_timeout {
error!("{}", ping.to_string());
msg = Some(BroadcastCommands::Error(uuid.clone()));
}
if timer.elapsed() >= Duration::minutes(SETTINGS.get_int("pingtimeout").unwrap_or(10)) {
msg = Some(BroadcastCommands::Timeout(uuid.clone()));
}
} else {
let (_, duration) = ping.map_err(|err| error!("{}", err.to_string())).expect("fatal error");
debug!("ping took {:?}", duration);
msg = Some(BroadcastCommands::Success(uuid.clone()));
};
}
let msg = msg.expect("fatal error");
let _ = tx.send(msg.clone());
if let BroadcastCommands::Success(..) = msg {
ping_map.insert(uuid.clone(), PingValue { ip: ip.clone(), online: true });
tokio::time::sleep(tokio::time::Duration::from_secs(60)).await;
}
trace!("remove {} from ping_map", uuid);
ping_map.remove(&uuid);
}
#[derive(Clone, Debug, PartialEq)]
pub enum BroadcastCommands {
Success(String),
Timeout(String),
Error(String),
}
pub async fn status_websocket(mut socket: WebSocket, state: Arc<AppState>) {
trace!("wait for ws message (uuid)");
let msg = socket.recv().await;
let uuid = msg.unwrap().unwrap().into_text().unwrap();
trace!("Search for uuid: {:?}", uuid);
let device_exists = state.ping_map.contains_key(&uuid);
match device_exists {
true => {
let _ = socket.send(process_device(state.clone(), uuid).await).await;
},
false => {
debug!("didn't find any device");
let _ = socket.send(Message::Text(format!("notfound_{}", uuid))).await;
},
};
let _ = socket.close().await;
}
async fn process_device(state: Arc<AppState>, uuid: String) -> Message {
let pm = state.ping_map.clone().into_read_only();
let device = pm.get(&uuid).expect("fatal error");
debug!("got device: {} (online: {})", device.ip, device.online);
match device.online {
true => {
debug!("already started");
Message::Text(format!("start_{}", uuid))
},
false => {
loop{
trace!("wait for tx message");
let message = state.ping_send.subscribe().recv().await.expect("fatal error");
trace!("got message {:?}", message);
return match message {
BroadcastCommands::Success(msg_uuid) => {
if msg_uuid != uuid { continue; }
trace!("message == uuid success");
Message::Text(format!("start_{}", uuid))
},
BroadcastCommands::Timeout(msg_uuid) => {
if msg_uuid != uuid { continue; }
trace!("message == uuid timeout");
Message::Text(format!("timeout_{}", uuid))
},
BroadcastCommands::Error(msg_uuid) => {
if msg_uuid != uuid { continue; }
trace!("message == uuid error");
Message::Text(format!("error_{}", uuid))
}
}
}
}
}
}
|
