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

   1 /*
   2  * SonarQube
   3  * Copyright (C) 2009-2017 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.server.computation.task.projectanalysis.step;
  21
  22 import com.google.common.base.Optional;
  23 import javax.annotation.Nullable;
  24 import org.sonar.api.measures.CoreMetrics;
  25 import org.sonar.api.notifications.Notification;
  26 import org.sonar.api.utils.log.Logger;
  27 import org.sonar.api.utils.log.Loggers;
  28 import org.sonar.server.computation.task.projectanalysis.component.Component;
  29 import org.sonar.server.computation.task.projectanalysis.component.ComponentVisitor;
  30 import org.sonar.server.computation.task.projectanalysis.component.CrawlerDepthLimit;
  31 import org.sonar.server.computation.task.projectanalysis.component.DepthTraversalTypeAwareCrawler;
  32 import org.sonar.server.computation.task.projectanalysis.component.TreeRootHolder;
  33 import org.sonar.server.computation.task.projectanalysis.component.TypeAwareVisitorAdapter;
  34 import org.sonar.server.computation.task.projectanalysis.event.Event;
  35 import org.sonar.server.computation.task.projectanalysis.event.EventRepository;
  36 import org.sonar.server.computation.task.projectanalysis.measure.Measure;
  37 import org.sonar.server.computation.task.projectanalysis.measure.MeasureRepository;
  38 import org.sonar.server.computation.task.projectanalysis.measure.QualityGateStatus;
  39 import org.sonar.server.computation.task.projectanalysis.metric.Metric;
  40 import org.sonar.server.computation.task.projectanalysis.metric.MetricRepository;
  41 import org.sonar.server.computation.task.step.ComputationStep;
  42 import org.sonar.server.notification.NotificationService;
  43
  44 /**
  45  * This step must be executed after computation of quality gate measure {@link QualityGateMeasuresStep}
  46  */
  47 public class QualityGateEventsStep implements ComputationStep {
  48   private static final Logger LOGGER = Loggers.get(QualityGateEventsStep.class);
  49
  50   private final TreeRootHolder treeRootHolder;
  51   private final MetricRepository metricRepository;
  52   private final MeasureRepository measureRepository;
  53   private final EventRepository eventRepository;
  54   private final NotificationService notificationService;
  55
  56   public QualityGateEventsStep(TreeRootHolder treeRootHolder,
  57     MetricRepository metricRepository, MeasureRepository measureRepository, EventRepository eventRepository,
  58     NotificationService notificationService) {
  59     this.treeRootHolder = treeRootHolder;
  60     this.metricRepository = metricRepository;
  61     this.measureRepository = measureRepository;
  62     this.eventRepository = eventRepository;
  63     this.notificationService = notificationService;
  64   }
  65
  66   @Override
  67   public void execute() {
  68     new DepthTraversalTypeAwareCrawler(
  69       new TypeAwareVisitorAdapter(CrawlerDepthLimit.PROJECT, ComponentVisitor.Order.PRE_ORDER) {
  70         @Override
  71         public void visitProject(Component project) {
  72           executeForProject(project);
  73         }
  74       }).visit(treeRootHolder.getRoot());
  75   }
  76
  77   private void executeForProject(Component project) {
  78     Metric metric = metricRepository.getByKey(CoreMetrics.ALERT_STATUS_KEY);
  79     Optional<Measure> rawStatus = measureRepository.getRawMeasure(project, metric);
  80     if (!rawStatus.isPresent() || !rawStatus.get().hasQualityGateStatus()) {
  81       return;
  82     }
  83
  84     checkQualityGateStatusChange(project, metric, rawStatus.get().getQualityGateStatus());
  85   }
  86
  87   private void checkQualityGateStatusChange(Component project, Metric metric, QualityGateStatus rawStatus) {
  88     Optional<Measure> baseMeasure = measureRepository.getBaseMeasure(project, metric);
  89     if (!baseMeasure.isPresent()) {
  90       checkNewQualityGate(project, rawStatus);
  91       return;
  92     }
  93
  94     if (!baseMeasure.get().hasQualityGateStatus()) {
  95       LOGGER.warn(String.format("Previous Quality gate status for project %s is not a supported value. Can not compute Quality Gate event", project.getKey()));
  96       checkNewQualityGate(project, rawStatus);
  97       return;
  98     }
  99     QualityGateStatus baseStatus = baseMeasure.get().getQualityGateStatus();
 100
 101     if (baseStatus.getStatus() != rawStatus.getStatus()) {
 102       // The QualityGate status has changed
 103       String label = String.format("%s (was %s)", rawStatus.getStatus().getColorName(), baseStatus.getStatus().getColorName());
 104       createEvent(project, label, rawStatus.getText());
 105       boolean isNewKo = rawStatus.getStatus() == Measure.Level.OK;
 106       notifyUsers(project, label, rawStatus, isNewKo);
 107     }
 108   }
 109
 110   private void checkNewQualityGate(Component project, QualityGateStatus rawStatus) {
 111     if (rawStatus.getStatus() != Measure.Level.OK) {
 112       // There were no defined alerts before, so this one is a new one
 113       createEvent(project, rawStatus.getStatus().getColorName(), rawStatus.getText());
 114       notifyUsers(project, rawStatus.getStatus().getColorName(), rawStatus, true);
 115     }
 116   }
 117
 118   /**
 119    * @param label "Red (was Orange)"
 120    * @param rawStatus OK, WARN or ERROR + optional text
 121    */
 122   private void notifyUsers(Component project, String label, QualityGateStatus rawStatus, boolean isNewAlert) {
 123     Notification notification = new Notification("alerts")
 124       .setDefaultMessage(String.format("Alert on %s: %s", project.getName(), label))
 125       .setFieldValue("projectName", project.getName())
 126       .setFieldValue("projectKey", project.getKey())
 127       .setFieldValue("projectUuid", project.getUuid())
 128       .setFieldValue("alertName", label)
 129       .setFieldValue("alertText", rawStatus.getText())
 130       .setFieldValue("alertLevel", rawStatus.getStatus().toString())
 131       .setFieldValue("isNewAlert", Boolean.toString(isNewAlert));
 132     notificationService.deliver(notification);
 133   }
 134
 135   private void createEvent(Component project, String name, @Nullable String description) {
 136     eventRepository.add(project, Event.createAlert(name, null, description));
 137   }
 138
 139   @Override
 140   public String getDescription() {
 141     return "Generate Quality gate events";
 142   }
 143 }