WIP: also pickle DigestProtoaevri/check_spawn_ci_working

1 files changed, 14 insertions, 0 deletions

diff --git a/src/buildstream/_scheduler/jobs/jobpickler.py b/src/buildstream/_scheduler/jobs/jobpickler.py
index 13ad2d0e3..4eb915e73 100644
--- a/src/buildstream/_scheduler/jobs/jobpickler.py
+++ b/src/buildstream/_scheduler/jobs/jobpickler.py
@@ -23,6 +23,7 @@ import io
 import pickle
 
 from ..._protos.buildstream.v2.artifact_pb2 import Artifact as ArtifactProto
+from ..._protos.build.bazel.remote.execution.v2.remote_execution_pb2 import Digest as DigestProto
 
 # BuildStream toplevel imports
 from ..._loader import Loader
@@ -100,6 +101,7 @@ def pickle_child_job(child_job, projects):
     for cls in source_classes:
         pickler.dispatch_table[cls] = _reduce_plugin
     pickler.dispatch_table[ArtifactProto] = _reduce_artifact_proto
+    pickler.dispatch_table[DigestProto] = _reduce_digest_proto
     pickler.dispatch_table[Loader] = _reduce_object
     pickler.dispatch_table[Messenger] = _reduce_object
 
@@ -127,6 +129,18 @@ def _new_artifact_proto_from_reduction_args(data):
     return instance
 
 
+def _reduce_digest_proto(instance):
+    assert isinstance(instance, DigestProto)
+    data = instance.SerializeToString()
+    return (_new_digest_proto_from_reduction_args, (data,))
+
+
+def _new_digest_proto_from_reduction_args(data):
+    instance = DigestProto()
+    instance.ParseFromString(data)
+    return instance
+
+
 def _reduce_plugin(plugin):
     project, meta_kind, state = plugin._get_args_for_child_job_pickling()
     assert project