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

   1 /*
   2  * SonarQube
   3  * Copyright (C) 2009-2016 SonarSource SA
   4  * mailto:contact 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.taskprocessor.report;
  21
  22 import java.util.Collections;
  23 import java.util.Set;
  24 import javax.annotation.CheckForNull;
  25 import org.sonar.core.platform.ComponentContainer;
  26 import org.sonar.db.ce.CeTaskTypes;
  27 import org.sonar.server.computation.container.ComputeEngineContainer;
  28 import org.sonar.server.computation.container.ContainerFactory;
  29 import org.sonar.ce.queue.CeTask;
  30 import org.sonar.ce.queue.CeTaskResult;
  31 import org.sonar.server.computation.step.ComputationStepExecutor;
  32 import org.sonar.ce.taskprocessor.CeTaskProcessor;
  33 import org.sonar.server.computation.taskprocessor.TaskResultHolder;
  34 import org.sonar.server.devcockpit.DevCockpitBridge;
  35
  36 public class ReportTaskProcessor implements CeTaskProcessor {
  37
  38   private static final Set<String> HANDLED_TYPES = Collections.singleton(CeTaskTypes.REPORT);
  39
  40   private final ContainerFactory containerFactory;
  41   private final ComponentContainer serverContainer;
  42   @CheckForNull
  43   private final DevCockpitBridge devCockpitBridge;
  44
  45   /**
  46    * Used when Developer Cockpit plugin is installed
  47    */
  48   public ReportTaskProcessor(ContainerFactory containerFactory, ComponentContainer serverContainer, DevCockpitBridge devCockpitBridge) {
  49     this.containerFactory = containerFactory;
  50     this.serverContainer = serverContainer;
  51     this.devCockpitBridge = devCockpitBridge;
  52   }
  53
  54   /**
  55    * Used when Developer Cockpit plugin is not installed
  56    */
  57   public ReportTaskProcessor(ContainerFactory containerFactory, ComponentContainer serverContainer) {
  58     this.containerFactory = containerFactory;
  59     this.serverContainer = serverContainer;
  60     this.devCockpitBridge = null;
  61   }
  62
  63   @Override
  64   public Set<String> getHandledCeTaskTypes() {
  65     return HANDLED_TYPES;
  66   }
  67
  68   @Override
  69   public CeTaskResult process(CeTask task) {
  70     ComputeEngineContainer ceContainer = containerFactory.create(serverContainer, task, devCockpitBridge);
  71     try {
  72       ceContainer.getComponentByType(ComputationStepExecutor.class).execute();
  73       return ceContainer.getComponentByType(TaskResultHolder.class).getResult();
  74     } finally {
  75       ceContainer.cleanup();
  76     }
  77   }
  78 }