Constellation, Spacedust, Slingshot, UFOs: atproto crates and services for microcosm
jetstream: handle socket timeout in read loop
hopefully this will actually cause it to try to reconnect?
+8
-1
+8
-1
link_aggregator/src/consumer/jetstream.rs
+8
-1
link_aggregator/src/consumer/jetstream.rs
···
2
2
use metrics::{
3
3
counter, describe_counter, describe_gauge, describe_histogram, gauge, histogram, Unit,
4
4
};
5
-
use std::io::{Cursor, Read};
5
+
use std::io::{Cursor, ErrorKind, Read};
6
6
use std::net::ToSocketAddrs;
7
7
use std::thread;
8
8
use std::time;
···
183
183
Err(TError::ConnectionClosed) => {
184
184
counter!("jetstream_read_fail", "url" => stream.clone(), "reason" => "clean close").increment(1);
185
185
println!("jetstream closed the websocket cleanly.");
186
+
break;
187
+
}
188
+
Err(TError::Io(e))
189
+
if matches!(e.kind(), ErrorKind::WouldBlock | ErrorKind::TimedOut) =>
190
+
{
191
+
counter!("jetstream_read_fail", "url" => stream.clone(), "reason" => "timed out").increment(1);
192
+
println!("jetstream socket timed out. bailing to reconnect -- should we be trying to close first?");
186
193
break;
187
194
}
188
195
r => eprintln!("jetstream: close result after error: {r:?}"),