Add the method InMemoryNodeEntry#getGroupedDirectDeps so callers can access the ordered list of grouped deps.

--
MOS_MIGRATED_REVID=87969259

1 files changed, 10 insertions, 0 deletions

diff --git a/src/main/java/com/google/devtools/build/skyframe/InMemoryNodeEntry.java b/src/main/java/com/google/devtools/build/skyframe/InMemoryNodeEntry.java
index 2872c90601..df9d0c6245 100644
--- a/src/main/java/com/google/devtools/build/skyframe/InMemoryNodeEntry.java
+++ b/src/main/java/com/google/devtools/build/skyframe/InMemoryNodeEntry.java
@@ -182,6 +182,16 @@ public class InMemoryNodeEntry implements NodeEntry, Serializable {
     return GroupedList.<SkyKey>create(directDeps).toSet();
   }
 
+  /**
+   * If {@code isDone()}, returns the ordered list of sets of grouped direct dependencies that were
+   * added in {@link addTemporaryDirectDeps}.
+   */
+  public synchronized Iterable<Iterable<SkyKey>> getGroupedDirectDeps() {
+    assertKeepEdges();
+    Preconditions.checkState(isDone(), "no deps until done. ValueEntry: %s", this);
+    return GroupedList.<SkyKey>create(directDeps);
+  }
+
   @Override
   @Nullable
   public synchronized ErrorInfo getErrorInfo() {