_damo_records: Implement a function for wait recording

Signed-off-by: SeongJae Park <sj@kernel.org>

1 files changed, 6 insertions, 0 deletions

diff --git a/_damo_records.py b/_damo_records.py
index 9bd8d653..21b6438f 100644
--- a/_damo_records.py
+++ b/_damo_records.py
@@ -694,6 +694,12 @@ def start_recording(tracepoint, file_path, file_format, file_permission,
             pipe, profile_pipe, kdamonds, poll_add_child_tasks,
             poll_add_mem_footprint)
 
+def wait_recording(handle):
+    while poll_target_pids(handle):
+        time.sleep(1)
+
+    _damon.wait_kdamonds_turned_off()
+
 def finish_recording(handle):
     try:
         handle.perf_pipe.send_signal(signal.SIGINT)