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
|
use std::borrow::Cow;
use std::sync::Arc;
use axum::extract::{ws::WebSocket};
use axum::extract::ws::{CloseFrame, Message};
use dashmap::DashMap;
use tokio::sync::broadcast::{Sender};
use tracing::{debug, trace, warn};
use crate::AppState;
use crate::error::WebolError;
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) -> Result<(), WebolError> {
let payload = [0; 8];
// TODO: Better while
let mut cont = true;
while cont {
let ping = surge_ping::ping(
ip.parse().map_err(WebolError::IpParse)?,
&payload
).await;
if let Err(ping) = ping {
cont = matches!(ping, surge_ping::SurgeError::Timeout { .. });
if !cont {
return Err(ping).map_err(WebolError::Ping)
}
} else {
let (_, duration) = ping.unwrap();
debug!("Ping took {:?}", duration);
cont = false;
handle_broadcast_send(&tx, ip.clone(), ping_map, uuid.clone()).await;
};
}
Ok(())
}
async fn handle_broadcast_send(tx: &Sender<BroadcastCommands>, ip: String, ping_map: &PingMap, uuid: String) {
debug!("sending pingsuccess message");
ping_map.insert(uuid.clone(), PingValue { ip: ip.clone(), online: true });
let _ = tx.send(BroadcastCommands::PingSuccess(ip));
tokio::time::sleep(tokio::time::Duration::from_secs(60)).await;
trace!("remove {} from ping_map", uuid);
ping_map.remove(&uuid);
}
#[derive(Clone, Debug)]
pub enum BroadcastCommands {
PingSuccess(String)
}
pub async fn status_websocket(mut socket: WebSocket, state: Arc<AppState>) {
warn!("{:?}", state.ping_map);
trace!("wait for ws message (uuid)");
let msg = socket.recv().await;
let uuid = msg.unwrap().unwrap().into_text().unwrap();
trace!("Search for uuid: {:?}", uuid);
// TODO: Handle Error
let device = state.ping_map.get(&uuid).unwrap().to_owned();
trace!("got device: {:?}", device);
match device.online {
true => {
debug!("already started");
// TODO: What's better?
// socket.send(Message::Text(format!("start_{}", uuid))).await.unwrap();
// socket.close().await.unwrap();
socket.send(Message::Close(Some(CloseFrame { code: 4001, reason: Cow::from(format!("start_{}", uuid)) }))).await.unwrap();
},
false => {
let ip = device.ip.to_owned();
loop{
trace!("wait for tx message");
let message = state.ping_send.subscribe().recv().await.unwrap();
trace!("GOT = {:?}", message);
// if let BroadcastCommands::PingSuccess(msg_ip) = message {
// if msg_ip == ip {
// trace!("message == ip");
// break;
// }
// }
let BroadcastCommands::PingSuccess(msg_ip) = message;
if msg_ip == ip {
trace!("message == ip");
break;
}
};
socket.send(Message::Close(Some(CloseFrame { code: 4000, reason: Cow::from(format!("start_{}", uuid)) }))).await.unwrap();
// socket.send(Message::Text(format!("start_{}", uuid))).await.unwrap();
// socket.close().await.unwrap();
warn!("{:?}", state.ping_map);
}
}
}
|
