File Coverage

blib/lib/DBR/Query/Count.pm

Criterion	Covered	Total	%
statement	30	31	96.7
branch	6	10	60.0
condition			n/a
subroutine	8	8	100.0
pod	0	2	0.0
total	44	51	86.2

# The contents of this file are Copyright (c) 2010 Daniel Norman
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License as
# published by the Free Software Foundation.

###########################################
package DBR::Query::Count;

use strict;
use base 'DBR::Query';
use Carp;

sub _params    { qw (tables where limit quiet_error) }
sub _reqparams { qw (tables) }
sub _validate_self{ 1 } # If I exist, I'm valid



sub sql{
      my $self = shift;
      my $conn   = $self->instance->connect('conn') or return $self->_error('failed to connect');
      my $sql;

      my $tables = join(',', map { $_->sql( $conn ) } @{$self->{tables}} );
      my $fields = join(',', map { $_->sql( $conn ) } @{$self->{fields}} );

      $sql = "SELECT count(*) FROM $tables";
      $sql .= ' WHERE ' . $self->{where}->sql($conn) if $self->{where};
      $sql .= ' LIMIT ' . $self->{limit}             if $self->{limit};

      $self->_logDebug2( $sql );
      return $sql;
}

sub run {
      my $self = shift;

      my $conn     = $self->instance->connect('conn') or confess 'failed to connect';
      my ($count)  = $conn->selectrow_array($self->sql);
      defined($count) or confess "failed to retrieve count";

      return $count;

}


1;