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.projectanalysis.step;
  21
  22 import java.util.Objects;
  23 import java.util.function.Function;
  24 import java.util.stream.StreamSupport;
  25 import org.sonar.api.utils.System2;
  26 import org.sonar.ce.task.projectanalysis.analysis.AnalysisMetadataHolder;
  27 import org.sonar.ce.task.projectanalysis.component.Component;
  28 import org.sonar.ce.task.projectanalysis.component.TreeRootHolder;
  29 import org.sonar.ce.task.projectanalysis.event.Event;
  30 import org.sonar.ce.task.projectanalysis.event.EventRepository;
  31 import org.sonar.ce.task.step.ComputationStep;
  32 import org.sonar.core.util.UuidFactory;
  33 import org.sonar.db.DbClient;
  34 import org.sonar.db.DbSession;
  35 import org.sonar.db.event.EventDto;
  36
  37 public class PersistEventsStep implements ComputationStep {
  38
  39   private final DbClient dbClient;
  40   private final System2 system2;
  41   private final AnalysisMetadataHolder analysisMetadataHolder;
  42   private final TreeRootHolder treeRootHolder;
  43   private final EventRepository eventRepository;
  44   private final UuidFactory uuidFactory;
  45
  46   public PersistEventsStep(DbClient dbClient, System2 system2, TreeRootHolder treeRootHolder, AnalysisMetadataHolder analysisMetadataHolder,
  47     EventRepository eventRepository, UuidFactory uuidFactory) {
  48     this.dbClient = dbClient;
  49     this.system2 = system2;
  50     this.treeRootHolder = treeRootHolder;
  51     this.analysisMetadataHolder = analysisMetadataHolder;
  52     this.eventRepository = eventRepository;
  53     this.uuidFactory = uuidFactory;
  54   }
  55
  56   @Override
  57   public void execute(ComputationStep.Context context) {
  58     try (DbSession dbSession = dbClient.openSession(false)) {
  59       long analysisDate = analysisMetadataHolder.getAnalysisDate();
  60       new PersistEvent(dbSession, analysisDate).process(treeRootHolder.getRoot());
  61       dbSession.commit();
  62     }
  63   }
  64
  65   @Override
  66   public String getDescription() {
  67     return "Persist events";
  68   }
  69
  70   private class PersistEvent {
  71     private final DbSession session;
  72     private final long analysisDate;
  73
  74     PersistEvent(DbSession session, long analysisDate) {
  75       this.session = session;
  76       this.analysisDate = analysisDate;
  77     }
  78
  79     public void process(Component project) {
  80       processEvents(session, project, analysisDate);
  81       saveVersionEvent(session, project, analysisDate);
  82     }
  83
  84     private void processEvents(DbSession session, Component component, Long analysisDate) {
  85       Function<Event, EventDto> eventToEventDto = event -> newBaseEvent(component, analysisDate)
  86         .setName(event.getName())
  87         .setCategory(convertCategory(event.getCategory()))
  88         .setDescription(event.getDescription())
  89         .setData(event.getData());
  90       // FIXME bulk insert
  91       for (EventDto batchEventDto : StreamSupport.stream(eventRepository.getEvents().spliterator(), false).map(eventToEventDto).toList()) {
  92         dbClient.eventDao().insert(session, batchEventDto);
  93       }
  94     }
  95
  96     private void saveVersionEvent(DbSession session, Component component, Long analysisDate) {
  97       String projectVersion = component.getProjectAttributes().getProjectVersion();
  98       deletePreviousEventsHavingSameVersion(session, projectVersion, component);
  99       dbClient.eventDao().insert(session, newBaseEvent(component, analysisDate)
 100         .setName(projectVersion)
 101         .setCategory(EventDto.CATEGORY_VERSION));
 102     }
 103
 104     private void deletePreviousEventsHavingSameVersion(DbSession session, String version, Component component) {
 105       for (EventDto dto : dbClient.eventDao().selectByComponentUuid(session, component.getUuid())) {
 106         if (Objects.equals(dto.getCategory(), EventDto.CATEGORY_VERSION) && Objects.equals(dto.getName(), version)) {
 107           dbClient.eventDao().delete(session, dto.getUuid());
 108         }
 109       }
 110     }
 111
 112     private EventDto newBaseEvent(Component component, Long analysisDate) {
 113       return new EventDto()
 114         .setUuid(uuidFactory.create())
 115         .setAnalysisUuid(analysisMetadataHolder.getUuid())
 116         .setComponentUuid(component.getUuid())
 117         .setCreatedAt(system2.now())
 118         .setDate(analysisDate);
 119     }
 120
 121     private String convertCategory(Event.Category category) {
 122       return switch (category) {
 123         case ALERT -> EventDto.CATEGORY_ALERT;
 124         case PROFILE -> EventDto.CATEGORY_PROFILE;
 125         case ISSUE_DETECTION -> EventDto.CATEGORY_ISSUE_DETECTION;
 126       };
 127     }
 128
 129   }
 130 }