From 59317c93fbb33c525eff6af00d749beb6aa00dd0 Mon Sep 17 00:00:00 2001
From: kohsuke <kohsuke@71c3de6d-444a-0410-be80-ed276b4c234a>
Date: Thu, 24 Jun 2010 00:17:13 +0000
Subject: [PATCH] [HUDSON-6819] tweaking the implementation a bit so that the
 code never ever have multiple edges between the same src and dst.

Not that I was able to spot where the previous code did that, but this implementation is bit more straightforward.

git-svn-id: https://hudson.dev.java.net/svn/hudson/trunk/hudson/main@32256 71c3de6d-444a-0410-be80-ed276b4c234a
---
 .../java/hudson/model/DependencyGraph.java    | 43 ++++++++-----------
 1 file changed, 18 insertions(+), 25 deletions(-)

diff --git a/core/src/main/java/hudson/model/DependencyGraph.java b/core/src/main/java/hudson/model/DependencyGraph.java
index 280a4e1eab..6e14cc334c 100644
--- a/core/src/main/java/hudson/model/DependencyGraph.java
+++ b/core/src/main/java/hudson/model/DependencyGraph.java
@@ -83,8 +83,8 @@ import java.awt.image.BufferedImage;
  */
 public final class DependencyGraph implements Comparator<AbstractProject> {
 
-    private Map<AbstractProject, List<Dependency>> forward = new HashMap<AbstractProject, List<Dependency>>();
-    private Map<AbstractProject, List<Dependency>> backward = new HashMap<AbstractProject, List<Dependency>>();
+    private Map<AbstractProject, List<DependencyGroup>> forward = new HashMap<AbstractProject, List<DependencyGroup>>();
+    private Map<AbstractProject, List<DependencyGroup>> backward = new HashMap<AbstractProject, List<DependencyGroup>>();
 
     private boolean built;
 
@@ -136,8 +136,8 @@ public final class DependencyGraph implements Comparator<AbstractProject> {
         return get(backward,p,true);
     }
 
-    private List<AbstractProject> get(Map<AbstractProject, List<Dependency>> map, AbstractProject src, boolean up) {
-        List<Dependency> v = map.get(src);
+    private List<AbstractProject> get(Map<AbstractProject, List<DependencyGroup>> map, AbstractProject src, boolean up) {
+        List<DependencyGroup> v = map.get(src);
         if(v==null) return Collections.emptyList();
         List<AbstractProject> result = new ArrayList<AbstractProject>(v.size());
         for (Dependency d : v) result.add(up ? d.getUpstreamProject() : d.getDownstreamProject());
@@ -158,9 +158,9 @@ public final class DependencyGraph implements Comparator<AbstractProject> {
         return get(backward,p);
     }
 
-    private List<Dependency> get(Map<AbstractProject, List<Dependency>> map, AbstractProject src) {
-        List<Dependency> v = map.get(src);
-        if(v!=null) return v;
+    private List<Dependency> get(Map<AbstractProject, List<DependencyGroup>> map, AbstractProject src) {
+        List<DependencyGroup> v = map.get(src);
+        if(v!=null) return Collections.<Dependency>unmodifiableList(v);
         else        return Collections.emptyList();
     }
 
@@ -250,7 +250,7 @@ public final class DependencyGraph implements Comparator<AbstractProject> {
         return getTransitive(forward,src,false);
     }
 
-    private Set<AbstractProject> getTransitive(Map<AbstractProject, List<Dependency>> direction, AbstractProject src, boolean up) {
+    private Set<AbstractProject> getTransitive(Map<AbstractProject, List<DependencyGroup>> direction, AbstractProject src, boolean up) {
         Set<AbstractProject> visited = new HashSet<AbstractProject>();
         Stack<AbstractProject> queue = new Stack<AbstractProject>();
 
@@ -268,32 +268,26 @@ public final class DependencyGraph implements Comparator<AbstractProject> {
         return visited;
     }
 
-    private void add(Map<AbstractProject, List<Dependency>> map, AbstractProject key, Dependency dep) {
-        List<Dependency> set = map.get(key);
+    private void add(Map<AbstractProject, List<DependencyGroup>> map, AbstractProject key, Dependency dep) {
+        List<DependencyGroup> set = map.get(key);
         if(set==null) {
-            set = new ArrayList<Dependency>();
+            set = new ArrayList<DependencyGroup>();
             map.put(key,set);
         }
-        for (ListIterator<Dependency> it = set.listIterator(); it.hasNext();) {
-            Dependency d = it.next();
+        for (ListIterator<DependencyGroup> it = set.listIterator(); it.hasNext();) {
+            DependencyGroup d = it.next();
             // Check for existing edge that connects the same two projects:
             if (d.getUpstreamProject()==dep.getUpstreamProject() && d.getDownstreamProject()==dep.getDownstreamProject()) {
-                if (d.equals(dep))
-                    return; // identical with existing edge
-
-                if (d instanceof DependencyGroup)
-                    ((DependencyGroup)d).add(dep);
-                else
-                    it.set(new DependencyGroup(d,dep));
+                d.add(dep);
                 return;
             }
         }
         // Otherwise add to list:
-        set.add(dep);
+        set.add(new DependencyGroup(dep));
     }
 
-    private Map<AbstractProject, List<Dependency>> finalize(Map<AbstractProject, List<Dependency>> m) {
-        for (Entry<AbstractProject, List<Dependency>> e : m.entrySet()) {
+    private Map<AbstractProject, List<DependencyGroup>> finalize(Map<AbstractProject, List<DependencyGroup>> m) {
+        for (Entry<AbstractProject, List<DependencyGroup>> e : m.entrySet()) {
             Collections.sort( e.getValue(), NAME_COMPARATOR );
             e.setValue( Collections.unmodifiableList(e.getValue()) );
         }
@@ -480,10 +474,9 @@ public final class DependencyGraph implements Comparator<AbstractProject> {
     private static class DependencyGroup extends Dependency {
         private Set<Dependency> group = new LinkedHashSet<Dependency>();
 
-        DependencyGroup(Dependency first, Dependency second) {
+        DependencyGroup(Dependency first) {
             super(first.getUpstreamProject(), first.getDownstreamProject());
             group.add(first);
-            group.add(second);
         }
 
         void add(Dependency next) {
-- 
GitLab