@@ -25,7 +25,7 @@ use tokio::{
25
25
} ;
26
26
use tokio_stream:: wrappers:: ReceiverStream ;
27
27
use tokio_util:: { io:: SyncIoBridge , sync:: CancellationToken } ;
28
- use tracing:: { error, info_span, instrument, trace, trace_span, warn, Instrument } ;
28
+ use tracing:: { error, info , info_span, instrument, trace, trace_span, warn, Instrument } ;
29
29
30
30
use crate :: {
31
31
bincode_input_closed,
@@ -2540,14 +2540,7 @@ pub struct TerminateContainer(Option<(String, Command)>);
2540
2540
2541
2541
impl TerminateContainer {
2542
2542
pub fn new ( name : String , command : Command ) -> Self {
2543
- let was_inserted = TRACKED_CONTAINERS
2544
- . lock ( )
2545
- . unwrap_or_else ( |e| e. into_inner ( ) )
2546
- . insert ( name. clone ( ) . into ( ) ) ;
2547
-
2548
- if !was_inserted {
2549
- error ! ( %name, "This container was already tracked; duplicates are bad logic" ) ;
2550
- }
2543
+ Self :: start_tracking ( & name) ;
2551
2544
2552
2545
Self ( Some ( ( name, command) ) )
2553
2546
}
@@ -2571,13 +2564,31 @@ impl TerminateContainer {
2571
2564
Ok ( ( ) )
2572
2565
}
2573
2566
2567
+ #[ instrument]
2568
+ fn start_tracking ( name : & str ) {
2569
+ let was_inserted = TRACKED_CONTAINERS
2570
+ . lock ( )
2571
+ . unwrap_or_else ( |e| e. into_inner ( ) )
2572
+ . insert ( name. into ( ) ) ;
2573
+
2574
+ if was_inserted {
2575
+ info ! ( %name, "Started tracking container" ) ;
2576
+ } else {
2577
+ error ! ( %name, "Started tracking container, but it was already tracked" ) ;
2578
+ }
2579
+ }
2580
+
2581
+ #[ instrument]
2574
2582
fn stop_tracking ( name : & str ) {
2575
2583
let was_tracked = TRACKED_CONTAINERS
2576
2584
. lock ( )
2577
2585
. unwrap_or_else ( |e| e. into_inner ( ) )
2578
2586
. remove ( name) ;
2579
- if !was_tracked {
2580
- error ! ( %name, "Container was not in the tracking set" ) ;
2587
+
2588
+ if was_tracked {
2589
+ info ! ( %name, "Stopped tracking container" ) ;
2590
+ } else {
2591
+ error ! ( %name, "Stopped tracking container, but it was not in the tracking set" ) ;
2581
2592
}
2582
2593
}
2583
2594
0 commit comments