File Coverage

blib/lib/Class/DBI/Plugin/AggregateFunction.pm

Criterion	Covered	Total	%
statement	15	37	40.5
branch	0	6	0.0
condition	0	6	0.0
subroutine	5	8	62.5
pod	0	1	0.0
total	20	58	34.4

package Class::DBI::Plugin::AggregateFunction;
use strict;
use vars qw/$VERSION/;
$VERSION = '0.02';

use SQL::Abstract;

sub import {
    my $class = shift;
    my $pkg = caller(0);

    no strict 'refs';
    *{"$pkg\::mk_aggregate_function"} = \&mk_aggregate_function;
}

sub mk_aggregate_function {
    my $class = shift;
    my ($aggregate_func, $alias) = @_;
    $alias ||= $aggregate_func;

    $class->set_sql( "AggregateFunction_$aggregate_func" => <<__SQL__ );
  SELECT $aggregate_func( %s )
  FROM   __TABLE__
  WHERE  %s
__SQL__

    no strict 'refs';
    *{"$class\::$alias"} = sub {
	my $proto = shift;
	my $class = ref($proto) || $proto;
	my $aggregate_column = shift;

	my $where = (ref $_[0]) ? $_[0]          : { @_ };
	my $attr  = (ref $_[0]) ? $_[1]          : undef;

	my $sql = SQL::Abstract->new(%$attr);
	my($phrase, @bind) = $sql->where($where);
	$phrase =~ s/^\s*WHERE\s*//i;
	$phrase = ' 1 = 1 ' unless $phrase;

	my $sql_method = "sql_AggregateFunction_$aggregate_func";
	my $sth = $class->$sql_method( $aggregate_column, $phrase );

	return $sth->select_val( @bind );
    }
}

1;
__END__