package Gearman::WorkerSpawner::BaseWorker::Client;

=head1 NAME

Gearman::WorkerSpawner::BaseWorker::Client - Gearman::Client wrapper for use
with Gearman::WorkerSpawner::BaseWorker workers

=head1 SYNOPSIS

    my $client = Gearman::WorkerSpawner::BaseWorker::Client->new;
    $client->job_servers('127.0.0.1');

    # foreground job
    my @retvals = $client->run_method(adder => { right_hand => 3 });

    # background job
    $client->run_method_background(log => { message => 'beep' });

=head1 DESCRIPTION

This class implements the run_method function from Gearman::WorkerSpawner in
the synchronous Gearman::Client client. Instead of calling an on_complete
callback, the return value of the job is returned.

The run_method_background method may be used to dispatch background jobs. It
has no meaningful return value.

=cut

use strict;
use warnings;

use base 'Gearman::Client';

use fields qw/ method_suffix /;

use Storable qw/ nfreeze thaw /;

sub new {
    my $ref = shift;
    my $class = ref $ref || $ref;

    my Gearman::WorkerSpawner::BaseWorker::Client $self = fields::new($class);
    $self->SUPER::new(@_);

    $self->{method_suffix} = '_m';

    return $self;
}

sub method_suffix {
    my Gearman::WorkerSpawner::BaseWorker::Client $self = shift;
    $self->{method_suffix} = shift if @_;;
    return $self->{method_suffix};
}

sub run_method {
    my Gearman::WorkerSpawner::BaseWorker::Client $self = shift;
    my ($methodname, $arg) = @_;

    $methodname .= $self->{method_suffix};

    my $frozen_arg = \nfreeze([$arg]);

    my $ref_to_frozen_retval = $self->do_task($methodname => $frozen_arg);

    if (!$ref_to_frozen_retval || ref $ref_to_frozen_retval ne 'SCALAR') {
        die "marshaling error";
    }

    my $rets = eval { thaw($$ref_to_frozen_retval) };
    die "unmarshaling error: $@" if $@;
    die "unmarshaling error (incompatible clients?)" if ref $rets ne 'ARRAY';

    return @$rets;
}

sub run_method_background {
    my Gearman::WorkerSpawner::BaseWorker::Client $self = shift;
    my ($methodname, $arg) = @_;

    $methodname .= $self->{method_suffix};

    my $frozen_arg = \nfreeze([$arg]);

    $self->dispatch_background($methodname => $frozen_arg);

    return;
}

1;