File Coverage

blib/lib/Mojo/PgX/Cursor/Cursor.pm

Criterion	Covered	Total	%
statement	3	23	13.0
branch	0	10	0.0
condition	0	14	0.0
subroutine	1	5	20.0
pod	3	3	100.0
total	7	55	12.7

package Mojo::PgX::Cursor::Cursor;

require UUID::Tiny;

use Mojo::Base -base;

has [qw(bind db name query)];

sub DESTROY {
  my $self = shift;
  if ($self->{close} && $self->db && $self->db->ping) { $self->close }
}

sub close {
  my $self = shift;
  my $query = sprintf('close %s', $self->db->dbh->quote_identifier($self->name));
  $self->db->query($query) if delete $self->{close};
}

sub fetch {
  my $self = shift;
  my $cb = ref $_[-1] eq 'CODE' ? pop : undef;
  my $fetch = shift || 100;
  my $query = sprintf('fetch %s from %s', $fetch, $self->db->dbh->quote_identifier($self->name));
  my @query_params = $query;
  push @query_params, $cb if $cb;
  return $self->db->query(@query_params);
}

sub new {
  my $self = shift->SUPER::new(
    bind => [],
    name => UUID::Tiny::create_uuid_as_string(UUID::Tiny::UUID_V4()),
    @_,
  );
  return unless defined $self->db
    and defined $self->query and length $self->query;
  my $query = sprintf('declare %s cursor with hold for %s',
    $self->db->dbh->quote_identifier($self->name), $self->query);
  $self->db->query($query, @{$self->bind});
  $self->{close} = 1;
  return $self;
}

1;
__END__