org.sonarsource.sonarlint.core.tracking.ServerIssueTracker Maven / Gradle / Ivy

/*
 * SonarLint Core - Implementation
 * Copyright (C) 2016-2021 SonarSource SA
 * mailto:info AT sonarsource DOT com
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */
package org.sonarsource.sonarlint.core.tracking;

import java.util.Collection;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;
import org.sonar.api.utils.log.Logger;
import org.sonar.api.utils.log.Loggers;
import org.sonarsource.sonarlint.core.client.api.connected.ConnectedSonarLintEngine;
import org.sonarsource.sonarlint.core.client.api.connected.ProjectBinding;
import org.sonarsource.sonarlint.core.client.api.connected.ServerConfiguration;
import org.sonarsource.sonarlint.core.client.api.connected.ServerIssue;
import org.sonarsource.sonarlint.core.client.api.exceptions.DownloadException;

public class ServerIssueTracker {

  private static final Logger LOGGER = Loggers.get(ServerIssueTracker.class);

  private final CachingIssueTracker issueTracker;

  public ServerIssueTracker(CachingIssueTracker issueTracker) {
    this.issueTracker = issueTracker;
  }

  public void update(ServerConfiguration serverConfiguration, ConnectedSonarLintEngine engine, ProjectBinding projectBinding, Collection fileKeys) {
    update(fileKeys, fileKey -> fetchServerIssues(serverConfiguration, engine, projectBinding, fileKey));
  }

  public void update(ConnectedSonarLintEngine engine, ProjectBinding projectBinding, Collection fileKeys) {
    update(fileKeys, fileKey -> engine.getServerIssues(projectBinding, fileKey));
  }

  private void update(Collection fileKeys, Function> issueGetter) {
    try {
      for (String fileKey : fileKeys) {
        List serverIssues = issueGetter.apply(fileKey);
        Collection serverIssuesTrackable = serverIssues.stream().map(ServerIssueTrackable::new).collect(Collectors.toList());
        issueTracker.matchAndTrackAsBase(fileKey, serverIssuesTrackable);
      }
    } catch (Exception e) {
      String message = "error while fetching and matching server issues";
      LOGGER.error(message, e);
    }
  }

  private static List fetchServerIssues(ServerConfiguration serverConfiguration, ConnectedSonarLintEngine engine,
    ProjectBinding projectBinding, String ideFilePath) {
    try {
      LOGGER.debug("fetchServerIssues projectKey=" + projectBinding.projectKey() + ", ideFilePath=" + ideFilePath);
      return engine.downloadServerIssues(serverConfiguration, projectBinding, ideFilePath);
    } catch (DownloadException e) {
      LOGGER.debug("Failed to download server issues", e);
      return engine.getServerIssues(projectBinding, ideFilePath);
    }
  }
}