@@ -43,7 +43,7 @@ pub struct BackgroundProcessor {
43
43
stop_thread : Arc < AtomicBool > ,
44
44
/// May be used to retrieve and handle the error if `BackgroundProcessor`'s thread
45
45
/// exits due to an error while persisting.
46
- pub thread_handle : Option < JoinHandle < Result < ( ) , std:: io:: Error > > > ,
46
+ thread_handle : Option < JoinHandle < Result < ( ) , std:: io:: Error > > > ,
47
47
}
48
48
49
49
#[ cfg( not( test) ) ]
@@ -84,21 +84,25 @@ ChannelManagerPersister<Signer, M, T, K, F, L> for Fun where
84
84
}
85
85
86
86
impl BackgroundProcessor {
87
- /// Start a background thread that takes care of responsibilities enumerated in the top-level
88
- /// documentation.
87
+ /// Start a background thread that takes care of responsibilities enumerated in the [ top-level
88
+ /// documentation] .
89
89
///
90
- /// If `persist_manager` returns an error, then this thread will return said error (and
91
- /// `start()` will need to be called again to restart the `BackgroundProcessor`). Users should
92
- /// wait on [`thread_handle`]'s `join()` method to be able to tell if and when an error is
93
- /// returned, or implement `persist_manager` such that an error is never returned to the
94
- /// `BackgroundProcessor`
90
+ /// The thread runs indefinitely unless the object is dropped, [`stop`] is called, or
91
+ /// `persist_manager` returns an error. In case of an error, the error is retrieved by calling
92
+ /// either [`join`] or [`stop`].
93
+ ///
94
+ /// Typically, users should either implement [`ChannelManagerPersister`] to never return an
95
+ /// error or call [`join`] and handle any error that may arise. For the latter case, the
96
+ /// `BackgroundProcessor` must be restarted by calling `start` again after handling the error.
95
97
///
96
98
/// `persist_manager` is responsible for writing out the [`ChannelManager`] to disk, and/or
97
99
/// uploading to one or more backup services. See [`ChannelManager::write`] for writing out a
98
100
/// [`ChannelManager`]. See [`FilesystemPersister::persist_manager`] for Rust-Lightning's
99
101
/// provided implementation.
100
102
///
101
- /// [`thread_handle`]: BackgroundProcessor::thread_handle
103
+ /// [top-level documentation]: Self
104
+ /// [`join`]: Self::join
105
+ /// [`stop`]: Self::stop
102
106
/// [`ChannelManager`]: lightning::ln::channelmanager::ChannelManager
103
107
/// [`ChannelManager::write`]: lightning::ln::channelmanager::ChannelManager#impl-Writeable
104
108
/// [`FilesystemPersister::persist_manager`]: lightning_persister::FilesystemPersister::persist_manager
@@ -161,14 +165,30 @@ impl BackgroundProcessor {
161
165
Self { stop_thread : stop_thread_clone, thread_handle : Some ( handle) }
162
166
}
163
167
164
- /// Stop `BackgroundProcessor`'s thread.
168
+ /// Join `BackgroundProcessor`'s thread, returning any error that occurred while persisting
169
+ /// [`ChannelManager`].
170
+ ///
171
+ /// [`ChannelManager`]: lightning::ln::channelmanager::ChannelManager
172
+ pub fn join ( mut self ) -> Result < ( ) , std:: io:: Error > {
173
+ assert ! ( self . thread_handle. is_some( ) ) ;
174
+ self . join_thread ( )
175
+ }
176
+
177
+ /// Stop `BackgroundProcessor`'s thread, returning any error that occurred while persisting
178
+ /// [`ChannelManager`].
179
+ ///
180
+ /// [`ChannelManager`]: lightning::ln::channelmanager::ChannelManager
165
181
pub fn stop ( mut self ) -> Result < ( ) , std:: io:: Error > {
166
182
assert ! ( self . thread_handle. is_some( ) ) ;
167
183
self . stop_and_join_thread ( )
168
184
}
169
185
170
186
fn stop_and_join_thread ( & mut self ) -> Result < ( ) , std:: io:: Error > {
171
187
self . stop_thread . store ( true , Ordering :: Release ) ;
188
+ self . join_thread ( )
189
+ }
190
+
191
+ fn join_thread ( & mut self ) -> Result < ( ) , std:: io:: Error > {
172
192
match self . thread_handle . take ( ) {
173
193
Some ( handle) => handle. join ( ) . unwrap ( ) ,
174
194
None => Ok ( ( ) ) ,
@@ -430,7 +450,7 @@ mod tests {
430
450
let persister = |_: & _ | Err ( std:: io:: Error :: new ( std:: io:: ErrorKind :: Other , "test" ) ) ;
431
451
let event_handler = |_| { } ;
432
452
let bg_processor = BackgroundProcessor :: start ( persister, event_handler, nodes[ 0 ] . chain_monitor . clone ( ) , nodes[ 0 ] . node . clone ( ) , nodes[ 0 ] . peer_manager . clone ( ) , nodes[ 0 ] . logger . clone ( ) ) ;
433
- match bg_processor. stop ( ) {
453
+ match bg_processor. join ( ) {
434
454
Ok ( _) => panic ! ( "Expected error persisting manager" ) ,
435
455
Err ( e) => {
436
456
assert_eq ! ( e. kind( ) , std:: io:: ErrorKind :: Other ) ;
0 commit comments