Highlight port in red if tunnel failed to open
This commit is contained in:
parent
75bfd4d5cf
commit
95e47c708c
|
@ -1,6 +1,6 @@
|
|||
use std::pin::Pin;
|
||||
use std::sync::Arc;
|
||||
use std::sync::atomic::{AtomicUsize, Ordering};
|
||||
use std::sync::atomic::{AtomicBool, AtomicUsize, Ordering};
|
||||
use std::task::{Context, Poll};
|
||||
|
||||
use pin_project_lite::pin_project;
|
||||
|
@ -14,6 +14,7 @@ pub struct Stats {
|
|||
connections: AtomicUsize,
|
||||
rx: AtomicUsize,
|
||||
tx: AtomicUsize,
|
||||
failed: AtomicBool,
|
||||
}
|
||||
|
||||
impl Stats {
|
||||
|
@ -33,6 +34,14 @@ impl Stats {
|
|||
self.connections.load(Ordering::Relaxed)
|
||||
}
|
||||
|
||||
pub fn failed(&self) -> bool {
|
||||
self.failed.load(Ordering::Relaxed)
|
||||
}
|
||||
|
||||
pub fn set_failed(&self, failed: bool) {
|
||||
self.failed.store(failed, Ordering::Relaxed);
|
||||
}
|
||||
|
||||
pub fn rx(&self) -> Unit {
|
||||
Unit::new(self.rx.load(Ordering::Relaxed), "B")
|
||||
}
|
||||
|
|
|
@ -40,7 +40,12 @@ impl TunnelInner {
|
|||
&self.internal_address,
|
||||
self.port,
|
||||
)
|
||||
.await?;
|
||||
.await
|
||||
.inspect_err(|_| {
|
||||
self.stats.set_failed(true);
|
||||
})?;
|
||||
|
||||
self.stats.set_failed(false);
|
||||
|
||||
Ok(TrackStats::new(channel.into_stream(), self.stats.clone()))
|
||||
}
|
||||
|
|
|
@ -17,9 +17,15 @@ pub struct TunnelRow {
|
|||
|
||||
impl From<&TunnelRow> for Vec<Span<'static>> {
|
||||
fn from(row: &TunnelRow) -> Self {
|
||||
let port = if row.stats.failed() {
|
||||
row.port.clone().red()
|
||||
} else {
|
||||
row.port.clone()
|
||||
};
|
||||
|
||||
vec![
|
||||
row.name.clone(),
|
||||
row.port.clone(),
|
||||
port,
|
||||
row.access.clone(),
|
||||
row.address.clone(),
|
||||
row.stats.connections().to_string().into(),
|
||||
|
|
Loading…
Reference in New Issue
Block a user