app/models/chouette/vehicle_journey_at_stop.rb
module Chouette
class VehicleJourneyAtStop < ::ActiveRecord::Base
include Chouette::ForBoardingEnumerations
include Chouette::ForAlightingEnumerations
include ChecksumSupport
acts_as_copy_target
DAY_OFFSET_MAX = 2
@@day_offset_max = DAY_OFFSET_MAX
mattr_accessor :day_offset_max
belongs_to :stop_area, optional: true
belongs_to :stop_point
belongs_to :vehicle_journey
attr_accessor :_destroy, :dummy
validate :arrival_must_be_before_departure
def arrival_must_be_before_departure
# security against nil values
return unless arrival_time && departure_time
if TimeDuration.exceeds_gap?(4.hours, arrival_time, departure_time)
errors.add(
:arrival_time,
I18n.t("activerecord.errors.models.vehicle_journey_at_stop.arrival_must_be_before_departure")
)
end
end
validate :day_offset_must_be_within_range
after_initialize :set_virtual_attributes
def set_virtual_attributes
@_destroy = false
@dummy = false
end
def departure_time_of_day
@departure_time_of_day ||= TimeOfDay.unserialize(attributes_before_type_cast["departure_time"], day_offset: departure_day_offset)
end
def departure_local_time_of_day
@departure_local_time_of_day ||= departure_time_of_day&.with_utc_offset(time_zone_offset)
end
def departure_time_of_day=(time_of_day)
time_of_day = time_of_day.without_utc_offset
@departure_time_of_day = time_of_day
@departure_local_time_of_day = nil
self.departure_time = time_of_day.to_hms
self.departure_day_offset = time_of_day.day_offset
end
def arrival_time_of_day
@arrival_time_of_day ||= TimeOfDay.unserialize(attributes_before_type_cast["arrival_time"], day_offset: arrival_day_offset)
end
def arrival_local_time_of_day
@arrival_local_time_of_day ||= arrival_time_of_day&.with_utc_offset(time_zone_offset)
end
def arrival_time_of_day=(time_of_day)
time_of_day = time_of_day.without_utc_offset
@arrival_time_of_day = time_of_day
@arrival_local_time_of_day = nil
self.arrival_time = time_of_day.to_hms
self.arrival_day_offset = time_of_day.day_offset
end
%i[departure_time arrival_time].each do |attr|
define_method "#{attr}=" do |val|
self[attr] = convert_string_time_to_utc_time(val)
end
define_method attr do
self[attr]&.utc
end
# departure_local_time=
# arrival_local_time=
define_method "#{attr.to_s.split('_').join('_local_')}=" do |local_time|
self.send "#{attr}=", format_time(local_time(convert_string_time_to_utc_time(local_time), -time_zone_offset))
end
end
def convert_string_time_to_utc_time(val)
return unless val.present?
if val && val.is_a?(String)
tz = Time.zone
Time.zone = 'UTC'
val = Time.zone.parse val
Time.zone = tz
end
base_date = '2000/01/01 00:00:00 UTC'.to_time.utc
base_date.change hour: val.utc.hour, min: val.utc.min, sec: val.sec
end
def day_offset_must_be_within_range
if day_offset_outside_range?(arrival_day_offset)
errors.add(
:arrival_day_offset,
I18n.t(
'vehicle_journey_at_stops.errors.day_offset_must_not_exceed_max',
short_id: vehicle_journey&.get_objectid&.short_id,
max: day_offset_max
)
)
end
if day_offset_outside_range?(departure_day_offset)
errors.add(
:departure_day_offset,
I18n.t(
'vehicle_journey_at_stops.errors.day_offset_must_not_exceed_max',
short_id: vehicle_journey&.get_objectid&.short_id,
max: day_offset_max
)
)
end
end
# Compare absolute day offset value with day_offset_max (if defined)
def day_offset_outside_range?(offset)
return false if day_offset_max.blank?
offset ||= 0
offset < -1 || offset > day_offset_max
end
def checksum_attributes(_db_lookup = true)
[].tap do |attrs|
[self.departure_time, self.arrival_time].each do |time|
time = time&.utc
time = time && "%.2d:%.2d" % [time.hour, time.min]
attrs << time
end
attrs << self.departure_day_offset.to_s
attrs << self.arrival_day_offset.to_s
attrs << self.stop_area_id.to_s if self.stop_area_id.present?
end
end
def time_zone
ActiveSupport::TimeZone[stop_point&.stop_area_light&.time_zone || "UTC"]
end
def time_zone_offset
return 0 unless stop_point&.stop_area_light&.time_zone.present?
time_zone&.utc_offset || 0
end
########## DEPRECATED ##########
def departure
format_time departure_time.utc
end
def arrival
format_time arrival_time.utc
end
def departure_local_time offset=nil
local_time departure_time, offset
end
def arrival_local_time offset=nil
local_time arrival_time, offset
end
def departure_local
format_time departure_local_time
end
def arrival_local
format_time arrival_local_time
end
def departure_time_with_zone
departure_time.in_time_zone(time_zone).change(day: 1)
end
def arrival_time_with_zone
arrival_time.in_time_zone(time_zone).change(day: 1)
end
# Reads light VehicleJourneyAtStops for the current scope
#
# Uses a database cursor and returns Light::VehicleJourneyAtStop
def self.find_each_light(&block)
vehicle_journey_at_stop = Light::VehicleJourneyAtStop.new
each_row(block_size: 10_000) do |row|
vehicle_journey_at_stop.attributes = row
block.call vehicle_journey_at_stop
end
end
module Light
class VehicleJourneyAtStop
attr_accessor :id, :vehicle_journey_id, :stop_point_id, :stop_area_id
attr_accessor :arrival_time, :departure_time, :departure_day_offset, :arrival_day_offset
attr_accessor :for_boarding, :for_alighting, :checksum, :checksum_source
attr_writer :time_zone
def initialize(attributes = {})
self.attributes = attributes
@attributes = attributes
end
attr_accessor :attributes
def attributes=(attributes)
@id = attributes["id"]
@vehicle_journey_id = attributes["vehicle_journey_id"]
@stop_point_id = attributes["stop_point_id"]
@arrival_time = attributes["arrival_time"]
@departure_time = attributes["departure_time"]
@for_boarding = attributes["for_boarding"]
@for_alighting = attributes["for_alighting"]
@departure_day_offset = attributes["departure_day_offset"]
@arrival_day_offset = attributes["arrival_day_offset"]
@stop_area_id = attributes["stop_area_id"]
@checksum = attributes["checksum"]
@checksum_source = attributes["checksum_source"]
@attributes = attributes
end
def method_missing(name, *args)
stringified_name = name.to_s
if @attributes.has_key?(stringified_name)
return @attributes[stringified_name]
end
super
end
def respond_to?(name, *args)
return true if @attributes.has_key?(name.to_s)
super
end
def arrival_time_of_day
TimeOfDay.parse(arrival_time, day_offset: arrival_day_offset) if arrival_time
end
def arrival_local_time_of_day
arrival_time_of_day&.with_zone(time_zone)
end
def departure_time_of_day
TimeOfDay.parse(departure_time, day_offset: departure_day_offset) if departure_time
end
def departure_local_time_of_day
departure_time_of_day&.with_zone(time_zone)
end
end
end
def self.departures
departure_query = <<~SQL
(
SELECT vehicle_journey_at_stops.*
FROM #{departure_arrival_base_query}
WHERE vehicle_journey_at_stops.departure = 'true'
) AS vehicle_journey_at_stops
SQL
select('*').from(departure_query)
end
def self.arrivals
arrival_query = <<~SQL
(
SELECT vehicle_journey_at_stops.*
FROM #{departure_arrival_base_query}
WHERE vehicle_journey_at_stops.arrival = 'true'
) AS vehicle_journey_at_stops
SQL
select('*').from(arrival_query)
end
def self.departure_arrival_base_query
<<~SQL
(
SELECT
vehicle_journey_at_stops.*,
(LAG(vehicle_journey_at_stops.id) OVER vehicle_journey_stops) IS NULL AS departure,
(LEAD(vehicle_journey_at_stops.id) OVER vehicle_journey_stops) IS NULL AS arrival
FROM vehicle_journey_at_stops
INNER JOIN stop_points ON vehicle_journey_at_stops.stop_point_id = stop_points.id
WINDOW vehicle_journey_stops AS (
PARTITION BY vehicle_journey_id
ORDER BY stop_points.position
)
) vehicle_journey_at_stops
SQL
end
private
def local_time time, offset=nil
return nil unless time
(time + (offset || time_zone_offset)).utc
end
def format_time time
time.strftime "%H:%M" if time
end
################################
end
end