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

   1 /*
   2  * SonarQube
   3  * Copyright (C) 2009-2023 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.projectexport.steps;
  21
  22 import com.sonarsource.governance.projectdump.protobuf.ProjectDump;
  23 import java.util.List;
  24 import org.slf4j.LoggerFactory;
  25 import org.sonar.ce.task.step.ComputationStep;
  26 import org.sonar.db.DbClient;
  27 import org.sonar.db.DbSession;
  28 import org.sonar.db.component.ProjectLinkDto;
  29 import org.sonar.db.project.ProjectExportMapper;
  30
  31 import static java.lang.String.format;
  32 import static org.apache.commons.lang.StringUtils.defaultString;
  33
  34 public class ExportLinksStep implements ComputationStep {
  35
  36   private final DbClient dbClient;
  37   private final ProjectHolder projectHolder;
  38   private final DumpWriter dumpWriter;
  39
  40   public ExportLinksStep(DbClient dbClient, ProjectHolder projectHolder, DumpWriter dumpWriter) {
  41     this.dbClient = dbClient;
  42     this.projectHolder = projectHolder;
  43     this.dumpWriter = dumpWriter;
  44   }
  45
  46   @Override
  47   public void execute(Context context) {
  48     long count = 0L;
  49
  50     try {
  51       try (DbSession dbSession = dbClient.openSession(false);
  52         StreamWriter<ProjectDump.Link> linksWriter = dumpWriter.newStreamWriter(DumpElement.LINKS)) {
  53         ProjectDump.Link.Builder builder = ProjectDump.Link.newBuilder();
  54         List<ProjectLinkDto> links = dbSession.getMapper(ProjectExportMapper.class).selectLinksForExport(projectHolder.projectDto().getUuid());
  55         for (ProjectLinkDto link : links) {
  56           builder
  57             .clear()
  58             .setUuid(link.getUuid())
  59             .setName(defaultString(link.getName()))
  60             .setHref(defaultString(link.getHref()))
  61             .setType(defaultString(link.getType()));
  62           linksWriter.write(builder.build());
  63           ++count;
  64         }
  65
  66         LoggerFactory.getLogger(getClass()).debug("{} links exported", count);
  67       }
  68     } catch (Exception e) {
  69       throw new IllegalStateException(format("Link export failed after processing %d link(s) successfully", count), e);
  70     }
  71   }
  72
  73   @Override
  74   public String getDescription() {
  75     return "Export links";
  76   }
  77 }