worker: add EFA support

1 files changed, 101 insertions, 1 deletions

diff --git a/lib/Travelynx/Model/InTransit.pm b/lib/Travelynx/Model/InTransit.pm
index 662df54..4f76468 100644
--- a/lib/Travelynx/Model/InTransit.pm
+++ b/lib/Travelynx/Model/InTransit.pm
@@ -176,7 +176,7 @@ sub add {
 				train_type         => $journey->type // q{},
 				train_line         => $journey->line,
 				train_no           => $journey->number // q{},
-				train_id           => $journey->id,
+				train_id           => $opt{trip_id},
 				sched_departure    => $stop->sched_dep,
 				real_departure     => $stop->rt_dep // $stop->sched_dep,
 				route              => $json->encode( \@route ),
@@ -995,6 +995,43 @@ sub update_departure_dbris {
 	);
 }
 
+sub update_departure_efa {
+	my ( $self, %opt ) = @_;
+	my $uid     = $opt{uid};
+	my $db      = $opt{db} // $self->{pg}->db;
+	my $dep_eva = $opt{dep_eva};
+	my $arr_eva = $opt{arr_eva};
+	my $journey = $opt{journey};
+	my $stop    = $opt{stop};
+	my $json    = JSON->new;
+
+	my $res_h = $db->select( 'in_transit', ['data'], { user_id => $uid } )
+	  ->expand->hash;
+	my $ephemeral_data = $res_h ? $res_h->{data} : {};
+	if ( $stop->rt_dep ) {
+		$ephemeral_data->{rt} = 1;
+	}
+
+	say "UPDATE dep WHERE $uid $opt{trip_id} $dep_eva $arr_eva";
+
+	# selecting on user_id and train_no avoids a race condition if a user checks
+	# into a new train while we are fetching data for their previous journey. In
+	# this case, the new train would receive data from the previous journey.
+	$db->update(
+		'in_transit',
+		{
+			data           => $json->encode($ephemeral_data),
+			real_departure => $stop->rt_dep,
+		},
+		{
+			user_id             => $uid,
+			train_id            => $opt{trip_id},
+			checkin_station_id  => $dep_eva,
+			checkout_station_id => $arr_eva,
+		}
+	);
+}
+
 sub update_departure_motis {
 	my ( $self, %opt ) = @_;
 	my $uid      = $opt{uid};
@@ -1188,6 +1225,69 @@ sub update_arrival_dbris {
 	);
 }
 
+sub update_arrival_efa {
+	my ( $self, %opt ) = @_;
+	my $uid     = $opt{uid};
+	my $db      = $opt{db} // $self->{pg}->db;
+	my $dep_eva = $opt{dep_eva};
+	my $arr_eva = $opt{arr_eva};
+	my $journey = $opt{journey};
+	my $stop    = $opt{stop};
+	my $json    = JSON->new;
+
+	my $res_h
+	  = $db->select( 'in_transit', [ 'data', 'route' ], { user_id => $uid } )
+	  ->expand->hash;
+	my $ephemeral_data = $res_h ? $res_h->{data}  : {};
+	my $old_route      = $res_h ? $res_h->{route} : [];
+
+	if ( $stop->rt_arr ) {
+		$ephemeral_data->{rt} = 1;
+	}
+
+	my @route;
+	for my $j_stop ( $journey->route ) {
+		push(
+			@route,
+			[
+				$j_stop->full_name,
+				$j_stop->id_num,
+				{
+					sched_arr => _epoch( $j_stop->sched_arr ),
+					sched_dep => _epoch( $j_stop->sched_dep ),
+					rt_arr    => _epoch( $j_stop->rt_arr ),
+					rt_dep    => _epoch( $j_stop->rt_dep ),
+					arr_delay => $j_stop->arr_delay,
+					dep_delay => $j_stop->dep_delay,
+					efa_load  => $j_stop->occupancy,
+					lat       => $j_stop->latlon->[0],
+					lon       => $j_stop->latlon->[1],
+				}
+			]
+		);
+	}
+
+	say "UPDATE arr WHERE $uid $opt{trip_id} $dep_eva $arr_eva";
+
+	# selecting on user_id and train_no avoids a race condition if a user checks
+	# into a new train while we are fetching data for their previous journey. In
+	# this case, the new train would receive data from the previous journey.
+	$db->update(
+		'in_transit',
+		{
+			data         => $json->encode($ephemeral_data),
+			real_arrival => $stop->rt_arr,
+			route        => $json->encode( [@route] ),
+		},
+		{
+			user_id             => $uid,
+			train_id            => $opt{trip_id},
+			checkin_station_id  => $dep_eva,
+			checkout_station_id => $arr_eva,
+		}
+	);
+}
+
 sub update_arrival_motis {
 	my ( $self, %opt ) = @_;
 	my $uid      = $opt{uid};