soruh/centralex
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

change some logging

Browse Source
This commit is contained in:
soruh 2023-03-19 00:40:50 +01:00
parent 0686673e72
commit 60fedabb01
1 changed files with 23 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/ports.rs
View File

@ -13,7 +13,7 @@ use std::{
use anyhow::anyhow; use anyhow::anyhow;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use tokio::{net::TcpListener, sync::Mutex, task::JoinHandle, time::Instant}; use tokio::{net::TcpListener, sync::Mutex, task::JoinHandle, time::Instant};
use tracing::{error, info, warn}; use tracing::{debug, error, info, warn};
use crate::{ use crate::{
constants::{PORT_OWNERSHIP_TIMEOUT, PORT_RETRY_TIME}, constants::{PORT_OWNERSHIP_TIMEOUT, PORT_RETRY_TIME},
@ -239,7 +239,7 @@ impl PortHandler {
} }
pub fn store(&self, cache: &Path) -> anyhow::Result<()> { pub fn store(&self, cache: &Path) -> anyhow::Result<()> {
info!("storing cache"); debug!("storing cache");
let temp_file = cache.with_extension(".temp"); let temp_file = cache.with_extension(".temp");
serde_json::to_writer(BufWriter::new(File::create(&temp_file)?), self)?; serde_json::to_writer(BufWriter::new(File::create(&temp_file)?), self)?;
@ -392,16 +392,19 @@ impl Rejector {
impl PortHandler { impl PortHandler {
pub fn allocate_port_for_number(&mut self, config: &Config, number: Number) -> Option<Port> { pub fn allocate_port_for_number(&mut self, config: &Config, number: Number) -> Option<Port> {
if let Some(port) = self.allocated_ports.get(&number) { let port = if let Some(port) = self.allocated_ports.get(&number) {
let already_connected = self let already_connected = self
.port_state .port_state
.get(port) .get(port)
.map(|state| state.status != PortStatus::Disconnected) .map(|state| state.status != PortStatus::Disconnected)
.unwrap_or(false); .unwrap_or(false);
return if already_connected { None } else { Some(*port) }; if already_connected {
None
} else {
Some(*port)
} }
} else {
let port = if let Some(&port) = self.free_ports.iter().next() { let port = if let Some(&port) = self.free_ports.iter().next() {
self.register_update(); self.register_update();
self.free_ports.remove(&port); self.free_ports.remove(&port);
@ -412,6 +415,13 @@ impl PortHandler {
assert!(self.allocated_ports.insert(number, port).is_none()); assert!(self.allocated_ports.insert(number, port).is_none());
Some(port) Some(port)
};
if let Some(port) = port {
info!(port, "allocated port");
}
port
} }
fn try_recover_port(&mut self, config: &Config) -> Option<Port> { fn try_recover_port(&mut self, config: &Config) -> Option<Port> {