After complete callback done

bastion-rs · Oct 20, 2019 · 3838dec · 3838dec
1 parent 1b0f39b
commit 3838dec
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/lightproc/examples/proc_run.rs b/lightproc/examples/proc_run.rs
@@ -7,6 +7,7 @@ use std::thread;
 use crossbeam::channel;
 use futures::executor;
 use lightproc::prelude::*;
+use std::sync::atomic::AtomicUsize;
 
 fn spawn_on_thread<F, R>(fut: F) -> ProcHandle<R>
     where
@@ -26,7 +27,15 @@ fn spawn_on_thread<F, R>(fut: F) -> ProcHandle<R>
     let (proc, handle) = LightProc::build(
         future,
         schedule,
-        ProcStack::default()
+        ProcStack {
+            pid: AtomicUsize::new(1),
+            after_complete: Some(Arc::new(|| {
+                println!("After complete");
+            })),
+            after_start: Some(Arc::new(|| {
+                println!("After start");
+            }))
+        }
     );
 
     proc.schedule();

diff --git a/lightproc/src/raw_proc.rs b/lightproc/src/raw_proc.rs
@@ -438,6 +438,10 @@ where
                                 (*raw.pdata).notify();
                             }
 
+                            if let Some(after_complete_cb) = &(*raw.stack).after_complete {
+                                (*after_complete_cb)();
+                            }
+
                             // Drop the task reference.
                             Self::decrement(ptr);
                             break;