Bug 31252: Advanced search in staff interface should call barcodedecode if the search...
[koha-ffzg.git] / Koha / BackgroundJob.pm
index 583ada7..cb0f9ad 100644 (file)
@@ -16,8 +16,7 @@ package Koha::BackgroundJob;
 # along with Koha; if not, see <http://www.gnu.org/licenses>.
 
 use Modern::Perl;
-use JSON qw( decode_json encode_json );
-use Encode qw( encode_utf8 );
+use JSON;
 use Carp qw( croak );
 use Net::Stomp;
 use Try::Tiny qw( catch try );
@@ -26,6 +25,7 @@ use C4::Context;
 use Koha::DateUtils qw( dt_from_string );
 use Koha::Exceptions;
 use Koha::Plugins;
+use Koha::Exceptions::BackgroundJob;
 
 use base qw( Koha::Object );
 
@@ -47,7 +47,7 @@ my $job_id = Koha::BackgroundJob->enqueue(
 );
 
 Consumer:
-Koha::BackgrounJobs->find($job_id)->process;
+Koha::BackgroundJobs->find($job_id)->process;
 See also C<misc/background_jobs_worker.pl> for a full example
 
 =head1 API
@@ -95,13 +95,17 @@ Return the job_id of the newly created job.
 sub enqueue {
     my ( $self, $params ) = @_;
 
-    my $job_type = $self->job_type;
-    my $job_size = $params->{job_size};
-    my $job_args = $params->{job_args};
-    my $job_queue = $params->{job_queue} // 'default';
+    my $job_type    = $self->job_type;
+    my $job_size    = $params->{job_size};
+    my $job_args    = $params->{job_args};
+    my $job_context = $params->{job_context} // C4::Context->userenv;
+    my $job_queue   = $params->{job_queue}  // 'default';
+    my $json = $self->json;
 
     my $borrowernumber = (C4::Context->userenv) ? C4::Context->userenv->{number} : undef;
-    my $json_args = encode_json $job_args;
+    $job_context->{interface} = C4::Context->interface;
+    my $json_context = $json->encode($job_context);
+    my $json_args = $json->encode($job_args);
 
     $self->set(
         {
@@ -110,6 +114,7 @@ sub enqueue {
             queue          => $job_queue,
             size           => $job_size,
             data           => $json_args,
+            context        => $json_context,
             enqueued_on    => dt_from_string,
             borrowernumber => $borrowernumber,
         }
@@ -125,7 +130,7 @@ sub enqueue {
     };
     return unless $conn;
 
-    $json_args = encode_json $job_args;
+    $json_args = $json->encode($job_args);
     try {
         # This namespace is wrong, it must be a vhost instead.
         # But to do so it needs to be created on the server => much more work when a new Koha instance is created.
@@ -161,9 +166,139 @@ sub process {
 
     $args ||= {};
 
+    if ( $self->context ) {
+        my $context = $self->json->decode($self->context);
+        C4::Context->_new_userenv(-1);
+        C4::Context->interface( $context->{interface} );
+        C4::Context->set_userenv(
+            $context->{number},       $context->{id},
+            $context->{cardnumber},   $context->{firstname},
+            $context->{surname},      $context->{branch},
+            $context->{branchname},   $context->{flags},
+            $context->{emailaddress}, undef,
+            $context->{desk_id},      $context->{desk_name},
+            $context->{register_id},  $context->{register_name}
+        );
+    }
+    else {
+        Koha::Logger->get->warn("A background job didn't have context defined (" . $self->id . ")");
+    }
+
     return $derived_class->process( $args );
 }
 
+=head3 start
+
+    $self->start;
+
+Marks the job as started.
+
+=cut
+
+sub start {
+    my ($self) = @_;
+
+    Koha::Exceptions::BackgroundJob::InconsistentStatus->throw(
+        current_status  => $self->status,
+        expected_status => 'new'
+    ) unless $self->status eq 'new';
+
+    return $self->set(
+        {
+            started_on => \'NOW()',
+            progress   => 0,
+            status     => 'started',
+        }
+    )->store;
+}
+
+=head3 step
+
+    $self->step;
+
+Makes the job record a step has taken place.
+
+=cut
+
+sub step {
+    my ($self) = @_;
+
+    Koha::Exceptions::BackgroundJob::InconsistentStatus->throw(
+        current_status  => $self->status,
+        expected_status => 'started'
+    ) unless $self->status eq 'started';
+
+    # reached the end of the tasks already
+    Koha::Exceptions::BackgroundJob::StepOutOfBounds->throw()
+        unless $self->progress < $self->size;
+
+    return $self->progress( $self->progress + 1 )->store;
+}
+
+=head3 finish
+
+    $self->finish;
+
+Makes the job record as finished. If the job status is I<cancelled>, it is kept.
+
+=cut
+
+sub finish {
+    my ( $self, $data ) = @_;
+
+    $self->status('finished') unless $self->status eq 'cancelled' or $self->status eq 'failed';
+
+    return $self->set(
+        {
+            ended_on => \'NOW()',
+            data     => $self->json->encode($data),
+        }
+    )->store;
+}
+
+=head3 json
+
+   my $JSON_object = $self->json;
+
+Returns a JSON object with utf8 disabled. Encoding to UTF-8 should be
+done later.
+
+=cut
+
+sub json {
+    my ( $self ) = @_;
+    $self->{_json} //= JSON->new->utf8(0); # TODO Should we allow_nonref ?
+    return $self->{_json};
+}
+
+=head3 decoded_data
+
+    my $job_data = $self->decoded_data;
+
+Returns the decoded JSON contents from $self->data.
+
+=cut
+
+sub decoded_data {
+    my ($self) = @_;
+
+    return $self->data ? $self->json->decode( $self->data ) : undef;
+}
+
+=head3 set_encoded_data
+
+    $self->set_encoded_data( $data );
+
+Serializes I<$data> as a JSON string and sets the I<data> attribute with it.
+
+=cut
+
+sub set_encoded_data {
+    my ( $self, $data ) = @_;
+
+    return $self->data( $data ? $self->json->encode($data) : undef );
+}
+
 =head3 job_type
 
 Return the job type of the job. Must be a string.
@@ -182,7 +317,7 @@ sub messages {
     my ( $self ) = @_;
 
     my @messages;
-    my $data_dump = decode_json encode_utf8 $self->data;
+    my $data_dump = $self->json->decode($self->data);
     if ( exists $data_dump->{messages} ) {
         @messages = @{ $data_dump->{messages} };
     }
@@ -199,7 +334,7 @@ Report of the job.
 sub report {
     my ( $self ) = @_;
 
-    my $data_dump = decode_json encode_utf8 $self->data;
+    my $data_dump = $self->json->decode($self->data);
     return $data_dump->{report} || {};
 }
 
@@ -251,7 +386,7 @@ sub _derived_class {
 
 =head3 type_to_class_mapping
 
-    my $mapping = Koha::BackgrounJob->new->type_to_class_mapping;
+    my $mapping = Koha::BackgroundJob->new->type_to_class_mapping;
 
 Returns the available types to class mappings.
 
@@ -260,7 +395,7 @@ Returns the available types to class mappings.
 sub type_to_class_mapping {
     my ($self) = @_;
 
-    my $plugins_mapping = $self->plugin_types_to_classes;
+    my $plugins_mapping = ( C4::Context->config("enable_plugins") ) ? $self->plugin_types_to_classes : {};
 
     return ($plugins_mapping)
       ? { %{ $self->core_types_to_classes }, %$plugins_mapping }
@@ -269,7 +404,7 @@ sub type_to_class_mapping {
 
 =head3 core_types_to_classes
 
-    my $mappings = Koha::BackgrounJob->new->core_types_to_classes
+    my $mappings = Koha::BackgroundJob->new->core_types_to_classes
 
 Returns the core background jobs types to class mappings.
 
@@ -285,6 +420,10 @@ sub core_types_to_classes {
         batch_item_record_modification      => 'Koha::BackgroundJob::BatchUpdateItem',
         batch_hold_cancel                   => 'Koha::BackgroundJob::BatchCancelHold',
         update_elastic_index                => 'Koha::BackgroundJob::UpdateElasticIndex',
+        update_holds_queue_for_biblios      => 'Koha::BackgroundJob::BatchUpdateBiblioHoldsQueue',
+        stage_marc_for_import               => 'Koha::BackgroundJob::StageMARCForImport',
+        marc_import_commit_batch            => 'Koha::BackgroundJob::MARCImportCommitBatch',
+        marc_import_revert_batch            => 'Koha::BackgroundJob::MARCImportRevertBatch',
     };
 }
 
@@ -333,6 +472,44 @@ sub plugin_types_to_classes {
     return $self->{_plugin_mapping};
 }
 
+=head3 to_api
+
+    my $json = $job->to_api;
+
+Overloaded method that returns a JSON representation of the Koha::BackgroundJob object,
+suitable for API output.
+
+=cut
+
+sub to_api {
+    my ( $self, $params ) = @_;
+
+    my $json = $self->SUPER::to_api( $params );
+
+    $json->{context} = $self->json->decode($self->context)
+      if defined $self->context;
+    $json->{data} = $self->decoded_data;
+
+    return $json;
+}
+
+=head3 to_api_mapping
+
+This method returns the mapping for representing a Koha::BackgroundJob object
+on the API.
+
+=cut
+
+sub to_api_mapping {
+    return {
+        id             => 'job_id',
+        borrowernumber => 'patron_id',
+        ended_on       => 'ended_date',
+        enqueued_on    => 'enqueued_date',
+        started_on     => 'started_date',
+    };
+}
+
 =head3 _type
 
 =cut