File Coverage

blib/lib/WWW/VastAI/API/Workergroups.pm

Criterion	Covered	Total	%
statement	41	41	100.0
branch	9	18	50.0
condition	2	10	20.0
subroutine	11	11	100.0
pod	6	6	100.0
total	69	86	80.2

package WWW::VastAI::API::Workergroups;
our $VERSION = '0.001';
# ABSTRACT: Workergroup management for Vast.ai serverless

use Moo;
use Carp qw(croak);
use WWW::VastAI::Workergroup;
use namespace::clean;

has client => (
    is       => 'ro',
    required => 1,
    weak_ref => 1,
);

sub _wrap {
    my ($self, $data) = @_;
    return WWW::VastAI::Workergroup->new(
        client => $self->client,
        data   => $data,
    );
}

sub list {
    my ($self) = @_;
    my $result = $self->client->request_op('listWorkergroups');
    my $groups = ref $result eq 'HASH' ? ($result->{workergroups} || $result->{results} || []) : ($result || []);
    return [ map { $self->_wrap($_) } @{$groups} ];
}

sub create {
    my ($self, %params) = @_;
    croak "template_hash required" unless $params{template_hash};
    croak "endpoint_name required" unless $params{endpoint_name};

    my $result = $self->client->request_op('createWorkergroup', body => \%params);
    my $group = ref $result eq 'HASH' ? ($result->{workergroup} || $result) : $result;
    return $self->_wrap($group);
}

sub update {
    my ($self, $id, %params) = @_;
    croak "workergroup id required" unless defined $id;

    my $result = $self->client->request_op(
        'updateWorkergroup',
        path => { id => $id },
        body => \%params,
    );

    my $group = ref $result eq 'HASH' ? ($result->{workergroup} || $result) : $result;
    return $self->_wrap($group);
}

sub delete {
    my ($self, $id) = @_;
    croak "workergroup id required" unless defined $id;
    return $self->client->request_op('deleteWorkergroup', path => { id => $id });
}

sub logs {
    my ($self, $id, %params) = @_;
    croak "workergroup id required" unless defined $id;
    my %body = ( workergroup_id => $id, %params );
    return $self->client->request_op('getWorkergroupLogs', body => \%body);
}

sub workers {
    my ($self, $id) = @_;
    croak "workergroup id required" unless defined $id;
    return $self->client->request_op('getWorkergroupWorkers', body => { workergroup_id => $id });
}

1;

__END__