| DBIx-DBO documentation | Contained in the DBIx-DBO distribution. |
DBIx::DBO::Query - An OO interface to SQL queries and results. Encapsulates an entire query in an object.
# Create a Query object by JOINing 2 tables
my $query = $dbo->query('my_table', 'my_other_table');
# Get the Table objects from the query
my ($table1, $table2) = $query->tables;
# Add a JOIN ON clause
$query->join_on($table1 ** 'login', '=', $table2 ** 'username');
# Find our ancestors, and order by age (oldest first)
$query->where('name', '=', 'Adam');
$query->where('name', '=', 'Eve');
$query->order_by({ COL => 'age', ORDER => 'DESC' });
# New Query using a LEFT JOIN
($query, $table1) = $dbo->query('my_table');
$table2 = $query->join_table('another_table', 'LEFT');
$query->join_on($table1 ** 'parent_id', '=', $table2 ** 'child_id');
# Find those not aged between 20 and 30.
$query->where($table1 ** 'age', '<', 20, FORCE => 'OR'); # Force OR so that we get: (age < 20 OR age > 30)
$query->where($table1 ** 'age', '>', 30, FORCE => 'OR'); # instead of the default: (age < 20 AND age > 30)
A Query object represents rows from a database (from one or more tables). This module makes it easy, not only to fetch and use the data in the returned rows, but also to modify the query to return a different result set.
newDBIx::DBO::Query->new($dbo, $table1, ...);
Create a new Query object from the tables specified.
In scalar context, just the Query object will be returned.
In list context, the Query object and Table objects will be returned for each table specified.
reset$query->reset;
Reset the query, start over with a clean slate.
NB: This will not remove the JOINs or JOIN ON clauses.
tablesReturn a list of Table objects for this query.
columnsReturn a list of column names.
column$query->column($alias_or_column_name);
Returns a reference to a column for use with other methods.
show $query->show(@columns);
$query->show($table1, {COL => $table2 ** 'name', AS => 'name2'});
$query->show($table1 ** 'id', {FUNC => 'UCASE(?)', COL => 'name', AS => 'alias'}, ...
Specify which columns to show as an array. If the array is empty all columns will be shown. If you use a Table object, all the columns from that table will be shown.
distinct$query->distinct(1); my $is_distinct = $query->distinct();
Takes a boolean argument to add or remove the DISTINCT clause for the returned rows. Returns the previous setting.
join_table$query->join_table($table, $join_type); $query->join_table([$schema, $table], $join_type); $query->join_table($table_object, $join_type);
Join a table onto the query, creating a Table object if needed. This will perform a comma (", ") join unless $join_type is specified.
Valid join types are any accepted by the DB. Eg: 'JOIN', 'LEFT', 'RIGHT', undef (for comma join), 'INNER', 'OUTER', ...
Returns the Table object.
join_on$query->join_on($table_object, $expression1, $operator, $expression2); $query->join_on($table2, $table1 ** 'id', '=', $table2 ** 'id');
Join tables on a specific WHERE clause. The first argument is the table object being joined onto. Then a JOIN ON condition follows, which uses the same arguments as where.
open_join_on_bracket, close_join_on_bracket$query->open_join_on_bracket($table, 'OR'); $query->join_on(... $query->close_join_on_bracket($table);
Equivalent to open_bracket, but for the JOIN ON clause. The first argument is the table being joined onto.
whereRestrict the query with the condition specified (WHERE clause).
$query->where($expression1, $operator, $expression2); $query->where($table1 ** 'id', '=', $table2 ** 'id');
$operator is one of: '=', '<', '>', 'IN', 'NOT IN', 'BETWEEN', 'NOT BETWEEN', ...
$expressions can be any of the following:
123 or 'hello' (or for $expression1 a column name: 'id') \"22 * 3" (These are passed unquoted in the SQL statement!) [1, 3, 5] (Used with IN and BETWEEN etc) $table ** 'id' or $table->column('id') For a more complex where expression it can be passed as a hash reference. Possibly containing scalars, arrays or Column objects.
$query->where('name', '=', { FUNC => 'COALESCE(?,?)', VAL => [$name, 'Unknown'] });
$query->where('string', '=', { FUNC => "CONCAT('Mr. ',?)", COL => 'name' });
The keys to the hash in a complex expression are:
VAL => A scalar, scalar reference or an array reference. COL => The name of a column or a Column object. AS => An alias name. FUNC => A string to be inserted into the SQL, possibly containing "?" placeholders. COLLATE => The collation for this value/field. ORDER => To order by a column (Used only in group_by and order_by).Multiple where expressions are combined cleverly using the preferred aggregator 'AND' (unless open_bracket was used to change this). So that when you add where expressions to the query, they will be ANDed together. However some expressions that refer to the same column will automatically be ORed instead where this makes sense, currently: '=', 'IS NULL', '<=>', 'IN' and 'BETWEEN'. Similarly, when the preferred aggregator is 'OR' the following operators will be ANDed together: '!=', 'IS NOT NULL', '<>', 'NOT IN' and 'NOT BETWEEN'.
$query->where('id', '=', 5);
$query->where('name', '=', 'Bob');
$query->where('id', '=', 7);
$query->where(...
# Produces: WHERE ("id" = 5 OR "id" = 7) AND "name" = 'Bob' AND ...
unwhere$query->unwhere(); $query->unwhere($column);
Removes all previously added where restrictions for a column. If no column is provided, the whole WHERE clause is removed.
open_bracket, close_bracket $query->open_bracket('OR');
$query->where( ...
$query->where( ...
$query->close_bracket;
Used to group where expressions together in parenthesis using either 'AND' or 'OR' as the preferred aggregator.
All the where calls made between open_bracket and close_bracket will be inside the parenthesis.
Without any parenthesis 'AND' is the preferred aggregator.
group_by $query->group_by('column', ...);
$query->group_by($table ** 'column', ...);
$query->group_by({ COL => $table ** 'column', ORDER => 'DESC' }, ...);
Group the results by the column(s) listed. This will replace the GROUP BY clause.
To remove the GROUP BY clause simply call group_by without any columns.
havingRestrict the query with the condition specified (HAVING clause). This takes the same arguments as where.
$query->having($expression1, $operator, $expression2);
unhaving$query->unhaving(); $query->unhaving($column);
Removes all previously added having restrictions for a column. If no column is provided, the whole HAVING clause is removed.
order_by $query->order_by('column', ...);
$query->order_by($table ** 'column', ...);
$query->order_by({ COL => $table ** 'column', ORDER => 'DESC' }, ...);
Order the results by the column(s) listed. This will replace the ORDER BY clause.
To remove the ORDER BY clause simply call order_by without any columns.
limit$query->limit; $query->limit($rows); $query->limit($rows, $offset);
Limit the maximum number of rows returned to $rows, optionally skipping the first $offset rows.
When called without arguments or if $rows is undefined, the limit is removed.
arrayref$query->arrayref; $query->arrayref(\%attr);
Run the query using DBI->selectall_arrayref which returns the result as an arrayref.
You can specify a slice by including a 'Slice' or 'Columns' attribute in %attr - See DBI->selectall_arrayref.
hashref$query->hashref($key_field); $query->hashref($key_field, \%attr);
Run the query using DBI->selectall_hashref which returns the result as an hashref.
$key_field defines which column, or columns, are used as keys in the returned hash.
col_arrayref$query->col_arrayref; $query->col_arrayref(\%attr);
Run the query using DBI->selectcol_arrayref which returns the result as an arrayref of the values of each row in one array. By default it pushes all the columns requested by the show method onto the result array (this differs from the DBI). Or to specify which columns to include in the result use the 'Columns' attribute in %attr - see DBI->selectcol_arrayref.
fetchmy $row = $query->fetch;
Fetch the next row from the query. This will run/rerun the query if needed.
Returns a Row object or undefined if there are no more rows.
rowmy $row = $query->row;
Returns the Row object for the current row from the query or an empty Row object if there is no current row.
run$query->run;
Run/rerun the query. This is called automatically before fetching the first row.
rowsmy $row_count = $query->rows;
Count the number of rows returned. Returns undefined if the number is unknown.
count_rowsmy $row_count = $query->count_rows;
Count the number of rows that would be returned. Returns undefined if there is an error.
found_rows$query->config(CalcFoundRows => 1); # Only applicable to MySQL my $total_rows = $query->found_rows;
Return the number of rows that would have been returned if there was no limit clause. Before runnning the query the CalcFoundRows config option can be enabled for improved performance on supported databases.
Returns undefined if there is an error or is unable to determine the number of found rows.
sqlmy $sql = $query->sql;
Returns the SQL statement string.
finish$query->finish;
Calls DBI->finish on the statement handle, if it's active.
Restarts cached queries from the first row (if created using the CacheQuery config).
This ensures that the next call to fetch will return the first row from the query.
These methods are accessible from all DBIx::DBO* objects.
dbhThe read-write DBI handle.
rdbhThe read-only DBI handle, or if there is no read-only connection, the read-write DBI handle.
do$query->do($statement) or die $query->dbh->errstr; $query->do($statement, \%attr) or die $query->dbh->errstr; $query->do($statement, \%attr, @bind_values) or die ...
This provides access to DBI->do method. It defaults to using the read-write DBI handle.
config$query_setting = $query->config($option); $query->config($option => $query_setting);
Get or set this Query object's config settings. When setting an option, the previous value is returned. When getting an option's value, if the value is undefined, the DBIx::DBO's value is returned.
When subclassing DBIx::DBO::Query, please note that Query objects created with the new method are blessed into a DBD driver specific module.
For example, if using MySQL, a new Query object will be blessed into DBIx::DBO::Query::DBD::mysql which inherits from DBIx::DBO::Query.
However if objects are created from a subclass called MySubClass the new object will be blessed into MySubClass::DBD::mysql which will inherit from both MySubClass and DBIx::DBO::Query::DBD::mysql.
Classes can easily be created for tables in your database.
Assume you want to create a Query and Row class for a "Users" table:
package My::Users;
use base 'DBIx::DBO::Query';
sub new {
my $class = shift;
my $dbo = shift;
my $self = $class->SUPER::new($dbo, 'Users'); # Create the Query for the "Users" table
# We could even add some JOINs or other clauses here
return $self;
}
sub _row_class { 'My::User' } # Rows are blessed into this class
package My::User;
use base 'DBIx::DBO::Row';
sub new {
my $class = shift;
my ($dbo, $parent) = @_;
$parent ||= My::Users->new($dbo); # The Row will use the same table as it's parent
$class->SUPER::new($dbo, $parent);
}
| DBIx-DBO documentation | Contained in the DBIx-DBO distribution. |
package DBIx::DBO::Query; use strict; use warnings; use DBIx::DBO::Common; use Devel::Peek 'SvREFCNT'; use Carp 'croak'; our @ISA = qw(DBIx::DBO::Common); BEGIN { if ($] < 5.008_009) { require XSLoader; XSLoader::load(__PACKAGE__, $DBIx::DBO::VERSION); } else { require Hash::Util; *_hv_store = \&Hash::Util::hv_store; } }
sub new { my $proto = shift; UNIVERSAL::isa($_[0], 'DBIx::DBO') or croak 'Invalid DBO Object'; my $class = ref($proto) || $proto; $class = $class->_set_dbd_inheritance($_[0]{dbd}); $class->_init(@_); } sub _init { my $class = shift; my $me = { DBO => shift, sql => undef, Columns => [] }; croak 'No table specified in new Query' unless @_; bless $me, $class; for my $table (@_) { $me->join_table($table); } $me->reset; return wantarray ? ($me, $me->tables) : $me; }
sub reset { my $me = shift; $me->finish; $me->unwhere; $me->distinct(0); $me->show; $me->group_by; $me->order_by; $me->limit; # FIXME: Should we be deleting this? delete $me->{Config}; }
sub tables { @{$_[0]->{Tables}}; } sub _table_idx { my ($me, $tbl) = @_; for my $i (0 .. $#{$me->{Tables}}) { return $i if $tbl == $me->{Tables}[$i]; } return undef; } sub _table_alias { my ($me, $tbl) = @_; return undef if $me == $tbl; # This means it's checking for an aliased column my $i = $me->_table_idx($tbl); croak 'The table is not in this query' unless defined $i; # Don't use aliases, when there's only 1 table @{$me->{Tables}} > 1 ? 't'.($i + 1) : (); }
sub columns { @{$_[0]->{Columns}}; }
sub column { my ($me, $col, $_check_aliases) = @_; $_check_aliases = $me->_alias_preference('column') unless defined $_check_aliases; my $column; return $column if $_check_aliases == 1 and $column = $me->_check_alias($col); for my $tbl ($me->tables) { return $tbl->column($col) if exists $tbl->{Column_Idx}{$col}; } return $column if $_check_aliases == 2 and $column = $me->_check_alias($col); croak 'No such column'.($_check_aliases ? '/alias' : '').': '.$me->_qi($col); } sub _check_alias { my ($me, $col) = @_; for my $fld (@{$me->{build_data}{Showing}}) { return $me->{Column}{$col} ||= bless [$me, $col], 'DBIx::DBO::Column' if ref($fld) eq 'ARRAY' and exists $fld->[2]{AS} and $col eq $fld->[2]{AS}; } }
# TODO: Keep track of all aliases in use and die if a used alias is removed sub show { my $me = shift; undef $me->{sql}; undef $me->{build_data}{from}; undef $me->{build_data}{show}; undef @{$me->{build_data}{Showing}}; for my $fld (@_) { if (UNIVERSAL::isa($fld, 'DBIx::DBO::Table')) { croak 'Invalid table field' unless defined $me->_table_idx($fld); push @{$me->{build_data}{Showing}}, $fld; next; } # If the $fld is just a scalar use it as a column name not a value push @{$me->{build_data}{Showing}}, [ $me->_parse_col_val($fld, Aliases => 0) ]; } }
sub distinct { my $me = shift; undef $me->{sql}; undef $me->{build_data}{show}; my $distinct = $me->{build_data}{Show_Distinct}; $me->{build_data}{Show_Distinct} = shift() ? 1 : undef if @_; return $distinct; }
## # Comma, INNER, NATURAL, LEFT, RIGHT, FULL ## sub join_table { my ($me, $tbl, $type) = @_; if (UNIVERSAL::isa($tbl, 'DBIx::DBO::Table')) { croak 'This table is already in this query' if $me->_table_idx($tbl); } else { $tbl = $me->_table_class->new($me->{DBO}, $tbl); } if (defined $type) { $type =~ s/^\s*/ /; $type =~ s/\s*$/ /; $type = uc $type; $type .= 'JOIN ' if $type !~ /\bJOIN\b/; } else { $type = ', '; } push @{$me->{Tables}}, $tbl; push @{$me->{build_data}{Join}}, $type; push @{$me->{build_data}{Join_On}}, undef; push @{$me->{Join_Bracket_Refs}}, []; push @{$me->{Join_Brackets}}, []; undef $me->{sql}; undef $me->{build_data}{from}; return $tbl; }
sub join_on { my $me = shift; my $t2 = shift; my $i = $me->_table_idx($t2) or croak 'Invalid table object to join onto'; my ($col1, $col1_func, $col1_opt) = $me->_parse_col_val(shift); my $op = shift; my ($col2, $col2_func, $col2_opt) = $me->_parse_col_val(shift); # Validate the fields $me->_validate_where_fields(@$col1, @$col2); # Force a new search undef $me->{sql}; undef $me->{build_data}{from}; $me->{build_data}{Join}[$i] = ' JOIN ' if $me->{build_data}{Join}[$i] eq ', '; $me->_add_where($me->{build_data}{Join_On}[$i] ||= [], $op, $col1, $col1_func, $col1_opt, $col2, $col2_func, $col2_opt, @_); }
sub open_join_on_bracket { my $me = shift; my $i = $me->_table_idx(shift) or croak 'No such table object in the join'; $me->_open_bracket($me->{Join_Brackets}[$i], $me->{Join_Bracket_Refs}[$i], $me->{build_data}{Join_On}[$i] ||= [], @_); } sub close_join_on_bracket { my $me = shift; my $i = $me->_table_idx(shift) or croak 'No such table object in the join'; $me->_close_bracket($me->{Join_Brackets}[$i], $me->{Join_Bracket_Refs}[$i]); }
sub where { my $me = shift; # If the $fld is just a scalar use it as a column name not a value my ($fld, $fld_func, $fld_opt) = $me->_parse_col_val(shift); my $op = shift; my ($val, $val_func, $val_opt) = $me->_parse_val(shift, Check => 'Auto'); # Validate the fields $me->_validate_where_fields(@$fld, @$val); # Force a new search undef $me->{sql}; undef $me->{build_data}{where}; # Find the current Where_Data reference my $ref = $me->{build_data}{Where_Data} ||= []; $ref = $ref->[$_] for (@{$me->{Where_Bracket_Refs}}); $me->_add_where($ref, $op, $fld, $fld_func, $fld_opt, $val, $val_func, $val_opt, @_); }
sub unwhere { my $me = shift; $me->_del_where('Where', @_); } sub _validate_where_fields { my $me = shift; for my $f (@_) { if (UNIVERSAL::isa($f, 'DBIx::DBO::Column')) { $me->_valid_col($f); } elsif (my $type = ref $f) { croak 'Invalid value type: '.$type if $type ne 'SCALAR'; } } } sub _del_where { my $me = shift; my $clause = shift; if (@_) { require Data::Dumper; my ($fld, $fld_func, $fld_opt) = $me->_parse_col_val(shift); # TODO: Validate the fields? return unless exists $me->{build_data}{$clause.'_Data'}; # Find the current Where_Data reference my $ref = $me->{build_data}{$clause.'_Data'}; $ref = $ref->[$_] for (@{$me->{$clause.'_Bracket_Refs'}}); local $Data::Dumper::Indent = 0; my @match = grep { Data::Dumper::Dumper($fld, $fld_func, $fld_opt) eq Data::Dumper::Dumper(@{$ref->[$_]}[1,2,3]) } 0 .. $#$ref; if (@_) { my $op = shift; my ($val, $val_func, $val_opt) = $me->_parse_val(shift, Check => 'Auto'); @match = grep { Data::Dumper::Dumper($op, $val, $val_func, $val_opt) eq Data::Dumper::Dumper(@{$ref->[$_]}[0,4,5,6]) } @match; } splice @$ref, $_, 1 for reverse @match; } else { delete $me->{build_data}{$clause.'_Data'}; $me->{$clause.'_Bracket_Refs'} = []; $me->{$clause.'_Brackets'} = []; } # This forces a new search undef $me->{sql}; undef $me->{build_data}{lc $clause}; } ## # This will add an arrayref to the $ref given. # The arrayref will contain 5 values: # $op, $fld_func, $fld, $val_func, $val, $force # $op is the operator (those supported differ by DBD) # $fld_func is undef or a scalar of the form '? AND ?' or 'POSITION(? IN ?)' # $fld is an arrayref of columns/values for use with $fld_func # $val_func is similar to $fld_func # $val is an arrayref of values for use with $val_func # $force is one of undef / 'AND' / 'OR' which if defined, overrides the default aggregator ## sub _add_where { my $me = shift; my ($ref, $op, $fld, $fld_func, $fld_opt, $val, $val_func, $val_opt, %opt) = @_; croak 'Invalid option, FORCE must be AND or OR' if defined $opt{FORCE} and $opt{FORCE} ne 'AND' and $opt{FORCE} ne 'OR'; # Deal with NULL values if (@$val == 1 and !defined $val->[0] and !defined $val_func) { if ($op eq '=') { $op = 'IS'; $val_func = 'NULL'; delete $val->[0]; } elsif ($op eq '!=') { $op = 'IS NOT'; $val_func = 'NULL'; delete $val->[0]; } } # Deal with array values: BETWEEN & IN unless (defined $val_func) { if ($op eq 'BETWEEN' or $op eq 'NOT BETWEEN') { croak 'Invalid value argument, BETWEEN requires 2 values' if ref $val ne 'ARRAY' or @$val != 2; $val_func = $me->PLACEHOLDER.' AND '.$me->PLACEHOLDER; } elsif ($op eq 'IN' or $op eq 'NOT IN') { if (ref $val eq 'ARRAY') { croak 'Invalid value argument, IN requires at least 1 value' if @$val == 0; } else { $val = [ $val ]; } # Add to previous 'IN' and 'NOT IN' Where expressions unless ($opt{FORCE} and $opt{FORCE} ne _op_ag($op)) { for my $lim (grep $$_[0] eq $op, @{$ref}) { next if defined $$lim[1] xor defined $fld; next if defined $$lim[1] and defined $fld and $$lim[1] != $fld; last if ($$lim[5] and $$lim[5] ne _op_ag($op)); last if $$lim[4] ne '('.join(',', ($me->PLACEHOLDER) x @{$$lim[2]}).')'; push @{$$lim[2]}, @$val; $$lim[4] = '('.join(',', ($me->PLACEHOLDER) x @{$$lim[2]}).')'; return; } } $val_func = '('.join(',', ($me->PLACEHOLDER) x @$val).')'; } elsif (@$val != 1) { # Check that there is only 1 placeholder croak 'Wrong number of fields/values, called with '.@$val.' while needing 1'; } } push @{$ref}, [ $op, $fld, $fld_func, $fld_opt, $val, $val_func, $val_opt, $opt{FORCE} ]; } sub _parse_col_val { my $me = shift; my $col = shift; my %c = (Check => 'Column', @_); unless (defined $c{Aliases}) { (my $method = (caller(1))[3]) =~ s/.*:://; $c{Aliases} = $me->_alias_preference($method); } return $me->_parse_val($col, %c) if ref $col; return [ $me->_parse_col($col, $c{Aliases}) ]; }
sub open_bracket { my $me = shift; $me->_open_bracket($me->{Where_Brackets}, $me->{Where_Bracket_Refs}, $me->{build_data}{Where_Data} ||= [], @_); } sub _open_bracket { my ($me, $brackets, $bracket_refs, $ref, $ag) = @_; croak 'Invalid argument MUST be AND or OR' if !$ag or $ag !~ /^(AND|OR)$/; my $last = @$brackets ? $brackets->[-1] : 'AND'; if ($ag ne $last) { # Find the current data reference $ref = $ref->[$_] for @$bracket_refs; push @$ref, []; push @$bracket_refs, $#$ref; } push @$brackets, $ag; } sub close_bracket { my $me = shift; $me->_close_bracket($me->{Where_Brackets}, $me->{Where_Bracket_Refs}); } sub _close_bracket { my ($me, $brackets, $bracket_refs) = @_; my $ag = pop @{$brackets} or croak "Can't close bracket with no open bracket!"; my $last = @$brackets ? $brackets->[-1] : 'AND'; pop @$bracket_refs if $last ne $ag; return $ag; }
sub group_by { my $me = shift; undef $me->{sql}; undef $me->{build_data}{group}; undef @{$me->{build_data}{GroupBy}}; for my $col (@_) { my @group = $me->_parse_col_val($col); push @{$me->{build_data}{GroupBy}}, \@group; } }
sub having { my $me = shift; # If the $fld is just a scalar use it as a column name not a value my ($fld, $fld_func, $fld_opt) = $me->_parse_col_val(shift); my $op = shift; my ($val, $val_func, $val_opt) = $me->_parse_val(shift, Check => 'Auto'); # Validate the fields $me->_validate_where_fields(@$fld, @$val); # Force a new search undef $me->{sql}; undef $me->{build_data}{having}; # Find the current Having_Data reference my $ref = $me->{build_data}{Having_Data} ||= []; $ref = $ref->[$_] for (@{$me->{Having_Bracket_Refs}}); $me->_add_where($ref, $op, $fld, $fld_func, $fld_opt, $val, $val_func, $val_opt, @_); }
sub unhaving { my $me = shift; $me->_del_where('Having', @_); }
sub order_by { my $me = shift; undef $me->{sql}; undef $me->{build_data}{order}; undef @{$me->{build_data}{OrderBy}}; for my $col (@_) { my @order = $me->_parse_col_val($col); push @{$me->{build_data}{OrderBy}}, \@order; } }
sub limit { my ($me, $rows, $offset) = @_; undef $me->{sql}; undef $me->{build_data}{limit}; return undef $me->{build_data}{LimitOffset} unless defined $rows; /^\d+$/ or croak "Invalid argument '$_' in limit" for grep defined, $rows, $offset; @{$me->{build_data}{LimitOffset}} = ($rows, $offset); }
sub arrayref { my $me = shift; my $attr = shift; $me->_sql($me->sql, $me->_bind_params_select($me->{build_data})); $me->rdbh->selectall_arrayref($me->{sql}, $attr, $me->_bind_params_select($me->{build_data})); }
sub hashref { my $me = shift; my $key = shift; my $attr = shift; $me->_sql($me->sql, $me->_bind_params_select($me->{build_data})); $me->rdbh->selectall_hashref($me->{sql}, $key, $attr, $me->_bind_params_select($me->{build_data})); }
sub col_arrayref { my $me = shift; my $attr = shift; $me->_sql($me->sql, $me->_bind_params_select($me->{build_data})); my $sth = $me->rdbh->prepare($me->{sql}, $attr) or return; unless (defined $attr->{Columns}) { # Some drivers don't provide $sth->{NUM_OF_FIELDS} until after execute is called if ($sth->{NUM_OF_FIELDS}) { $attr->{Columns} = [1 .. $sth->{NUM_OF_FIELDS}]; } else { $sth->execute($me->_bind_params_select($me->{build_data})) or return; my @col; if (my $max = $attr->{MaxRows}) { push @col, @$_ while 0 < $max-- and $_ = $sth->fetch; } else { push @col, @$_ while $_ = $sth->fetch; } return \@col; } } return $me->rdbh->selectcol_arrayref($sth, $attr, $me->_bind_params_select($me->{build_data})); }
sub fetch { my $me = $_[0]; # Prepare and/or execute the query if needed $me->_sth and ($me->{Active} or $me->run) or croak $me->rdbh->errstr; # Detach the old row if there is still another reference to it if (defined $me->{Row} and SvREFCNT(${$me->{Row}}) > 1) { $me->{Row}->_detach; } my $row = $me->row; if (exists $me->{cache}) { if ($me->{cache}{idx} < @{$me->{cache}{data}}) { @{$me->{cache}{array}}[0..$#{$me->{cache}{array}}] = @{$me->{cache}{data}[$me->{cache}{idx}++]}; $$row->{array} = $me->{cache}{array}; $$row->{hash} = $me->{hash}; return $row; } undef $$row->{array}; $me->{cache}{idx} = 0; } else { # Fetch and store the data then return the Row on success and undef on failure or no more rows if ($$row->{array} = $me->{sth}->fetch) { $$row->{hash} = $me->{hash}; return $row; } $me->{Active} = 0; } $$row->{hash} = {}; return; }
sub row { my $me = $_[0]; $me->sql; # Build the SQL and detach the Row if needed $me->{Row} ||= $me->_row_class->new($me->{DBO}, $me); }
sub run { my $me = shift; $me->sql; # Build the SQL and detach the Row if needed if (defined $me->{Row}) { undef ${$me->{Row}}{array}; undef %{$me->{Row}}; } my $rv = $me->_execute or return undef; $me->{Active} = 1; $me->_bind_cols_to_hash; if ($me->config('CacheQuery')) { $me->{cache}{data} = $me->{sth}->fetchall_arrayref; $me->{cache}{idx} = 0; } else { delete $me->{cache}; } return $rv; } sub _execute { my $me = shift; $me->_sql($me->sql, $me->_bind_params_select($me->{build_data})); $me->_sth or return; $me->{sth}->execute($me->_bind_params_select($me->{build_data})); } sub _bind_cols_to_hash { my $me = shift; unless ($me->{hash}) { # Bind only to the first column of the same name @{$me->{Columns}} = @{$me->{sth}{NAME}}; if ($me->config('CacheQuery')) { @{$me->{cache}{array}} = (undef) x @{$me->{Columns}}; $me->{hash} = \my %hash; my $i = 0; for (@{$me->{Columns}}) { _hv_store(%hash, $_, $me->{cache}{array}[$i]) unless exists $hash{$_}; $i++; } } else { my $i; for (@{$me->{Columns}}) { $i++; $me->{sth}->bind_col($i, \$me->{hash}{$_}) unless exists $me->{hash}{$_}; } } } }
sub rows { my $me = shift; $me->sql; # Ensure the Row_Count is cleared if needed unless (defined $me->{Row_Count}) { $me->_sth and ($me->{sth}{Executed} or $me->run) or croak $me->rdbh->errstr; $me->{Row_Count} = $me->_sth->rows; $me->{Row_Count} = $me->count_rows if $me->{Row_Count} == -1; } $me->{Row_Count}; }
sub count_rows { my $me = shift; my $old_fr = $me->{Config}{CalcFoundRows}; $me->{Config}{CalcFoundRows} = 0; my $old_sb = delete $me->{build_data}{Show_Bind}; $me->{build_data}{show} = '1'; my $sql = 'SELECT COUNT(*) FROM ('.$me->_build_sql_select($me->{build_data}).') t'; $me->_sql($sql, $me->_bind_params_select($me->{build_data})); my ($count) = $me->rdbh->selectrow_array($sql, undef, $me->_bind_params_select($me->{build_data})); $me->{Config}{CalcFoundRows} = $old_fr if defined $old_fr; $me->{build_data}{Show_Bind} = $old_sb if $old_sb; undef $me->{build_data}{show}; return $count; }
sub found_rows { my $me = shift; if (not defined $me->{Found_Rows}) { $me->{build_data}{limit} = ''; $me->{Found_Rows} = $me->count_rows; undef $me->{build_data}{limit}; } $me->{Found_Rows}; }
sub sql { my $me = shift; $me->{sql} || $me->_build_sql; } sub _build_sql { my $me = shift; undef $me->{sth}; undef $me->{hash}; undef $me->{Row_Count}; undef $me->{Found_Rows}; delete $me->{cache}; $me->{Active} = 0; if (defined $me->{Row}) { if (SvREFCNT(${$me->{Row}}) > 1) { $me->{Row}->_detach; } else { undef ${$me->{Row}}{array}; undef %{$me->{Row}}; $me->{sql} = $me->_build_sql_select($me->{build_data}); $me->{Row}->_copy_build_data; return $me->{sql}; } } undef @{$me->{Columns}}; $me->{sql} = $me->_build_sql_select($me->{build_data}); } # Get the DBI statement handle for the query. # It may not have been executed yet. sub _sth { my $me = shift; # Ensure the sql is rebuilt if needed my $sql = $me->sql; $me->{sth} ||= $me->rdbh->prepare($sql); }
sub finish { my $me = shift; if (defined $me->{Row}) { if (SvREFCNT(${$me->{Row}}) > 1) { $me->{Row}->_detach; } else { undef ${$me->{Row}}{array}; undef %{$me->{Row}}; } } if (exists $me->{cache}) { $me->{cache}{idx} = 0; } else { $me->{sth}->finish if $me->{sth} and $me->{sth}{Active}; $me->{Active} = 0; } }
sub config { my $me = shift; my $opt = shift; return $me->_set_config($me->{Config} ||= {}, $opt, shift) if @_; return defined $me->{Config}{$opt} ? $me->{Config}{$opt} : $me->{DBO}->config($opt); } sub DESTROY { undef %{$_[0]}; } 1; __END__