org.opentripplanner.ext.restapi.mapping.AlertMapper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of application Show documentation

The newest version!

package org.opentripplanner.ext.restapi.mapping;

import java.time.Instant;
import java.util.Collection;
import java.util.Date;
import java.util.List;
import java.util.Locale;
import java.util.Optional;
import java.util.stream.Collectors;
import org.opentripplanner.ext.restapi.model.ApiAlert;
import org.opentripplanner.routing.alertpatch.TransitAlert;

public class AlertMapper {

  private final Locale locale;

  public AlertMapper(Locale locale) {
    this.locale = locale;
  }

  public List mapToApi(Collection alerts) {
    // Using {@code null} and not an empty set will minimize the JSON removing the
    // {@code alerts} from the result.
    if (alerts == null || alerts.isEmpty()) {
      return null;
    }

    return alerts.stream().map(this::mapToApi).collect(Collectors.toList());
  }

  ApiAlert mapToApi(TransitAlert domain) {
    ApiAlert api = new ApiAlert();
    api.alertHeaderText = domain.headerText().map(h -> h.toString(locale)).orElse(null);
    api.alertDescriptionText = domain.descriptionText().map(t -> t.toString(locale)).orElse(null);
    api.alertUrl = domain.url().map(u -> u.toString(locale)).orElse(null);

    api.effectiveStartDate = ofNullableInstant(domain.getEffectiveStartDate());
    api.effectiveEndDate = ofNullableInstant(domain.getEffectiveEndDate());

    return api;
  }

  private static Date ofNullableInstant(Instant instant) {
    return Optional.ofNullable(instant).map(Date::from).orElse(null);
  }
}