_basecache.py: Add has_open_grpc_channels() method

1 files changed, 12 insertions, 0 deletions

diff --git a/src/buildstream/_basecache.py b/src/buildstream/_basecache.py
index 431a7304e..9ad6c1277 100644
--- a/src/buildstream/_basecache.py
+++ b/src/buildstream/_basecache.py
@@ -59,6 +59,18 @@ class BaseCache():
         self._has_fetch_remotes = False
         self._has_push_remotes = False
 
+    # has_open_grpc_channels():
+    #
+    # Return whether there are gRPC channel instances. This is used to safeguard
+    # against fork() with open gRPC channels.
+    #
+    def has_open_grpc_channels(self):
+        for project_remotes in self._remotes.values():
+            for remote in project_remotes:
+                if remote.channel:
+                    return True
+        return False
+
     # release_resources():
     #
     # Release resources used by BaseCache.