org.opentripplanner.model.calendar.CalendarServiceData Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of otp Show documentation
The OpenTripPlanner multimodal journey planning system
The newest version!
/* This file is based on code copied from project OneBusAway, see the LICENSE file for further information. */
package org.opentripplanner.model.calendar;

import java.io.Serializable;
import java.time.LocalDate;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import org.opentripplanner.framework.time.ServiceDateUtils;
import org.opentripplanner.transit.model.framework.FeedScopedId;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class CalendarServiceData implements Serializable {

  private static final String CAL_SERVICE_FEED_ID = "CSID";
  private static final Logger LOG = LoggerFactory.getLogger(CalendarServiceData.class);

  private final Map> serviceDatesByServiceId = new HashMap<>();

  private final Map> serviceIdsByDate = new HashMap<>();

  public Set getServiceIds() {
    return Collections.unmodifiableSet(serviceDatesByServiceId.keySet());
  }

  public List getServiceDatesForServiceId(FeedScopedId serviceId) {
    return serviceDatesByServiceId.get(serviceId);
  }

  public Set getServiceIdsForDate(LocalDate date) {
    Set serviceIds = serviceIdsByDate.get(date);
    if (serviceIds == null) {
      serviceIds = new HashSet<>();
    }
    return serviceIds;
  }

  public void putServiceDatesForServiceId(FeedScopedId serviceId, List dates) {
    List serviceDates = sortedImmutableList(dates);
    serviceDatesByServiceId.put(serviceId, serviceDates);
    addDatesToServiceIdsByDate(serviceId, serviceDates);
  }

  /**
   * TODO OTP2 - This is NOT THREAD-SAFE and is used in the real-time updaters, we need to fix
   *           - this when doing the issue #3030.
   */
  public FeedScopedId getOrCreateServiceIdForDate(LocalDate serviceDate) {
    FeedScopedId serviceId = new FeedScopedId(
      CAL_SERVICE_FEED_ID,
      ServiceDateUtils.asCompactString(serviceDate)
    );
    if (serviceDatesByServiceId.containsKey(serviceId)) {
      return serviceId;
    }
    serviceDatesByServiceId.put(serviceId, List.of(serviceDate));
    serviceIdsByDate.computeIfAbsent(serviceDate, d -> new HashSet<>()).add(serviceId);

    LOG.info("Adding serviceId {} to CalendarService", serviceId);

    return serviceId;
  }

  public void add(CalendarServiceData other) {
    for (FeedScopedId serviceId : other.serviceDatesByServiceId.keySet()) {
      putServiceDatesForServiceId(serviceId, other.serviceDatesByServiceId.get(serviceId));
    }
  }

  public Optional getFirstDate() {
    return serviceIdsByDate.keySet().stream().min(LocalDate::compareTo);
  }

  public Optional getLastDate() {
    return serviceIdsByDate.keySet().stream().max(LocalDate::compareTo);
  }

  /* private methods */

  private static  List sortedImmutableList(Collection c) {
    return c.stream().sorted().toList();
  }

  private void addDatesToServiceIdsByDate(FeedScopedId serviceId, List serviceDates) {
    for (LocalDate serviceDate : serviceDates) {
      Set serviceIds = serviceIdsByDate.computeIfAbsent(
        serviceDate,
        k -> new HashSet<>()
      );
      serviceIds.add(serviceId);
    }
  }
}