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

   1 /*
   2  * SonarQube
   3  * Copyright (C) 2009-2019 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 java.util.HashMap;
  23 import java.util.List;
  24 import java.util.Map;
  25 import java.util.Optional;
  26 import javax.annotation.CheckForNull;
  27 import org.sonar.ce.task.projectanalysis.analysis.AnalysisMetadataHolder;
  28 import org.sonar.db.DbClient;
  29 import org.sonar.db.DbSession;
  30 import org.sonar.db.component.BranchDto;
  31 import org.sonar.db.component.ComponentDto;
  32
  33 import static com.google.common.base.Preconditions.checkState;
  34 import static org.sonar.db.component.ComponentDto.removeBranchAndPullRequestFromKey;
  35
  36 /**
  37  * Cache a map between component keys and uuids in the merge branch and optionally the target branch (for PRs, and only if this target branch is analyzed)
  38  */
  39 public class MergeAndTargetBranchComponentUuids {
  40   private final AnalysisMetadataHolder analysisMetadataHolder;
  41   private final DbClient dbClient;
  42   private Map<String, String> mergeBranchComponentsUuidsByKey;
  43   private Map<String, String> targetBranchComponentsUuidsByKey;
  44   private String mergeBranchName;
  45   private boolean hasMergeBranchAnalysis;
  46   private boolean hasTargetBranchAnalysis;
  47   @CheckForNull
  48   private String targetBranchUuid;
  49
  50   public MergeAndTargetBranchComponentUuids(AnalysisMetadataHolder analysisMetadataHolder, DbClient dbClient) {
  51     this.analysisMetadataHolder = analysisMetadataHolder;
  52     this.dbClient = dbClient;
  53   }
  54
  55   private void lazyInit() {
  56     if (mergeBranchComponentsUuidsByKey == null) {
  57       String mergeBranchUuid = analysisMetadataHolder.getBranch().getMergeBranchUuid();
  58
  59       mergeBranchComponentsUuidsByKey = new HashMap<>();
  60       targetBranchComponentsUuidsByKey = new HashMap<>();
  61
  62       try (DbSession dbSession = dbClient.openSession(false)) {
  63
  64         Optional<BranchDto> opt = dbClient.branchDao().selectByUuid(dbSession, mergeBranchUuid);
  65         checkState(opt.isPresent(), "Merge branch '%s' does not exist", mergeBranchUuid);
  66         mergeBranchName = opt.get().getKey();
  67
  68         initForMergeBranch(mergeBranchUuid, dbSession);
  69
  70         if (analysisMetadataHolder.isPullRequest()) {
  71           initForTargetBranch(mergeBranchUuid, dbSession);
  72         } else {
  73           hasTargetBranchAnalysis = false;
  74         }
  75       }
  76     }
  77   }
  78
  79   private void initForTargetBranch(String mergeBranchUuid, DbSession dbSession) {
  80     Optional<BranchDto> branchDtoOpt = dbClient.branchDao().selectByBranchKey(dbSession, analysisMetadataHolder.getProject().getUuid(),
  81       analysisMetadataHolder.getBranch().getTargetBranchName());
  82     targetBranchUuid = branchDtoOpt.map(BranchDto::getUuid).orElse(null);
  83     hasTargetBranchAnalysis = targetBranchUuid != null && dbClient.snapshotDao().selectLastAnalysisByRootComponentUuid(dbSession, targetBranchUuid).isPresent();
  84     if (hasTargetBranchAnalysis && !targetBranchUuid.equals(mergeBranchUuid)) {
  85       List<ComponentDto> targetComponents = dbClient.componentDao().selectByProjectUuid(targetBranchUuid, dbSession);
  86       for (ComponentDto dto : targetComponents) {
  87         targetBranchComponentsUuidsByKey.put(dto.getKey(), dto.uuid());
  88       }
  89     }
  90   }
  91
  92   private void initForMergeBranch(String mergeBranchUuid, DbSession dbSession) {
  93     hasMergeBranchAnalysis = dbClient.snapshotDao().selectLastAnalysisByRootComponentUuid(dbSession, mergeBranchUuid).isPresent();
  94
  95     if (hasMergeBranchAnalysis) {
  96       List<ComponentDto> components = dbClient.componentDao().selectByProjectUuid(mergeBranchUuid, dbSession);
  97       for (ComponentDto dto : components) {
  98         mergeBranchComponentsUuidsByKey.put(dto.getKey(), dto.uuid());
  99       }
 100     }
 101   }
 102
 103   public boolean hasMergeBranchAnalysis() {
 104     lazyInit();
 105     return hasMergeBranchAnalysis;
 106   }
 107
 108   public boolean hasTargetBranchAnalysis() {
 109     lazyInit();
 110     return hasTargetBranchAnalysis;
 111   }
 112
 113   public String getMergeBranchName() {
 114     lazyInit();
 115     return mergeBranchName;
 116   }
 117
 118   public boolean areTargetAndMergeBranchesDifferent() {
 119     lazyInit();
 120     return targetBranchUuid == null || !analysisMetadataHolder.getBranch().getMergeBranchUuid().equals(targetBranchUuid);
 121   }
 122
 123   @CheckForNull
 124   public String getMergeBranchComponentUuid(String dbKey) {
 125     lazyInit();
 126     String cleanComponentKey = removeBranchAndPullRequestFromKey(dbKey);
 127     return mergeBranchComponentsUuidsByKey.get(cleanComponentKey);
 128   }
 129
 130   @CheckForNull
 131   public String getTargetBranchComponentUuid(String dbKey) {
 132     lazyInit();
 133     String cleanComponentKey = removeBranchAndPullRequestFromKey(dbKey);
 134     return targetBranchComponentsUuidsByKey.get(cleanComponentKey);
 135   }
 136 }