source.dussan.org Git - archiva.git/blob

   1 package org.apache.maven.archiva.dependency.graph.walk;
   2
   3 /*
   4  * Licensed to the Apache Software Foundation (ASF) under one
   5  * or more contributor license agreements.  See the NOTICE file
   6  * distributed with this work for additional information
   7  * regarding copyright ownership.  The ASF licenses this file
   8  * to you under the Apache License, Version 2.0 (the
   9  * "License"); you may not use this file except in compliance
  10  * with the License.  You may obtain a copy of the License at
  11  *
  12  *  http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing,
  15  * software distributed under the License is distributed on an
  16  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  17  * KIND, either express or implied.  See the License for the
  18  * specific language governing permissions and limitations
  19  * under the License.
  20  */
  21
  22 import org.apache.commons.collections.Predicate;
  23 import org.apache.commons.collections.functors.NotPredicate;
  24 import org.apache.maven.archiva.dependency.graph.DependencyGraph;
  25 import org.apache.maven.archiva.dependency.graph.DependencyGraphEdge;
  26 import org.apache.maven.archiva.dependency.graph.DependencyGraphNode;
  27 import org.apache.maven.archiva.dependency.graph.functors.EdgeDisabledPredicate;
  28 import org.apache.maven.archiva.model.ArtifactReference;
  29
  30 import java.util.HashMap;
  31 import java.util.Iterator;
  32 import java.util.Map;
  33
  34 /**
  35  * Perform a walk of the graph using the DepthFirstSearch algorithm.
  36  *
  37  * NOTE: Default edgePredicate is to NOT traverse disabled edges.
  38  *
  39  * @version $Id$
  40  */
  41 public class WalkDepthFirstSearch
  42     implements DependencyGraphWalker
  43 {
  44     private Map nodeVisitStates = new HashMap();
  45
  46     private Predicate edgePredicate;
  47
  48     public WalkDepthFirstSearch()
  49     {
  50         this.edgePredicate = NotPredicate.getInstance( new EdgeDisabledPredicate() );
  51     }
  52
  53     public Predicate getEdgePredicate()
  54     {
  55         return this.edgePredicate;
  56     }
  57
  58     public void setEdgePredicate( Predicate edgePredicate )
  59     {
  60         this.edgePredicate = edgePredicate;
  61     }
  62
  63     public Integer getNodeVisitState( DependencyGraphNode node )
  64     {
  65         if ( node == null )
  66         {
  67             return SEEN;
  68         }
  69
  70         return (Integer) nodeVisitStates.get( node.getArtifact() );
  71     }
  72
  73     public Integer getNodeVisitState( ArtifactReference artifact )
  74     {
  75         return (Integer) nodeVisitStates.get( artifact );
  76     }
  77
  78     public void setNodeVisitState( DependencyGraphNode node, Integer state )
  79     {
  80         this.nodeVisitStates.put( node.getArtifact(), state );
  81     }
  82
  83     public void setNodeVisitState( ArtifactReference artifact, Integer state )
  84     {
  85         this.nodeVisitStates.put( artifact, state );
  86     }
  87
  88     private void visitEdge( DependencyGraph graph, DependencyGraphEdge e, DependencyGraphVisitor visitor )
  89     {
  90         visitor.discoverEdge( e );
  91
  92         DependencyGraphNode node = graph.getNode( e.getNodeTo() );
  93
  94         if ( getNodeVisitState( node ) == UNSEEN )
  95         {
  96             visitNode( graph, node, visitor );
  97         }
  98
  99         visitor.finishEdge( e );
 100     }
 101
 102     private void visitNode( DependencyGraph graph, DependencyGraphNode node, DependencyGraphVisitor visitor )
 103     {
 104         setNodeVisitState( node, PROCESSING );
 105
 106         visitor.discoverNode( node );
 107
 108         Iterator edges = graph.getEdgesFrom( node ).iterator();
 109         while ( edges.hasNext() )
 110         {
 111             DependencyGraphEdge e = (DependencyGraphEdge) edges.next();
 112             if ( this.edgePredicate.evaluate( e ) )
 113             {
 114                 visitEdge( graph, e, visitor );
 115             }
 116         }
 117
 118         visitor.finishNode( node );
 119
 120         setNodeVisitState( node, SEEN );
 121     }
 122
 123     public void visit( DependencyGraph graph, DependencyGraphVisitor visitor )
 124     {
 125         visit( graph, graph.getRootNode(), visitor );
 126     }
 127
 128     public void visit( DependencyGraph graph, DependencyGraphNode startNode, DependencyGraphVisitor visitor )
 129     {
 130         nodeVisitStates.clear();
 131
 132         Iterator nodes = graph.getNodes().iterator();
 133         while ( nodes.hasNext() )
 134         {
 135             setNodeVisitState( (DependencyGraphNode) nodes.next(), UNSEEN );
 136         }
 137
 138         visitor.discoverGraph( graph );
 139
 140         visitNode( graph, startNode, visitor );
 141
 142         visitor.finishGraph( graph );
 143     }
 144 }