Internal change.

PiperOrigin-RevId: 610602779 Change-Id: I1c55c1a126ed603741f52a31fb5891af05345385
bazelbuild · Feb 27, 2024 · 6302db1 · 6302db1
1 parent 1e77944
commit 6302db1
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/src/main/java/com/google/devtools/build/lib/skyframe/serialization/ObjectCodecs.java b/src/main/java/com/google/devtools/build/lib/skyframe/serialization/ObjectCodecs.java
@@ -84,6 +84,16 @@ public SerializationResult<ByteString> serializeMemoizedAndBlocking(Object subje
  return serialize(subject, serializationContext.getMemoizingAndBlockingOnWriteContext());
  }
 
+ public SerializationResult<ByteString> serializeMemoizedAndBlocking(
+ Object subject, ImmutableClassToInstanceMap<?> dependencyOverrides)
+ throws SerializationException {
+ return serialize(
+ subject,
+ serializationContext
+ .withDependencyOverrides(dependencyOverrides)
+ .getMemoizingAndBlockingOnWriteContext());
+ }
+
  private static ByteString serializeToByteString(Object subject, SerializationContext context)
  throws SerializationException {
  ByteString.Output resultOut = ByteString.newOutput();
@@ -140,6 +150,14 @@ public Object deserializeMemoized(ByteString data) throws SerializationException
  return deserializeMemoized(data.newCodedInput());
  }
 
+ public Object deserializeMemoized(
+ ByteString data, ImmutableClassToInstanceMap<?> dependencyOverrides)
+ throws SerializationException {
+ return deserialize(
+ data.newCodedInput(),
+ deserializationContext.withDependencyOverrides(dependencyOverrides).getMemoizingContext());
+ }
+
  public Object deserializeMemoized(CodedInputStream codedIn) throws SerializationException {
  return deserialize(codedIn, deserializationContext.getMemoizingContext());
  }