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

   1 /*
   2  * SonarQube
   3  * Copyright (C) 2009-2022 SonarSource SA
   4  * mailto:info AT sonarsource DOT com
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 3 of the License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19  */
  20 package org.sonar.ce.task.projectanalysis.component;
  21
  22 import com.google.common.collect.ImmutableMap;
  23 import java.util.Collection;
  24 import java.util.Collections;
  25 import java.util.List;
  26 import java.util.Map;
  27 import java.util.function.Function;
  28 import java.util.function.Predicate;
  29 import java.util.stream.Collectors;
  30 import javax.annotation.Nonnull;
  31 import org.sonar.api.utils.log.Loggers;
  32 import org.sonar.core.util.logs.Profiler;
  33
  34 import static com.google.common.collect.Iterables.concat;
  35 import static java.util.Objects.requireNonNull;
  36
  37 /**
  38  * This crawler make any number of {@link TypeAwareVisitor} or {@link PathAwareVisitor} defined in a list visit a component tree, component per component, in the order of the list
  39  */
  40 public class VisitorsCrawler implements ComponentCrawler {
  41
  42   private final boolean computeDuration;
  43   private final Map<ComponentVisitor, VisitorDuration> visitorCumulativeDurations;
  44   private final List<VisitorWrapper> preOrderVisitorWrappers;
  45   private final List<VisitorWrapper> postOrderVisitorWrappers;
  46
  47   public VisitorsCrawler(Collection<ComponentVisitor> visitors) {
  48     this(visitors, false);
  49   }
  50
  51   public VisitorsCrawler(Collection<ComponentVisitor> visitors, boolean computeDuration) {
  52     List<VisitorWrapper> visitorWrappers = visitors.stream().map(ToVisitorWrapper.INSTANCE).collect(Collectors.toList());
  53     this.preOrderVisitorWrappers = visitorWrappers.stream().filter(MathPreOrderVisitor.INSTANCE).collect(Collectors.toList());
  54     this.postOrderVisitorWrappers = visitorWrappers.stream().filter(MatchPostOrderVisitor.INSTANCE).collect(Collectors.toList());
  55     this.computeDuration = computeDuration;
  56     this.visitorCumulativeDurations = computeDuration ? visitors.stream().collect(Collectors.toMap(v -> v, v -> new VisitorDuration())) : Collections.emptyMap();
  57   }
  58
  59   public Map<ComponentVisitor, Long> getCumulativeDurations() {
  60     if (computeDuration) {
  61       return visitorCumulativeDurations.entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, e -> e.getValue().getDuration()));
  62     }
  63     return Collections.emptyMap();
  64   }
  65
  66   @Override
  67   public void visit(final Component component) {
  68     try {
  69       visitImpl(component);
  70     } catch (RuntimeException e) {
  71       VisitException.rethrowOrWrap(
  72         e,
  73         "Visit of Component {key=%s,type=%s} failed",
  74         component.getDbKey(), component.getType());
  75     }
  76   }
  77
  78   private void visitImpl(Component component) {
  79     MatchVisitorMaxDepth visitorMaxDepth = MatchVisitorMaxDepth.forComponent(component);
  80     List<VisitorWrapper> preOrderVisitorWrappersToExecute = preOrderVisitorWrappers.stream().filter(visitorMaxDepth).collect(Collectors.toList());
  81     List<VisitorWrapper> postOrderVisitorWrappersToExecute = postOrderVisitorWrappers.stream().filter(visitorMaxDepth).collect(Collectors.toList());
  82     if (preOrderVisitorWrappersToExecute.isEmpty() && postOrderVisitorWrappersToExecute.isEmpty()) {
  83       return;
  84     }
  85
  86     for (VisitorWrapper visitorWrapper : concat(preOrderVisitorWrappers, postOrderVisitorWrappers)) {
  87       visitorWrapper.beforeComponent(component);
  88     }
  89
  90     for (VisitorWrapper visitorWrapper : preOrderVisitorWrappersToExecute) {
  91       visitNode(component, visitorWrapper);
  92     }
  93
  94     visitChildren(component);
  95
  96     for (VisitorWrapper visitorWrapper : postOrderVisitorWrappersToExecute) {
  97       visitNode(component, visitorWrapper);
  98     }
  99
 100     for (VisitorWrapper visitorWrapper : concat(preOrderVisitorWrappersToExecute, postOrderVisitorWrappersToExecute)) {
 101       visitorWrapper.afterComponent(component);
 102     }
 103   }
 104
 105   private void visitChildren(Component component) {
 106     for (Component child : component.getChildren()) {
 107       visit(child);
 108     }
 109   }
 110
 111   private void visitNode(Component component, VisitorWrapper visitor) {
 112     Profiler profiler = Profiler.create(Loggers.get(visitor.getWrappedVisitor().getClass()))
 113       .startTrace("Visiting component {}", component.getDbKey());
 114     visitor.visitAny(component);
 115     switch (component.getType()) {
 116       case PROJECT:
 117         visitor.visitProject(component);
 118         break;
 119       case DIRECTORY:
 120         visitor.visitDirectory(component);
 121         break;
 122       case FILE:
 123         visitor.visitFile(component);
 124         break;
 125       case VIEW:
 126         visitor.visitView(component);
 127         break;
 128       case SUBVIEW:
 129         visitor.visitSubView(component);
 130         break;
 131       case PROJECT_VIEW:
 132         visitor.visitProjectView(component);
 133         break;
 134       default:
 135         throw new IllegalStateException(String.format("Unknown type %s", component.getType().name()));
 136     }
 137     long duration = profiler.stopTrace();
 138     incrementDuration(visitor, duration);
 139   }
 140
 141   private void incrementDuration(VisitorWrapper visitorWrapper, long duration) {
 142     if (computeDuration) {
 143       visitorCumulativeDurations.get(visitorWrapper.getWrappedVisitor()).increment(duration);
 144     }
 145   }
 146
 147   private enum ToVisitorWrapper implements Function<ComponentVisitor, VisitorWrapper> {
 148     INSTANCE;
 149
 150     @Override
 151     public VisitorWrapper apply(@Nonnull ComponentVisitor componentVisitor) {
 152       if (componentVisitor instanceof TypeAwareVisitor) {
 153         return new TypeAwareVisitorWrapper((TypeAwareVisitor) componentVisitor);
 154       } else if (componentVisitor instanceof PathAwareVisitor) {
 155         return new PathAwareVisitorWrapper((PathAwareVisitor) componentVisitor);
 156       } else {
 157         throw new IllegalArgumentException("Only TypeAwareVisitor and PathAwareVisitor can be used");
 158       }
 159     }
 160   }
 161
 162   private static class MatchVisitorMaxDepth implements Predicate<VisitorWrapper> {
 163     private static final Map<Component.Type, MatchVisitorMaxDepth> INSTANCES = buildInstances();
 164     private final Component.Type type;
 165
 166     private MatchVisitorMaxDepth(Component.Type type) {
 167       this.type = requireNonNull(type);
 168     }
 169
 170     private static Map<Component.Type, MatchVisitorMaxDepth> buildInstances() {
 171       ImmutableMap.Builder<Component.Type, MatchVisitorMaxDepth> builder = ImmutableMap.builder();
 172       for (Component.Type type : Component.Type.values()) {
 173         builder.put(type, new MatchVisitorMaxDepth(type));
 174       }
 175       return builder.build();
 176     }
 177
 178     public static MatchVisitorMaxDepth forComponent(Component component) {
 179       return INSTANCES.get(component.getType());
 180     }
 181
 182     @Override
 183     public boolean test(VisitorWrapper visitorWrapper) {
 184       CrawlerDepthLimit maxDepth = visitorWrapper.getMaxDepth();
 185       return maxDepth.isSameAs(type) || maxDepth.isDeeperThan(type);
 186     }
 187   }
 188
 189   private enum MathPreOrderVisitor implements Predicate<VisitorWrapper> {
 190     INSTANCE;
 191
 192     @Override
 193     public boolean test(VisitorWrapper visitorWrapper) {
 194       return visitorWrapper.getOrder() == ComponentVisitor.Order.PRE_ORDER;
 195     }
 196   }
 197
 198   private enum MatchPostOrderVisitor implements Predicate<VisitorWrapper> {
 199     INSTANCE;
 200
 201     @Override
 202     public boolean test(VisitorWrapper visitorWrapper) {
 203       return visitorWrapper.getOrder() == ComponentVisitor.Order.POST_ORDER;
 204     }
 205   }
 206
 207   private static final class VisitorDuration {
 208     private long duration = 0;
 209
 210     public void increment(long duration) {
 211       this.duration += duration;
 212     }
 213
 214     public long getDuration() {
 215       return duration;
 216     }
 217   }
 218 }