commit 299b19183f3aba8963be152b2f5d5ed2ba491f93 · nekomimi.pet/microcosm-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
       5
       -
       use std::io::{Cursor, Read};

     

       5
       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
       186
       +
                                   break;

     

       187
       187
       +
                               }

     

       188
       188
       +
                               Err(TError::Io(e))

     

       189
       189
       +
                                   if matches!(e.kind(), ErrorKind::WouldBlock | ErrorKind::TimedOut) =>

     

       190
       190
       +
                               {

     

       191
       191
       +
                                   counter!("jetstream_read_fail", "url" => stream.clone(), "reason" => "timed out").increment(1);

     

       192
       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:?}"),