add daemonize run binary option

Stepan Neretin · Stepan Neretin · commit 032f142140e6 · 2025-02-20T19:34:40.000+07:00
diff --git a/testgres/plugins/pg_probackup2/pg_probackup2/app.py b/testgres/plugins/pg_probackup2/pg_probackup2/app.py
@@ -45,6 +45,7 @@ class ProbackupApp:
 
     def __init__(self, test_class: unittest.TestCase,
                  pg_node, pb_log_path, test_env, auto_compress_alg, backup_dir, probackup_path=None):
+        self.process = None
         self.test_class = test_class
         self.pg_node = pg_node
         self.pb_log_path = pb_log_path
@@ -61,7 +62,7 @@ def __init__(self, test_class: unittest.TestCase,
         self.execution_time = None
 
     def run(self, command, gdb=False, old_binary=False, return_id=True, env=None,
-            skip_log_directory=False, expect_error=False, use_backup_dir=True):
+            skip_log_directory=False, expect_error=False, use_backup_dir=True, daemonize=False):
         """
         Run pg_probackup
         backup_dir: target directory for making backup
@@ -118,11 +119,35 @@ def run(self, command, gdb=False, old_binary=False, return_id=True, env=None,
                 logging.warning("pg_probackup gdb suspended, waiting gdb connection on localhost:{0}".format(gdb_port))
 
             start_time = time.time()
-            self.test_class.output = subprocess.check_output(
-                cmdline,
-                stderr=subprocess.STDOUT,
-                env=env
-            ).decode('utf-8', errors='replace')
+            if daemonize:
+
+                def stream_output(stream: subprocess.PIPE) -> None:
+                    for line in iter(stream.readline, ''):
+                        print(line)
+                    stream.close()
+
+                self.process = subprocess.Popen(
+                    cmdline,
+                    stdout=subprocess.PIPE,
+                    stderr=subprocess.PIPE,
+                    text=True,
+                    env=env
+                )
+                logging.info(f"Process started in background with PID: {self.process.pid}")
+
+                if self.process.stdout and self.process.stderr:
+                    stdout_thread = threading.Thread(target=stream_output, args=(self.process.stdout,))
+                    stderr_thread = threading.Thread(target=stream_output, args=(self.process.stderr,))
+
+                    stdout_thread.start()
+                    stderr_thread.start()
+                    return self.process.pid
+            else:
+                self.test_class.output = subprocess.check_output(
+                    cmdline,
+                    stderr=subprocess.STDOUT,
+                    env=env
+                ).decode('utf-8', errors='replace')
             end_time = time.time()
             self.execution_time = end_time - start_time
 
