github-mirrors/librespot
1
0
Fork 0
mirror of https://github.com/librespot-org/librespot.git synced 2024-12-18 17:11:53 +00:00
Code Issues Projects Releases Packages Wiki Activity

Rate-limit automatic re-connection attempts when spirc shuts down.

Browse source
This commit is contained in:
Konstantin Seiler 2020-01-23 19:09:26 +11:00
parent 17821b26aa
commit f26db01110
1 changed files with 13 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/main.rs
View file

@ -25,6 +25,7 @@ use std::str::FromStr;
use tokio_core::reactor::{Core, Handle};
use tokio_io::IoStream;
use url::Url;
use std::time::Instant;
use librespot::core::authentication::{get_credentials, Credentials};
use librespot::core::cache::Cache;
@ -382,6 +383,7 @@ struct Main {
shutdown: bool,
last_credentials: Option<Credentials>,
auto_connect_times: Vec<Instant>,
player_event_channel: Option<UnboundedReceiver<PlayerEvent>>,
player_event_program: Option<String>,
@ -406,6 +408,7 @@ impl Main {
spirc_task: None,
shutdown: false,
last_credentials: None,
auto_connect_times: Vec::new(),
signal: Box::new(tokio_signal::ctrl_c().flatten_stream()),
player_event_channel: None,
@ -454,6 +457,7 @@ impl Future for Main {
if let Some(ref spirc) = self.spirc {
spirc.shutdown();
}
self.auto_connect_times.clear();
self.credentials(creds);
progress = true;
@ -505,7 +509,16 @@ impl Future for Main {
} else {
warn!("Spirc shut down unexpectedly");
self.spirc_task = None;
while (!self.auto_connect_times.is_empty()) && ((Instant::now() - self.auto_connect_times[0]).as_secs() > 600) {
let _ = self.auto_connect_times.remove(0);
}
if self.auto_connect_times.len() >= 5 {
error!("Spirc shut down too often. Exiting to avoid too many login attempts.");
return Ok(Async::Ready(()));
}
if let Some(credentials) = self.last_credentials.clone() {
self.auto_connect_times.push(Instant::now());
self.credentials(credentials);
progress = true;
}