org.opentripplanner.graph_builder.DataImportIssueStore Maven / Gradle / Ivy

Go to download
package org.opentripplanner.graph_builder;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class DataImportIssueStore {

  private static final Logger ISSUE_LOG = LoggerFactory.getLogger("DATA_IMPORT_ISSUES");

  private final List issues = new ArrayList<>();

  private final boolean storeIssues;

  public DataImportIssueStore(boolean storeIssues) {
    this.storeIssues = storeIssues;
  }

  public void add(DataImportIssue issue) {
    ISSUE_LOG.debug("{} - {}", issue.getType(), issue.getMessage());
    if (storeIssues) {
      this.issues.add(issue);
    }
  }

  public void add(String type, String message) {
    add(Issue.issue(type, message));
  }

  public void add(String type, String message, Object ... arguments) {
    add(Issue.issue(type, message, arguments));
  }

  void summarize() {
    Map issueCounts = issues
        .stream()
        .map(DataImportIssue::getType)
        .collect(Collectors.groupingBy(Function.identity(), Collectors.counting()));

    int maxLength = issueCounts.keySet().stream().mapToInt(String::length).max().orElse(10);
    final String FMT = "  - %-" + maxLength + "s  %,7d";

    ISSUE_LOG.info("Issue summary (number of each type):");

    issueCounts.keySet().stream().sorted().forEach(issueType ->
        ISSUE_LOG.info(String.format(FMT, issueType, issueCounts.get(issueType)))
    );
  }

  public List getIssues() {
    return this.issues;
  }
}