travelynx/lib/Travelynx/Command/work.pm
2024-06-23 12:47:41 +02:00

348 lines
9 KiB
Perl

package Travelynx::Command::work;
# Copyright (C) 2020-2023 Birte Kristina Friesel
#
# SPDX-License-Identifier: AGPL-3.0-or-later
use Mojo::Base 'Mojolicious::Command';
use Mojo::Promise;
use DateTime;
use JSON;
use List::Util;
has description => 'Update real-time data of active journeys';
has usage => sub { shift->extract_usage };
sub run {
my ($self) = @_;
my $now = DateTime->now( time_zone => 'Europe/Berlin' );
my $checkin_deadline = $now->clone->subtract( hours => 48 );
my $json = JSON->new;
if ( -e 'maintenance' ) {
$self->app->log->debug('work: "maintenance" file found, aborting');
return;
}
my $num_incomplete = $self->app->in_transit->delete_incomplete_checkins(
earlier_than => $checkin_deadline );
if ($num_incomplete) {
$self->app->log->debug("Removed ${num_incomplete} incomplete checkins");
}
my $errors = 0;
for my $entry ( $self->app->in_transit->get_all_active ) {
if ( -e 'maintenance' ) {
$self->app->log->debug('work: "maintenance" file found, aborting');
return;
}
my $uid = $entry->{user_id};
my $dep = $entry->{dep_eva};
my $arr = $entry->{arr_eva};
my $train_id = $entry->{train_id};
if ( $train_id =~ m{[|]} ) {
$self->app->hafas->get_journey_p( trip_id => $train_id )->then(
sub {
my ($journey) = @_;
my $found_dep;
my $found_arr;
for my $stop ( $journey->route ) {
if ( $stop->loc->eva == $dep ) {
$found_dep = $stop;
}
if ( $arr and $stop->loc->eva == $arr ) {
$found_arr = $stop;
last;
}
}
if ( not $found_dep ) {
$self->app->log->debug(
"Did not find $dep within journey $train_id");
return;
}
if ( $found_dep->rt_dep ) {
$self->app->in_transit->update_departure_hafas(
uid => $uid,
journey => $journey,
stop => $found_dep,
dep_eva => $dep,
arr_eva => $arr
);
if ( $journey->class <= 16
and $found_dep->rt_dep->epoch > $now->epoch )
{
$self->app->add_wagonorder( $uid, 1, $train_id,
$found_dep->sched_dep, $journey->number );
$self->app->add_stationinfo( $uid, 1, $journey->id,
$found_dep->loc->eva );
}
}
if ( $found_arr and $found_arr->rt_arr ) {
$self->app->in_transit->update_arrival_hafas(
uid => $uid,
journey => $journey,
stop => $found_arr,
dep_eva => $dep,
arr_eva => $arr
);
if ( $journey->class <= 16
and $found_arr->rt_arr->epoch - $now->epoch < 600 )
{
$self->app->add_wagonorder( $uid, 0, $train_id,
$found_dep->sched_dep, $journey->number );
$self->app->add_stationinfo( $uid, 0, $journey->id,
$found_dep->loc->eva, $found_arr->loc->eva );
}
}
}
)->catch(
sub {
my ($err) = @_;
if ( $err =~ m{svcResL\[0\][.]err is (?:FAIL|PARAMETER)$} )
{
# HAFAS do be weird. These are not actionable.
$self->app->log->debug("work($uid)/journey: $err");
}
else {
$self->app->log->error("work($uid)/journey: $err");
}
}
)->wait;
if ( $arr
and $entry->{real_arr_ts}
and $now->epoch - $entry->{real_arr_ts} > 600 )
{
$self->app->checkout_p(
station => $arr,
force => 2,
dep_eva => $dep,
arr_eva => $arr,
uid => $uid
)->wait;
}
next;
}
# Note: IRIS data is not always updated in real-time. Both departure and
# arrival delays may take several minutes to appear, especially in case
# of large-scale disturbances. We work around this by continuing to
# update departure data for up to 15 minutes after departure and
# delaying automatic checkout by at least 10 minutes.
eval {
if ( $now->epoch - $entry->{real_dep_ts} < 900 ) {
my $status = $self->app->iris->get_departures(
station => $dep,
lookbehind => 30,
lookahead => 30
);
if ( $status->{errstr} ) {
die("get_departures($dep): $status->{errstr}\n");
}
my ($train) = List::Util::first { $_->train_id eq $train_id }
@{ $status->{results} };
if ( not $train ) {
$self->app->log->debug(
"could not find train $train_id at $dep\n");
return;
}
$self->app->in_transit->update_departure(
uid => $uid,
train => $train,
dep_eva => $dep,
arr_eva => $arr,
route => [ $self->app->iris->route_diff($train) ]
);
if ( $train->departure_is_cancelled and $arr ) {
my $checked_in
= $self->app->in_transit->update_departure_cancelled(
uid => $uid,
train => $train,
dep_eva => $dep,
arr_eva => $arr,
);
# depending on the amount of users in transit, some time may
# have passed between fetching $entry from the database and
# now. Only check out if the user is still checked into this
# train.
if ($checked_in) {
# check out (adds a cancelled journey and resets journey state
# to checkin
$self->app->checkout_p(
station => $arr,
force => 2,
dep_eva => $dep,
arr_eva => $arr,
uid => $uid
)->wait;
}
}
else {
$self->app->add_route_timestamps( $uid, $train, 1 );
$self->app->add_wagonorder( $uid, 1, $train->train_id,
$train->sched_departure, $train->train_no );
$self->app->add_stationinfo( $uid, 1, $train->train_id,
$dep, $arr );
}
}
};
if ($@) {
$errors += 1;
$self->app->log->error("work($uid)/departure: $@");
}
eval {
if (
$arr
and ( not $entry->{real_arr_ts}
or $now->epoch - $entry->{real_arr_ts} < 600 )
)
{
my $status = $self->app->iris->get_departures(
station => $arr,
lookbehind => 20,
lookahead => 220
);
if ( $status->{errstr} ) {
die("get_departures($arr): $status->{errstr}\n");
}
# Note that a train may pass the same station several times.
# Notable example: S41 / S42 ("Ringbahn") both starts and
# terminates at Berlin Südkreuz
my ($train) = List::Util::first {
$_->train_id eq $train_id
and $_->sched_arrival
and $_->sched_arrival->epoch > $entry->{sched_dep_ts}
}
@{ $status->{results} };
$train //= List::Util::first { $_->train_id eq $train_id }
@{ $status->{results} };
if ( not $train ) {
# If we haven't seen the train yet, its arrival is probably
# too far in the future. This is not critical.
return;
}
my $checked_in = $self->app->in_transit->update_arrival(
uid => $uid,
train => $train,
route => [ $self->app->iris->route_diff($train) ],
dep_eva => $dep,
arr_eva => $arr,
);
if ( $checked_in and $train->arrival_is_cancelled ) {
# check out (adds a cancelled journey and resets journey state
# to destination selection)
$self->app->checkout_p(
station => $arr,
force => 0,
dep_eva => $dep,
arr_eva => $arr,
uid => $uid
)->wait;
}
else {
$self->app->add_route_timestamps(
$uid, $train, 0,
(
defined $entry->{real_arr_ts}
and $now->epoch > $entry->{real_arr_ts}
) ? 1 : 0
);
$self->app->add_wagonorder( $uid, 0, $train->train_id,
$train->sched_departure, $train->train_no );
$self->app->add_stationinfo( $uid, 0, $train->train_id,
$dep, $arr );
}
}
elsif ( $entry->{real_arr_ts} ) {
my ( undef, $error ) = $self->app->checkout_p(
station => $arr,
force => 2,
dep_eva => $dep,
arr_eva => $arr,
uid => $uid
)->catch(
sub {
my ($error) = @_;
$self->app->log->error("work($uid)/arrival: $error");
$errors += 1;
}
)->wait;
}
};
if ($@) {
$self->app->log->error("work($uid)/arrival: $@");
$errors += 1;
}
eval { };
}
my $started_at = $now;
my $main_finished_at = DateTime->now( time_zone => 'Europe/Berlin' );
my $worker_duration = $main_finished_at->epoch - $started_at->epoch;
if ( $self->app->config->{influxdb}->{url} ) {
if ( $self->app->mode eq 'development' ) {
$self->app->log->debug( 'POST '
. $self->app->config->{influxdb}->{url}
. " worker runtime_seconds=${worker_duration},errors=${errors}"
);
}
else {
$self->app->ua->post_p( $self->app->config->{influxdb}->{url},
"worker runtime_seconds=${worker_duration},errors=${errors}" )
->wait;
}
}
if ( not $self->app->config->{traewelling}->{separate_worker} ) {
$self->app->start('traewelling');
}
# add_wagonorder and add_stationinfo assume a permanently running IOLoop
# and do not allow Mojolicious commands to wait until they have completed.
# Hence, some add_wagonorder and add_stationinfo calls made here may not
# complete before the work command exits, and thus have no effect.
#
# This is not ideal and will need fixing at some point. Until then, here
# is the pragmatic solution for 99% of the associated issues.
Mojo::Promise->timer(5)->wait;
}
1;
__END__
=head1 SYNOPSIS
Usage: index.pl work
Work Work Work.
Should be called from a cronjob every three minutes or so.