utils.py: Added _is_main_process()

A helper function to identify if we are running in the main process
or not.

1 files changed, 12 insertions, 0 deletions

diff --git a/buildstream/utils.py b/buildstream/utils.py
index 40705346b..5b246ced4 100644
--- a/buildstream/utils.py
+++ b/buildstream/utils.py
@@ -536,6 +536,18 @@ def save_file_atomic(filename, mode='w', *, buffering=-1, encoding=None,
 # to distinguish between a kwarg set to None and an unset kwarg.
 _sentinel = object()
 
+# Main process pid
+_main_pid = os.getpid()
+
+
+# _is_main_process()
+#
+# Return whether we are in the main process or not.
+#
+def _is_main_process():
+    assert _main_pid is not None
+    return os.getpid() == _main_pid
+
 
 # Recursively remove directories, ignoring file permissions as much as
 # possible.