File tree 2 files changed +13
-4
lines changed
2 files changed +13
-4
lines changed Original file line number Diff line number Diff line change @@ -745,12 +745,19 @@ impl GlobalBarrierManager {
745
745
746
746
send_latency_timer. observe_duration ( ) ;
747
747
748
- let node_to_collect = self
748
+ let node_to_collect = match self
749
749
. control_stream_manager
750
750
. inject_barrier ( command_ctx. clone ( ) )
751
- . inspect_err ( |_| {
751
+ {
752
+ Ok ( node_to_collect) => node_to_collect,
753
+ Err ( err) => {
754
+ for notifier in notifiers {
755
+ notifier. notify_failed ( err. clone ( ) ) ;
756
+ }
752
757
fail_point ! ( "inject_barrier_err_success" ) ;
753
- } ) ?;
758
+ return Err ( err) ;
759
+ }
760
+ } ;
754
761
755
762
// Notify about the injection.
756
763
let prev_paused_reason = self . state . paused_reason ( ) ;
Original file line number Diff line number Diff line change @@ -241,7 +241,9 @@ impl BarrierScheduler {
241
241
..Default :: default ( )
242
242
} ;
243
243
self . attach_notifiers ( vec ! [ notifier] , checkpoint) ?;
244
- rx. await . unwrap ( )
244
+ rx. await
245
+ . ok ( )
246
+ . context ( "failed to wait for barrier collect" ) ?
245
247
}
246
248
247
249
/// Run multiple commands and return when they're all completely finished. It's ensured that
You can’t perform that action at this time.
0 commit comments