blib000755 001751 000000 00000000000 10326255707 011630 5ustar00donwheel000000 000000 blib/lib000755 001751 000000 00000000000 10326255704 012373 5ustar00donwheel000000 000000 blib/lib/DBIx000755 001751 000000 00000000000 10326255704 013161 5ustar00donwheel000000 000000 blib/lib/DBIx/.exists000644 001751 000000 00000000000 10224067271 014543 0ustar00donwheel000000 000000 blib/lib/DBIx/Wrapper000755 001751 000000 00000000000 10326255704 014601 5ustar00donwheel000000 000000 blib/lib/DBIx/Wrapper/Statement.pm000444 001751 000000 00000002705 10140555673 017166 0ustar00donwheel000000 000000 # -*-perl-*- # Creation date: 2003-03-30 15:23:31 # Authors: Don # Change log: # $Id: Statement.pm,v 1.4 2004/10/30 00:09:31 don Exp $ use strict; { package DBIx::Wrapper::Statement; use vars qw($VERSION); $VERSION = do { my @r=(q$Revision: 1.4 $=~/\d+/g); sprintf "%d."."%02d"x$#r,@r }; sub new { my ($proto) = @_; my $self = bless {}, ref($proto) || $proto; return $self; } #################### # getters/setters sub _getSth { my ($self) = @_; return $$self{_sth}; } sub _setSth { my ($self, $sth) = @_; $$self{_sth} = $sth; } sub _getParent { my ($self) = @_; return $$self{_parent}; } sub _setParent { my ($self, $parent) = @_; $$self{_parent} = $parent; } sub _getQuery { my $self = shift; return $self->{_query}; } sub _setQuery { my $self = shift; my $query = shift; $self->{_query} = $query; } sub _getRequestObj { return shift()->{_request_obj}; } sub _setRequestObj { my $self = shift; $self->{_request_obj} = shift; } } 1; __END__ =pod =head1 NAME DBIx::Wrapper::Statement - Statement wrapper used by DBIx::Wrapper =head1 SYNOPSIS =head1 DESCRIPTION =head1 METHODS =head1 EXAMPLES =head1 BUGS =head1 AUTHOR =head1 VERSION $Id: Statement.pm,v 1.4 2004/10/30 00:09:31 don Exp $ =cut blib/lib/DBIx/Wrapper/DBIDelegator.pm000444 001751 000000 00000007116 10326224460 017441 0ustar00donwheel000000 000000 # Creation date: 2005-10-16 20:42:19 # Authors: don =pod =head1 NAME DBIx::Wrapper::DBIDelegator - =head1 SYNOPSIS =head1 DESCRIPTION =cut use strict; use warnings; package DBIx::Wrapper::DBIDelegator; use vars qw($VERSION); $VERSION = do { my @r=(q$Revision: 1.3 $=~/\d+/g); sprintf "%d."."%02d"x$#r,@r }; # use Scalar::Util qw(refaddr); use Carp qw(cluck); my %i_data; sub refaddr($) { my $obj = shift; my $pkg = ref($obj) or return undef; bless $obj, 'DBIx::Wrapper::Fake'; my $i = int($obj); bless $obj, $pkg; return $i; } =pod =head1 METHODS =cut sub _new { my $proto = shift; my $self = bless {}, ref($proto) || $proto; $i_data{ refaddr($self) } = {}; return $self; } sub TIEHASH { my $proto = shift; my $dbix_dbh = shift; my $self = $proto->_new; $i_data{ refaddr($self) }{_dbix_dbh} = $dbix_dbh; return $self; } sub _get_dbi { my $self = shift; return $i_data{ refaddr($self) }{_dbix_dbh}->get_dbi; } sub FETCH { my $self = shift; my $key = shift; if ($key =~ /\A_(?:dbh|username|auth|attr|data_source_str|dbd_driver|db_style|debug)\Z/) { my ($package, $filename, $line, $subroutine, $hasargs, $wantarray, $evaltext, $is_require, $hints, $bitmask); my $frame = 1; my $this_pkg = __PACKAGE__; ($package, $filename, $line, $subroutine, $hasargs, $wantarray, $evaltext, $is_require, $hints, $bitmask) = caller($frame); while (defined($package) and $package eq $this_pkg) { $frame++; ($package, $filename, $line, $subroutine, $hasargs, $wantarray, $evaltext, $is_require, $hints, $bitmask) = caller($frame); # if we get more than 10 something must be wrong last if $frame >= 10; } local($Carp::CarpLevel) = $frame; cluck "Accessing DBIx::Wrapper's internal data directly. Don't do that."; } return $self->_get_dbi()->{$key}; } sub STORE { my $self = shift; my $key = shift; my $value = shift; my $dbi = $self->_get_dbi; $self->_get_dbi()->{$key} = $value; return $value; } sub DELETE { my $self = shift; my $key = shift; return delete $self->_get_dbi()->{$key}; } sub CLEAR { my $self = shift; %{ $self->_get_dbi() } = (); } sub EXISTS { my $self = shift; my $key = shift; return exists $self->_get_dbi()->{$key}; } sub FIRSTKEY { my $self = shift; my $dbi = $self->_get_dbi; my $cnt = keys %$dbi; # reset each() iterator return each %$dbi; } sub NEXTKEY { my $self = shift; my $last_key = shift; return each %{ $self->_get_dbi }; } sub SCALAR { my $self = shift; return scalar(%{ $self->_get_dbi }); } sub UNTIE { # noop } sub DESTROY { my $self = shift; delete $i_data{ refaddr($self) }; return; } =pod =head1 EXAMPLES =head1 DEPENDENCIES =head1 AUTHOR Don Owens =head1 LICENSE AND COPYRIGHT Copyright (c) 2005 Don Owens . All rights reserved. This is free software; you can redistribute it and/or modify it under the same terms as Perl itself. See perlartistic. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. =head1 SEE ALSO =head1 VERSION 0.01 =cut 1; # Local Variables: # # mode: perl # # tab-width: 4 # # indent-tabs-mode: nil # # cperl-indent-level: 4 # # perl-indent-level: 4 # # End: # # vim:set ai si et sta ts=4 sw=4 sts=4: blib/lib/DBIx/Wrapper/Request.pm000444 001751 000000 00000015045 10323773702 016651 0ustar00donwheel000000 000000 # -*-perl-*- # Creation date: 2004-10-29 14:01:59 # Authors: Don # $Id: Request.pm,v 1.6 2005/10/14 18:24:02 don Exp $ use strict; { package DBIx::Wrapper::Request; use vars qw($VERSION); $VERSION = do { my @r=(q$Revision: 1.6 $=~/\d+/g); sprintf "%d."."%02d"x$#r,@r }; sub new { my $proto = shift; my $db_obj = shift; my $self = bless { _db_obj => $db_obj }, ref($proto) || $proto; return $self; } sub getDbObj { return shift()->{_db_obj}; } sub getQuery { shift()->{_query}; } sub setQuery { my $self = shift; $self->{_query} = shift; } sub getExecArgs { return shift()->{_exec_args} || []; } sub setExecArgs { my $self = shift; my $args = shift; if (ref($args) eq 'ARRAY') { $self->{_exec_args} = $args; } else { $self->{_exec_args} = [ $args ]; } } sub getExecReturnValue { return shift()->{_exec_return_value}; } sub setExecReturnValue { my $self = shift; $self->{_exec_return_value} = shift; } sub getReturnVal { return shift()->{_return_record}; } sub setReturnVal { my $self = shift; $self->{_return_record} = shift; } sub getStatementHandle { return shift()->{_statement_handle}; } sub setStatementHandle { my $self = shift; $self->{_statement_handle} = shift; } sub getErrorStr { return shift()->{_errstr}; } sub setErrorStr { my $self = shift; $self->{_errstr} = shift; } sub OK { return 1; } sub DECLINED { return 0; } } 1; __END__ =pod =head1 NAME DBIx::Wrapper::Request - Request object for database operations =head1 SYNOPSIS Objects of the class are created by DBIx::Wrapper objects and passed to hooks. You should never have to create one yourself. my $db = $req->getDbObj; my $query = $req->getQuery; $req->setQuery($query); my $exec_args = $req->getExecArgs; $req->setExecArgs(\@args); my $rv = $req->getExecReturnValue; $req->setExecReturnValue($rv); my $rv = $req->getReturnVal; $req->setReturnVal($rv); my $sth = $req->getStatementHandle; $req->setStatementHandle($sth); my $err_str = $req->getErrorStr; $req->setErrorStr($err_str); =head1 DESCRIPTION DBIx::Wrapper::Request objects are used to encapsulate date passed between DBIx::Wrapper methods at various stages of executing a query. =head1 METHODS =head2 getDbObj() Returns the DBIx::Wrapper object that created the Request object. =head2 getQuery() Returns the current query. =head2 setQuery($query) Sets the current query. =head2 getExecArgs() Returns a reference to the array of execute arguments passed to the DBIx::Wrapper method currently executing. =head2 setExecArgs(\@args); Sets the current execute arguments. =head2 getExecReturnValue() Returns the current execute() return value. =head2 setExecReturnValue($rv); Sets the current execute() return value. =head2 getReturnVal() Gets the current return value (from a fetch). =head2 setReturnVal($rv) Sets the current return value (from a fetch). =head2 getStatementHandle() Get the current statement handle being used. =head2 setStatementHandle($sth) Set the current statement handle to use. =head2 $req->getErrorStr() Get the error string. =head2 setErrorStr($err_str) Set the error string. =head1 EXAMPLES ################################################## # Pre prepare hook $db_obj->addPrePrepareHook(\&_db_pre_prepare_hook) sub _db_pre_prepare_hook { my $self = shift; my $r = shift; my $query = $r->getQuery; if ($query =~ /^\s*(?:update|delete|insert|replace|create|drop|alter)/i) { my $db = $r->getDbObj; unless ($db->ping) { # db connection has gone away, so try to reconnect my $msg = "UI DataProvider pre-prepare: db ping failed, reconnecting to "; $msg .= $db->_getDataSource; print STDERR $msg . "\n"; my $tries_left = 5; my $connected = 0; my $sleep_time = 0; while ($tries_left) { $sleep_time++; sleep $sleep_time; $tries_left--; $connected = $db->reconnect; last if $connected; } unless ($connected) { die "Couldn't reconnect to db after ping failure: dsn=" . $db->_getDataSource; } } } return $r->OK; } ################################################## # Post execute hook sub _db_post_exec_hook { my $self = shift; my $r = shift; my $exec_successful = $r->getExecReturnValue; unless ($exec_successful) { my $query = $r->getQuery; if ($r->getQuery =~ /^\s*(?:select|show)/i) { my $errstr = $r->getErrorStr; if ($errstr =~ /Lost connection to MySQL server during query/i) { my $db = $r->getDbObj; my $msg = "UI DataProvider post exec: lost connection to MySQL server "; $msg .= "during query, reconnecting to " . $db->_getDataSource; print STDERR $msg . "\n"; my $tries_left = 5; my $connected = 0; my $sleep_time = 0; while ($tries_left) { $sleep_time++; sleep $sleep_time; $tries_left--; $connected = $db->reconnect; last if $connected; } if ($connected) { my $sth = $db->prepare_no_hooks($r->getQuery); $r->setStatementHandle($sth); my $exec_args = $r->getExecArgs; my $rv = $sth->execute(@$exec_args); $r->setExecReturnValue($rv); } else { die "Couldn't reconnect to db after losing connection: dsn=" . $db->_getDataSource; } } } } return $r->OK; } =head1 BUGS =head1 AUTHOR =head1 VERSION $Id: Request.pm,v 1.6 2005/10/14 18:24:02 don Exp $ =cut blib/lib/DBIx/Wrapper/SelectLoop.pm000444 001751 000000 00000005231 10212444212 017252 0ustar00donwheel000000 000000 # -*-perl-*- # Creation date: 2003-03-30 15:25:00 # Authors: Don # Change log: # $Id: SelectLoop.pm,v 1.10 2005/03/06 00:02:18 don Exp $ use strict; { package DBIx::Wrapper::SelectLoop; use vars qw($VERSION); $VERSION = do { my @r=(q$Revision: 1.10 $=~/\d+/g); sprintf "%d."."%02d"x$#r,@r }; use base 'DBIx::Wrapper::Statement'; sub new { my ($proto, $parent, $query, $exec_args) = @_; my $self = { _query => $query, _exec_args => $exec_args, _cur_row_count => 0 }; bless $self, ref($proto) || $proto; $self->_setParent($parent); my ($sth, $rv, $r); if (scalar(@_) == 4) { ($sth, $rv, $r) = $parent->_getStatementHandleForQuery($query, $exec_args); } else { ($sth, $rv, $r) = $parent->_getStatementHandleForQuery($query); } return $sth unless $sth; $self->_setSth($sth); $self->_setRequestObj($r); return $self; } sub next { my ($self) = @_; my $sth = $self->_getSth; $self->{_cur_row_count}++; my $r = $self->_getRequestObj; $self->_getParent()->_runPreFetchHook($r); $sth = $r->getStatementHandle; my $result = $sth->fetchrow_hashref($self->_getParent()->getNameArg); $r->setReturnVal($result); $self->_getParent()->_runPostFetchHook($r); $result = $r->getReturnVal; return $result; } sub nextWithArrayRef { my ($self) = @_; my $sth = $self->_getSth; $self->{_cur_row_count}++; my $r = $self->_getRequestObj; $self->_getParent()->_runPreFetchHook($r); $sth = $r->getStatementHandle; my $row = $sth->fetchrow_arrayref; $r->setReturnVal($row); $self->_getParent()->_runPostFetchHook($r); $row = $r->getReturnVal; return [ @$row ] if $row; return undef; } *nextArrayRef = \&nextWithArrayRef; sub rowCountCurrent { my ($self) = @_; return $$self{_cur_row_count}; } sub rowCountTotal { my ($self) = @_; my $sth = $self->_getSth; return $sth->rows; } *count = \&rowCountTotal; sub DESTROY { my ($self) = @_; my $sth = $self->_getSth; $sth->finish if $sth; } } 1; __END__ =pod =head1 NAME DBIx::Wrapper::SelectLoop - Used by DBIx::Wrapper for SelectLoop methods =head1 SYNOPSIS =head1 DESCRIPTION =head1 METHODS =head1 EXAMPLES =head1 BUGS =head1 AUTHOR =head1 VERSION $Id: SelectLoop.pm,v 1.10 2005/03/06 00:02:18 don Exp $ =cut blib/lib/DBIx/Wrapper/SQLCommand.pm000444 001751 000000 00000001603 10070730427 017146 0ustar00donwheel000000 000000 # -*-perl-*- # Creation date: 2003-03-30 16:26:50 # Authors: Don # Change log: # $Id: SQLCommand.pm,v 1.3 2004/07/01 06:37:11 don Exp $ use strict; { package DBIx::Wrapper::SQLCommand; use vars qw($VERSION); $VERSION = do { my @r=(q$Revision: 1.3 $=~/\d+/g); sprintf "%d."."%02d"x$#r,@r }; sub new { my ($proto, $str) = @_; my $self = bless { _str => $str }, ref($proto) || $proto; return $self; } sub asString { my ($self) = @_; return $$self{_str}; } *as_string = \&asString; } 1; __END__ =pod =head1 NAME DBIx::Wrapper::SQLCommand - Used by DBIx::Wrapper to pass SQL as-is. This is deprecated. Use a scalar reference instead. =head1 SYNOPSIS =head1 DESCRIPTION =head1 METHODS =head1 EXAMPLES =head1 BUGS =head1 AUTHOR =head1 VERSION $Id: SQLCommand.pm,v 1.3 2004/07/01 06:37:11 don Exp $ =cut blib/lib/DBIx/Wrapper/StatementLoop.pm000444 001751 000000 00000004176 10212440752 020013 0ustar00donwheel000000 000000 # -*-perl-*- # Creation date: 2003-03-30 15:24:44 # Authors: Don # Change log: # $Id: StatementLoop.pm,v 1.8 2005/03/05 23:34:02 don Exp $ use strict; { package DBIx::Wrapper::StatementLoop; use vars qw($VERSION); $VERSION = do { my @r=(q$Revision: 1.8 $=~/\d+/g); sprintf "%d."."%02d"x$#r,@r }; use base 'DBIx::Wrapper::Statement'; sub new { my ($proto, $parent, $query) = @_; my $r = DBIx::Wrapper::Request->new($parent); $r->setQuery($query); $parent->_runPrePrepareHook($r); $query = $r->getQuery; my $sth = $parent->_getDatabaseHandle()->prepare($query); $r->setStatementHandle($sth); $parent->_runPostPrepareHook($r); $r->setStatementHandle($sth); unless ($sth) { $parent->_printDbiError("\nQuery was '$query'\n"); return $parent->setErr(0, $DBI::errstr); } my $self = bless {}, ref($proto) || $proto; $self->_setSth($sth); $self->_setParent($parent); $self->_setQuery($query); $self->_setRequestObj($r); return $self; } sub next { my ($self, $exec_args) = @_; if (scalar(@_) == 3) { $exec_args = [ $exec_args ] unless ref($exec_args); } $exec_args = [] unless $exec_args; my $r = $self->_getRequestObj; $r->setExecArgs($exec_args); my $sth = $self->_getSth; $self->_getParent()->_runPreExecHook($r); $exec_args = $r->getExecArgs; my $rv = $sth->execute(@$exec_args); $r->setExecReturnValue($rv); $self->_getParent()->_runPostExecHook($r); return $rv; } sub DESTROY { my ($self) = @_; my $sth = $self->_getSth; $sth->finish if $sth; } } 1; __END__ =pod =head1 NAME DBIx::Wrapper::StatementLoop - Used by DBIx::Wrapper to implement StatementLoop methods. =head1 SYNOPSIS =head1 DESCRIPTION =head1 METHODS =head1 EXAMPLES =head1 BUGS =head1 AUTHOR =head1 VERSION $Id: StatementLoop.pm,v 1.8 2005/03/05 23:34:02 don Exp $ =cut blib/lib/DBIx/Wrapper/Delegator.pm000444 001751 000000 00000001421 10325077523 017120 0ustar00donwheel000000 000000 # -*-perl-*- # Creation date: 2005-03-04 21:15:40 # Authors: Don # Change log: # $Id: Delegator.pm,v 1.5 2005/10/18 04:51:31 don Exp $ use strict; package DBIx::Wrapper::Delegator; use warnings; use vars qw($VERSION $AUTOLOAD); $VERSION = do { my @r=(q$Revision: 1.5 $=~/\d+/g); sprintf "%d."."%02d"x$#r,@r }; sub AUTOLOAD { my $self = shift; (my $func = $AUTOLOAD) =~ s/^.*::([^:]+)$/$1/; return undef if $func eq 'DESTROY'; my $key = $func; # turn method call into hash access return $self->{$func}; } =pod =head1 NAME DBIx::Wrapper::Delegator - =head1 SYNOPSIS =head1 DESCRIPTION =head1 METHODS =head1 EXAMPLES =head1 BUGS =head1 AUTHOR =head1 VERSION $Id: Delegator.pm,v 1.5 2005/10/18 04:51:31 don Exp $ =cut 1; blib/lib/DBIx/Wrapper/SelectExecLoop.pm000444 001751 000000 00000006774 10212444212 020074 0ustar00donwheel000000 000000 # -*-perl-*- # Creation date: 2004-04-21 10:45:30 # Authors: Don # Change log: # $Id: SelectExecLoop.pm,v 1.6 2005/03/06 00:02:18 don Exp $ use strict; { package DBIx::Wrapper::SelectExecLoop; use vars qw($VERSION); $VERSION = do { my @r=(q$Revision: 1.6 $=~/\d+/g); sprintf "%d."."%02d"x$#r,@r }; use base 'DBIx::Wrapper::Statement'; sub new { my ($proto, $parent, $query, $multi) = @_; my $r = DBIx::Wrapper::Request->new($parent); $r->setQuery($query); $parent->_runPrePrepareHook($r); $query = $r->getQuery; my $sth = $parent->_getDatabaseHandle()->prepare($query); $r->setStatementHandle($sth); $parent->_runPostPrepareHook($r); $r->setStatementHandle($sth); unless ($sth) { $parent->_printDbiError("\nQuery was '$query'\n"); return $parent->setErr(0, $DBI::errstr); } my $self = bless { _query => $query, _multi => $multi || '' }, ref($proto) || $proto; $self->_setSth($sth); $self->_setParent($parent); $self->_setQuery($query); $self->_setRequestObj($r); return $self; } sub next { my ($self, $exec_args) = @_; my $query = $self->_getQuery; my $sth = $self->_getSth; my $r = $self->_getRequestObj; $r->setExecArgs($exec_args); if ($$self{_multi}) { $self->_getParent()->_runPreExecHook($r); $exec_args = $r->getExecArgs; my $rv = $sth->execute(@$exec_args); $r->setExecReturnValue($rv); $self->_getParent()->_runPostExecHook($r); $self->_getParent()->_runPreFetchHook($r); $sth = $r->getStatementHandle; if ($rv) { my $rows = []; my $row = $sth->fetchrow_hashref($self->_getParent()->getNameArg); while ($row) { $r->setReturnVal($row); $self->_getParent()->_runPostFetchHook($r); $row = $r->getReturnVal; push @$rows, $row if $row; $self->_getParent()->_runPreFetchHook($r); $sth = $r->getStatementHandle; $row = $sth->fetchrow_hashref($self->_getParent()->getNameArg); } return $rows; } } else { $self->_getParent()->_runPreExecHook($r); $exec_args = $r->getExecArgs; my $rv = $sth->execute(@$exec_args); $r->setExecReturnValue($rv); $self->_getParent()->_runPostExecHook($r); if ($rv) { $self->_getParent()->_runPreFetchHook($r); $sth = $r->getStatementHandle; my $result = $sth->fetchrow_hashref($self->_getParent()->getNameArg); $r->setReturnVal($result); $self->_getParent()->_runPostFetchHook($r); $result = $r->getReturnVal; return $result; } } return undef; } sub DESTROY { my ($self) = @_; my $sth = $self->_getSth; $sth->finish if $sth; } } 1; __END__ =pod =head1 NAME DBIx::Wrapper::SelectExecLoop - Used by DBIx::Wrapper for SelectExecLoop methods. =head1 SYNOPSIS =head1 DESCRIPTION =head1 METHODS =head1 EXAMPLES =head1 BUGS =head1 AUTHOR =head1 VERSION $Id: SelectExecLoop.pm,v 1.6 2005/03/06 00:02:18 don Exp $ =cut blib/lib/DBIx/Wrapper.pm000444 001751 000000 00000212533 10326224460 015215 0ustar00donwheel000000 000000 # -*-perl-*- # Creation date: 2003-03-30 12:17:42 # Authors: Don # Change log: # $Id: Wrapper.pm,v 1.65 2005/10/21 17:45:20 don Exp $ # # Copyright (c) 2003-2005 Don Owens # # All rights reserved. This program is free software; you can # redistribute it and/or modify it under the same terms as Perl # itself. # Questions: # * When quoting tables, split on . or require more arguments for dbname, etc.? # * Convert named placeholders to normal ones or call the quote() method? # For DBD::mysql, quote() calls mysql_real_escape_string(), looks like execute() # calls mysql_stmt_bind_param in mysql 4.1 =pod =head1 NAME DBIx::Wrapper - A wrapper around the DBI =head1 SYNOPSIS use DBIx::Wrapper; my $db = DBIx::Wrapper->connect($dsn, $user, $auth, \%attr); my $db = DBIx::Wrapper->connect($dsn, $user, $auth, \%attr, { error_handler => sub { print $DBI::errstr }, debug_handler => sub { print $DBI::errstr }, }); my $dbi_obj = DBI->connect(...) my $db = DBIx::Wrapper->newFromDBI($dbi_obj); my $dbi_obj = $db->getDBI; my $rv = $db->insert($table, { id => 5, val => "myval", the_date => \"NOW()", }); my $rv = $db->insert($table, { id => 5, val => "myval", the_date => $db->command("NOW()"), }); my $rv = $db->replace($table, \%data); my $rv = $db->smartReplace($table, \%data) my $rv = $db->delete($table, \%keys); my $rv = $db->update($table, \%keys, \%data); my $rv = $db->smartUpdate($table, \%keys, \%data); my $row = $db->selectFromHash($table, \%keys, \@cols); my $row = $db->selectFromHashMulti($table, \%keys, \@cols); my $val = $db->selectValueFromHash($table, \%keys, $col); my $vals = $db->selectValueFromHashMulti($table, \%keys, \@cols); my $row = $db->nativeSelect($query, \@exec_args); my $loop = $db->nativeSelectExecLoop($query); foreach my $val (@vals) { my $row = $loop->next([ $val ]); } my $row = $db->nativeSelectWithArrayRef($query, \@exec_args); my $rows = $db->nativeSelectMulti($query, \@exec_args); my $loop = $db->nativeSelectMultiExecLoop($query) foreach my $val (@vals) { my $rows = $loop->next([ $val ]); } my $rows = $db->nativeSelectMultiWithArrayRef($query, \@exec_args); my $hash = $db->nativeSelectMapping($query, \@exec_args); my $hash = $db->nativeSelectDynaMapping($query, \@cols, \@exec_args); my $hash = $db->nativeSelectRecordMapping($query, \@exec_args); my $hash = $db->nativeSelectRecordDynaMapping($query, $col, \@exec_args); my $val = $db->nativeSelectValue($query, \@exec_args); my $vals = $db->nativeSelectValuesArray($query, \@exec_args); my $row = $db->abstractSelect($table, \@fields, \%where, \@order); my $rows = $db->abstractSelectMulti($table, \@fields, \%where, \@order); my $loop = $db->nativeSelectLoop($query, @exec_args); while (my $row = $loop->next) { my $id = $$row{id}; } my $rv = $db->nativeQuery($query, @exec_args); my $loop = $db->nativeQueryLoop("UPDATE my_table SET value=? WHERE id=?"); $loop->next([ 'one', 1]); $loop->next([ 'two', 2]); my $id = $db->getLastInsertId; $db->debugOn(\*FILE_HANDLE); $db->setNameArg($arg) $db->commit(); $db->ping(); $db->err(); Attributes Attributes accessed in DBIx::Wrapper object via hash access are passed on or retrieved from the underlying DBI object, e.g., $dbi_obj->{RaiseError} = 1 =head1 DESCRIPTION DBIx::Wrapper provides a wrapper around the DBI that makes it a bit easier on the programmer. This module allows you to execute a query with a single method call as well as make inserts easier, etc. It also supports running hooks at various stages of processing a query (see the section on Hooks). =head1 METHODS =cut use strict; package DBIx::Wrapper; use 5.006_00; # should have at least Perl 5.6.0 use warnings; no warnings 'once'; use vars qw($VERSION $AUTOLOAD $Heavy @CARP_NOT); use Carp (); # use Scalar::Util qw(refaddr); $Heavy = 0; BEGIN { $VERSION = '0.20pre1'; # update below in POD as well } use DBI; use DBIx::Wrapper::Request; use DBIx::Wrapper::SQLCommand; use DBIx::Wrapper::Statement; use DBIx::Wrapper::SelectLoop; use DBIx::Wrapper::SelectExecLoop; use DBIx::Wrapper::StatementLoop; use DBIx::Wrapper::Delegator; use DBIx::Wrapper::DBIDelegator; my %i_data; sub refaddr($) { my $obj = shift; my $pkg = ref($obj) or return undef; bless $obj, 'DBIx::Wrapper::Fake'; my $i = int($obj); bless $obj, $pkg; return $i; } sub _new { my ($proto) = @_; my $self = bless {}, ref($proto) || $proto; $i_data{ refaddr($self) } = {}; tie %$self, 'DBIx::Wrapper::DBIDelegator', $self; return $self; } sub _get_i_data { my $self = shift; return $i_data{ refaddr($self) }; } sub _get_i_val { my $self = shift; return $self->_get_i_data()->{ shift() }; } sub _set_i_val { my $self = shift; my $name = shift; my $val = shift; $self->_get_i_data()->{$name} = $val; } sub _delete_i_val { my $self = shift; my $name = shift; delete $self->_get_i_data()->{$name}; } sub import { my $class = shift; foreach my $e (@_) { if ($e eq ':heavy') { $Heavy = 1; } } } =pod =head2 connect($data_source, $username, $auth, \%attr, \%params) Connects to the given database. The first four parameters are the same parameters you would pass to the connect call when using DBI directly. The %params hash is optional and contains extra parameters to control the behaviour of DBIx::Wrapper itself. Following are the valid parameters. =over 4 =item error_handler and debug_handler These values should either be a reference to a subroutine, or a reference to an array whose first element is an object and whose second element is a method name to call on that object. The parameters passed to the error_handler callback are the current DBIx::Wrapper object and an error string, usually the query if appropriate. The parameters passed to the debug_handler callback are the current DBIx::Wrapper object, an error string, and the filehandle passed to the debugOn() method (defaults to STDERR). E.g., sub do_error { my ($db, $str) = @_; print $DBI::errstr; } sub do_debug { my ($db, $str, $fh) = @_; print $fh "query was: $str\n"; } my $db = DBIx::Wrapper->connect($ds, $un, $auth, \%attr, { error_handler => \&do_error, debug_handler => \&do_debug, }); =item db_style Used to control some database specific logic. The default value is 'mysql'. Currently, this is only used for the getLastInsertId() method. MSSQL is supported with a value of mssql for this parameter. =item heavy If set to a true value, any hashes returned will actually be objects on which you can call methods to get the values back. E.g., my $row = $db->nativeSelect($query); my $id = $row->id; or my $id = $row->{id}; =back =head2 new($data_source, $username, $auth, \%attr, \%params) An alias for connect(). =cut sub connect { my ($proto, $data_source, $username, $auth, $attr, $params) = @_; my $self = $proto->_new; $self->_set_i_val('_pre_prepare_hooks', []); $self->_set_i_val('_post_prepare_hooks', []); $self->_set_i_val('_pre_exec_hooks', []); $self->_set_i_val('_post_exec_hooks', []); $self->_set_i_val('_pre_fetch_hooks', []); $self->_set_i_val('_post_fetch_hooks', []); my $dsn = $data_source; $dsn = $self->_getDsnFromHash($data_source) if ref($data_source) eq 'HASH'; my $dbh = DBI->connect($dsn, $username, $auth, $attr); unless (ref($attr) eq 'HASH' and defined($$attr{PrintError}) and not $$attr{PrintError}) { # FIXME: make a way to set debug level here # $self->addDebugLevel(2); # print on error } unless ($dbh) { if ($self->_isDebugOn) { $self->_printDebug(Carp::longmess($DBI::errstr)); } else { $self->_printDbiError if not defined($$attr{PrintError}) or $$attr{PrintError}; } return undef; } $params = {} unless UNIVERSAL::isa($params, 'HASH'); $self->_setDatabaseHandle($dbh); $self->_setDataSource($data_source); $self->_setDataSourceStr($dsn); $self->_setUsername($username); $self->_setAuth($auth); $self->_setAttr($attr); $self->_setDisconnect(1); $self->_setErrorHandler($params->{error_handler}) if $params->{error_handler}; $self->_setDebugHandler($params->{debug_handler}) if $params->{debug_handler}; $self->_setDbStyle($params->{db_style}) if exists($params->{db_style}); $self->_setHeavy(1) if $params->{heavy}; my ($junk, $dbd_driver, @rest) = split /:/, $dsn; $self->_setDbdDriver(lc($dbd_driver)); return $self; } { no warnings; *new = \&connect; } =pod =head2 reconnect() Reconnect to the database using the same parameters that were given to the connect() method. It does not try to disconnect before attempting to connect again. =cut sub reconnect { my $self = shift; my $dsn = $self->_getDataSourceStr; my $dbh = DBI->connect($dsn, $self->_getUsername, $self->_getAuth, $self->_getAttr); if ($dbh) { $self->_setDatabaseHandle($dbh); return $self; } else { return undef; } } =pod =head2 disconnect() Disconnect from the database. This disconnects and frees up the underlying DBI object. =cut sub disconnect { my $self = shift; my $dbi_obj = $self->_getDatabaseHandle; $dbi_obj->disconnect if $dbi_obj; $self->_deleteDatabaseHandle; return 1; } =pod =head2 connect_one(\@cfg_list, \%attr) Connects to a random database out of the list. This is useful for connecting to a slave database out of a group for read-only access. Ths list should look similar to the following: my $cfg_list = [ { driver => 'mysql', host => 'db0.example.com', port => 3306, database => 'MyDB', user => 'dbuser', auth => 'dbpwd', attr => { RaiseError => 1 }, weight => 1, }, { driver => 'mysql', host => 'db1.example.com', port => 3306, database => 'MyDB', user => 'dbuser', auth => 'dbpwd', attr => { RaiseError => 1 }, weight => 2, }, ]; where the weight fields are optional (defaulting to 1). The attr field is also optional and corresponds to the 4th argument to DBI's connect() method. The \%attr passed to this method is an optional parameter specifying the defaults for \%attr to be passed to the connect() method. The attr field in the config for each database in the list overrides any in the \%attr parameter passed into the method. You may also pass the DSN string for the connect() method as the 'dsn' field in each config instead of the separate driver, host, port, and database fields, e.g., my $cfg_list = [ { dsn => 'dbi:mysql:host=db0.example.com;database=MyDB;port=3306', user => 'dbuser', auth => 'dbpwd', attr => { RaiseError => 1 }, weight => 1, }, ]; =cut sub connect_one { my $proto = shift; my $cfg_list = shift; my $attr = shift || {}; return undef unless $cfg_list and @$cfg_list; # make copy so we don't distrub the original datastructure $cfg_list = [ @$cfg_list ]; my $db = 0; while (not $db and scalar(@$cfg_list) > 0) { my ($cfg, $index) = $proto->_pick_one($cfg_list); my $this_attr = $cfg->{attr} || {}; $this_attr = { %$attr, %$this_attr }; eval { local($SIG{__DIE__}); $db = $proto->connect($cfg->{dsn} || $cfg, $cfg->{user}, $cfg->{auth}, $this_attr); }; splice(@$cfg_list, $index, 1) unless $db; } return $db; } sub _pick_one { my $proto = shift; my $cfg_list = shift; return undef unless $cfg_list and @$cfg_list; $cfg_list = [ grep { not defined($_->{weight}) or $_->{weight} != 0 } @$cfg_list ]; my $total_weight = 0; foreach my $cfg (@$cfg_list) { $total_weight += $cfg->{weight} || 1; } my $target = rand($total_weight); my $accumulated = 0; my $pick; my $index = 0; foreach my $cfg (@$cfg_list) { $accumulated += $cfg->{weight} || 1; if ($target < $accumulated) { $pick = $cfg; last; } $index++; } return wantarray ? ($pick, $index) : $pick; } sub _getDsnFromHash { my $self = shift; my $data_source = shift; my @dsn; push @dsn, "database=$$data_source{database}" if $data_source->{database}; push @dsn, "host=$$data_source{host}" if $data_source->{host}; push @dsn, "port=$$data_source{port}" if $data_source->{port}; push @dsn, "mysql_connect_timeout=$$data_source{mysql_connect_timeout}" if $data_source->{mysql_connect_timeout}; my $driver = $data_source->{driver} || $data_source->{type}; if ($data_source->{timeout}) { if ($driver eq 'mysql') { push @dsn, "mysql_connect_timeout=$$data_source{timeout}"; } } return "dbi:$driver:" . join(';', @dsn); } sub addDebugLevel { my $self = shift; my $level = shift; my $cur_level = $self->_get_i_val('_debug_level'); $cur_level |= $level; $self->_set_i_val('_debug_level', $cur_level); } sub getDebugLevel { return shift()->_get_i_data('_debug_level'); } =pod =head2 newFromDBI($dbh) Returns a new DBIx::Wrapper object from a DBI object that has already been created. Note that when created this way, disconnect() will not be called automatically on the underlying DBI object when the DBIx::Wrapper object goes out of scope. =cut sub newFromDBI { my ($proto, $dbh) = @_; my $self = $proto->_new; $self->_setDatabaseHandle($dbh); return $self; } *new_from_dbi = \&newFromDBI; =pod =head2 getDBI() Return the underlying DBI object used to query the database. =cut sub getDBI { my ($self) = @_; return $self->_getDatabaseHandle; } *get_dbi = \&getDBI; *getDbi = \&getDBI; sub _insert_replace { my ($self, $operation, $table, $data) = @_; my @values; my @fields; my @place_holders; while (my ($field, $value) = each %$data) { push @fields, $field; if (UNIVERSAL::isa($value, 'DBIx::Wrapper::SQLCommand')) { push @place_holders, $value->asString; } elsif (ref($value) eq 'SCALAR') { push @place_holders, $$value; } else { $value = '' unless defined($value); push @place_holders, '?'; push @values, $value; } } my $fields = join(",", @fields); my $place_holders = join(",", @place_holders); my $query = qq{$operation INTO $table ($fields) values ($place_holders)}; my ($sth, $rv) = $self->_getStatementHandleForQuery($query, \@values); return $sth unless $sth; $sth->finish; return $rv; } =pod =head2 insert($table, \%data) Insert the provided row into the database. $table is the name of the table you want to insert into. %data is the data you want to insert -- a hash with key/value pairs representing a row to be insert into the database. =cut sub insert { my ($self, $table, $data) = @_; return $self->_insert_replace('INSERT', $table, $data); } =pod =head2 replace($table, \%data) Same as insert(), except does a REPLACE instead of an INSERT for databases which support it. =cut sub replace { my ($self, $table, $data) = @_; my $style = lc($self->_getDbStyle); if ($style eq 'mssql') { # mssql doesn't support replace, so do an insert instead return $self->_insert_replace('INSERT', $table, $data); } else { return $self->_insert_replace('REPLACE', $table, $data); } } =pod =head2 smartReplace($table, \%data) This method is MySQL specific. If $table has an auto_increment column, the return value will be the value of the auto_increment column. So if that column was specified in \%data, that value will be returned, otherwise, an insert will be performed and the value of LAST_INSERT_ID() will be returned. If there is no auto_increment column, but primary keys are provided, the row containing the primary keys will be returned. Otherwise, a true value will be returned upon success. =cut sub smartReplace { my ($self, $table, $data, $keys) = @_; if (0 and $keys) { # ignore $keys for now } else { my $dbh = $self->_getDatabaseHandle; my $query = qq{DESCRIBE $table}; my $sth = $self->_getStatementHandleForQuery($query); return $sth unless $sth; my $auto_incr = undef; my $key_list = []; my $info_list = []; while (my $info = $sth->fetchrow_hashref('NAME_lc')) { push @$info_list, $info; push @$key_list, $$info{field} if lc($$info{key}) eq 'pri'; if ($$info{extra} =~ /auto_increment/i) { $auto_incr = $$info{field}; } } my $orig_auto_incr = $auto_incr; $auto_incr = lc($auto_incr); my $keys_provided = []; my $key_hash = { map { (lc($_) => 1) } @$key_list }; my $auto_incr_provided = 0; foreach my $key (keys %$data) { push @$keys_provided, $key if exists($$key_hash{lc($key)}); if (lc($key) eq $auto_incr) { $auto_incr_provided = 1; last; } } if (@$keys_provided) { # do replace and return the value of this field my $rv = $self->replace($table, $data); return $rv unless $rv; if (not defined($orig_auto_incr) or $orig_auto_incr eq '') { my %hash = map { ($_ => $$data{$_}) } @$keys_provided; my $row = $self->selectFromHash($table, \%hash); return $row if $row and %$row; return undef; } else { return $$data{$orig_auto_incr}; } } else { # do insert and return last insert id my $rv = $self->insert($table, $data); return $rv unless $rv; if (not defined($orig_auto_incr) or $orig_auto_incr eq '') { # FIXME: what do we do here? return 1; } else { my $id = $self->getLastInsertId(undef, undef, $table, $orig_auto_incr); return $id; } } } } *smart_replace = \&smartReplace; =pod =head2 delete($table, \%keys), delete($table, \@keys) Delete rows from table $table using the key/value pairs in %keys to specify the WHERE clause of the query. Multiple key/value pairs are joined with 'AND' in the WHERE clause. The cols parameter can optionally be an array ref instead of a hashref. E.g. $db->delete($table, [ key1 => $val1, key2 => $val2 ]) This is so that the order of the parameters in the WHERE clause are kept in the same order. This is required to use the correct multi field indexes in some databases. =cut sub delete { my ($self, $table, $keys) = @_; unless ($keys and (UNIVERSAL::isa($keys, 'HASH') or UNIVERSAL::isa($keys, 'ARRAY'))) { return $self->setErr(-1, 'DBIx::Wrapper: No keys passed to update()'); } my @keys; my @values; if (ref($keys) eq 'ARRAY') { # allow this to maintain order in the WHERE clause in # order to use the right indexes my @copy = @$keys; while (my $key = shift @copy) { push @keys, $key; my $val = shift @copy; # shift off the value } $keys = { @$keys }; } else { @keys = keys %$keys; } push @values, @$keys{@keys}; my $where = join(" AND ", map { "$_=?" } @keys); my $query = qq{DELETE FROM $table WHERE $where}; my ($sth, $rv) = $self->_getStatementHandleForQuery($query, \@values); return $sth unless $sth; $sth->finish; return $rv; } =pod =head2 update($table, \%keys, \%data), update($table, \@keys, \%data) Update the table using the key/value pairs in %keys to specify the WHERE clause of the query. %data contains the new values for the row(s) in the database. The keys parameter can optionally be an array ref instead of a hashref. E.g., $db->update($table, [ key1 => $val1, key2 => $val2 ], \%data); This is so that the order of the parameters in the WHERE clause are kept in the same order. This is required to use the correct multi field indexes in some databases. =cut sub update { my ($self, $table, $keys, $data) = @_; unless ($keys and (UNIVERSAL::isa($keys, 'HASH') or UNIVERSAL::isa($keys, 'ARRAY'))) { return $self->setErr(-1, 'DBIx::Wrapper: No keys passed to update()'); } unless ($data and UNIVERSAL::isa($data, 'HASH')) { return $self->setErr(-1, 'DBIx::Wrapper: No values passed to update()'); } unless (%$data) { return "0E"; } my @fields; my @values; my @set; while (my ($field, $value) = each %$data) { push @fields, $field; if (UNIVERSAL::isa($value, 'DBIx::Wrapper::SQLCommand')) { push @set, "$field=" . $value->asString; } elsif (ref($value) eq 'SCALAR') { push @set, "$field=" . $$value; } else { $value = "" unless defined $value; push @set, "$field=?"; push @values, $value; } } my @keys; if (ref($keys) eq 'ARRAY') { # allow this to maintain order in the WHERE clause in # order to use the right indexes my @copy = @$keys; while (my $key = shift @copy) { push @keys, $key; my $val = shift @copy; # shift off the value } $keys = { @$keys }; } else { @keys = keys %$keys; } push @values, @$keys{@keys}; my $set = join(",", @set); my $where = join(" AND ", map { "$_=?" } @keys); my $query = qq{UPDATE $table SET $set WHERE $where}; my ($sth, $rv) = $self->_getStatementHandleForQuery($query, \@values); return $sth unless $sth; $sth->finish; return $rv; } =pod =head2 selectFromHash($table, \%keys, \@cols); Select from table $table using the key/value pairs in %keys to specify the WHERE clause of the query. Multiple key/value pairs are joined with 'AND' in the WHERE clause. Returns a single row as a hashref. If %keys is empty or not passed, it is treated as "SELECT * FROM $table" with no WHERE clause. @cols is a list of columns you want back. If nothing is passed in @cols, all columns will be returned. If a value in the %keys hash is an array ref, the resulting query will search for records with any of those values. E.g., my $row = $db->selectFromHash('the_table', { id => [ 5, 6, 7 ] }); will result in a query like SELECT * FROM the_table WHERE (id=5 OR id=6 OR id=7) The call my $row = $db->selectFromHash('the_table', { id => [ 5, 6, 7 ], the_val => 'ten' }); will result in a query like SELECT * FROM the_table WHERE (id=5 OR id=6 OR id=7) AND the_val="ten" or, if a value was passed in for \@cols, e.g., my $row = $db->selectFromHash('the_table', { id => [ 5, 6, 7 ], the_val => 'ten' }, [ 'id' ]); the resulting query would be SELECT id FROM the_table WHERE (id=5 OR id=6 OR id=7) AND the_val="ten" =cut sub selectFromHash { my ($self, $table, $keys, $cols) = @_; my $sth = $self->_get_statement_handle_for_select_from_hash($table, $keys, $cols); return $sth unless $sth; my $info = $sth->fetchrow_hashref; my $rv; if ($info and %$info) { $rv = $info; } else { $rv = wantarray ? () : undef; } $sth->finish; return $rv; } *select_from_hash = \&selectFromHash; sub _get_statement_handle_for_select_from_hash { my ($self, $table, $keys, $cols) = @_; my $query; my $col_list = '*'; if (ref($cols) eq 'ARRAY') { if (@$cols) { $col_list = join(',', @$cols); } } elsif (defined($cols) and $cols ne '') { $col_list = $cols; } if ($keys and ((ref($keys) eq 'HASH' and %$keys) or (ref($keys) eq 'ARRAY' and @$keys))) { my ($where, $exec_args) = $self->_get_clause_for_select_from_hash($keys); $query = qq{SELECT $col_list FROM $table WHERE $where}; return $self->_getStatementHandleForQuery($query, $exec_args); } else { $query = qq{SELECT $col_list FROM $table}; return $self->_getStatementHandleForQuery($query); } } sub _get_clause_for_select_from_hash { my $self = shift; my $data = shift; my $parent_key = shift; my @values; my @where; if (ref($data) eq 'HASH') { my @keys = sort keys %$data; foreach my $key (@keys) { my $val = $data->{$key}; if (ref($val)) { my ($clause, $exec_args) = $self->_get_clause_for_select_from_hash($val, $key); push @where, "($clause)"; push @values, @$exec_args if $exec_args; } else { push @where, "$key=?"; push @values, $val; } } my $where = join(" AND ", @where); return wantarray ? ($where, \@values) : $where; } elsif (ref($data) eq 'ARRAY') { foreach my $val (@$data) { if (ref($val)) { my ($clause, $exec_args) = $self->_get_clause_for_select_from_hash($val, $parent_key); push @where, "($clause)"; push @values, @$exec_args if $exec_args; } else { push @where, "$parent_key=?"; push @values, $val; } } my $where = join(" OR ", @where); return wantarray ? ($where, \@values) : $where; } else { return wantarray ? ($data, []) : $data; } } =pod =head2 selectFromHashMulti($table, \%keys, \@cols) Like selectFromHash(), but returns all rows in the result. Returns a reference to an array of hashrefs. =cut sub selectFromHashMulti { my ($self, $table, $keys, $cols) = @_; my $sth = $self->_get_statement_handle_for_select_from_hash($table, $keys, $cols); return $sth unless $sth; my $results = []; while (my $info = $sth->fetchrow_hashref) { push @$results, $info; } $sth->finish; return $results; } *select_from_hash_multi = \&selectFromHashMulti; =pod =head2 selectValueFromHash($table, \%keys, $col) Combination of nativeSelectValue() and selectFromHash(). Returns the first column from the result of a query given by $table and %keys, as in selectFromHash(). $col is the column to return. =cut sub selectValueFromHash { my ($self, $table, $keys, $col) = @_; my $sth = $self->_get_statement_handle_for_select_from_hash($table, $keys, $col); return $sth unless $sth; my $info = $sth->fetchrow_arrayref; $sth->finish; my $rv; if ($info and @$info) { return $info->[0]; } else { return wantarray ? () : undef; } } *select_value_from_hash = \&selectValueFromHash; =pod =head2 selectValueFromHashMulti($table, \%keys, \@cols) Like selectValueFromhash(), but returns the first column of all rows in the result. =cut sub selectValueFromHashMulti { my ($self, $table, $keys, $col) = @_; my $sth = $self->_get_statement_handle_for_select_from_hash($table, $keys, $col); return $sth unless $sth; my $results = []; while (my $info = $sth->fetchrow_arrayref) { push @$results, $info->[0]; } $sth->finish; return $results; } *select_value_from_hash_multi = \&selectValueFromHashMulti; =pod =head2 smartUpdate($table, \%keys, \%data) Same as update(), except that a check is first made to see if there are any rows matching the data in %keys. If so, update() is called, otherwise, insert() is called. =cut sub smartUpdate { my ($self, $table, $keys, $data) = @_; unless (ref($data) eq 'HASH' and %$data) { return "0E"; } my $col; if (ref($keys) eq 'HASH') { $col = (keys %$keys)[0]; } my $info = $self->select_from_hash($table, $keys, $col); if ($info and %$info) { return $self->update($table, $keys, $data); } else { my %new_data = %$data; while (my ($key, $value) = each %$keys) { $new_data{$key} = $value unless exists $new_data{$key}; } return $self->insert($table, \%new_data); } } *smart_update = \&smartUpdate; sub _runHandler { my ($self, $handler_info, @args) = @_; return undef unless ref($handler_info); my ($handler, $custom_args) = @$handler_info; $custom_args = [] unless $custom_args; unshift @args, $self; if (ref($handler) eq 'ARRAY') { my $method = $handler->[1]; $handler->[0]->$method(@args, @$custom_args); } else { $handler->(@args, @$custom_args); } return 1; } sub _runHandlers { my ($self, $handlers, $r) = @_; return undef unless $handlers; my $rv = $r->OK; foreach my $handler_info (reverse @$handlers) { my ($handler, $custom_args) = @$handler_info; $custom_args = [] unless $custom_args; if (ref($handler) eq 'ARRAY') { my $method = $handler->[1]; $rv = $handler->[0]->$method($r); } else { $rv = $handler->($r); } last unless $rv == $r->DECLINED; } return $rv; } sub _defaultPrePrepareHandler { my $r = shift; return $r->OK; } sub _defaultPostPrepareHandler { my $r = shift; return $r->OK; } sub _defaultPreExecHandler { my $r = shift; return $r->OK; } sub _defaultPostExecHandler { my $r = shift; return $r->OK; } sub _defaultPreFetchHandler { my $r = shift; return $r->OK; } sub _defaultPostFetchHandler { my $r = shift; return $r->OK; } sub _runGenericHook { my ($self, $r, $default_handler, $custom_handler_field) = @_; my $handlers = [ $default_handler ]; if ($self->shouldBeHeavy) { if ($custom_handler_field eq '_post_fetch_hooks') { push @$handlers, [ \&_heavyPostFetchHook ]; } } my $custom_handlers = $self->_get_i_val($custom_handler_field); push @$handlers, @$custom_handlers if $custom_handlers; return $self->_runHandlers($handlers, $r); } sub _runPrePrepareHook { my $self = shift; my $r = shift; my $handlers = [ [ \&_defaultPrePrepareHandler ] ]; my $custom_handlers = $self->_get_i_val('_pre_prepare_hooks'); push @$handlers, @$custom_handlers if $custom_handlers; return $self->_runHandlers($handlers, $r); } sub _runPostPrepareHook { my $self = shift; my $r = shift; my $handlers = [ [ \&_defaultPostPrepareHandler ] ]; my $custom_handlers = $self->_get_i_val('_post_prepare_hooks'); push @$handlers, @$custom_handlers if $custom_handlers; return $self->_runHandlers($handlers, $r); } sub _runPreExecHook { my $self = shift; my $r = shift; my $handlers = [ [ \&_defaultPreExecHandler ] ]; my $custom_handlers = $self->_get_i_val('_pre_exec_hooks'); push @$handlers, @$custom_handlers if $custom_handlers; return $self->_runHandlers($handlers, $r); } sub _runPostExecHook { my $self = shift; my $r = shift; return $self->_runGenericHook($r, [ \&_defaultPostExecHandler ], '_post_exec_hooks'); } sub _runPreFetchHook { my $self = shift; my $r = shift; return $self->_runGenericHook($r, [ \&_defaultPreFetchHandler ], '_pre_fetch_hooks'); } sub _runPostFetchHook { my $self = shift; my $r = shift; return $self->_runGenericHook($r, [ \&_defaultPostFetchHandler ], '_post_fetch_hooks'); } sub _heavyPostFetchHook { my $r = shift; my $row = $r->getReturnVal; if (ref($row) eq 'HASH') { $r->setReturnVal(bless($row, 'DBIx::Wrapper::Delegator')); } elsif (ref($row) eq 'ARRAY') { # do nothing for now } } sub _bind_named_place_holders { my $self = shift; my $query = shift; my $exec_args = shift; my $dbi = $self->_getDatabaseHandle; # $query =~ s/(?quote($exec_args->{$2})/eg; # return wantarray ? ($query, []) : $query; my @new_args; # $query =~ s/(?{$2}); '?'/eg; # Convert :: to : instead of treating it as a placeholder $query =~ s{(::)|:([\'\"]?)(\w+)\2}{ if (defined($1) and $1 eq '::' ) { ':' . (defined $2 ? $2 : '') . (defined $3 ? $3 : '') . (defined $2 ? $2 : '') } else { push(@new_args, $exec_args->{$3}); '?' } }eg; return wantarray ? ($query, \@new_args) : $query; } sub _getStatementHandleForQuery { my ($self, $query, $exec_args) = @_; if (scalar(@_) == 3) { my $type = ref($exec_args); if ($type eq 'HASH') { # okay ($query, $exec_args) = $self->_bind_named_place_holders($query, $exec_args); } elsif ($type eq 'ARRAY') { # okay -- leave as is } else { $exec_args = [ $exec_args ]; } } $exec_args = [] unless $exec_args; $self->_printDebug($query); my $r = DBIx::Wrapper::Request->new($self); $r->setQuery($query); $r->setExecArgs($exec_args); $self->_runPrePrepareHook($r); $query = $r->getQuery; $exec_args = $r->getExecArgs; my $dbh = $self->_getDatabaseHandle; my $sth = $dbh->prepare($query); $r->setStatementHandle($sth); $r->setErrorStr($sth ? $dbh->errstr : ''); $self->_runPostPrepareHook($r); unless ($sth) { if ($self->_isDebugOn) { $self->_printDebug(Carp::longmess($dbh->errstr) . "\nQuery was '$query'\n"); } else { $self->_printDbiError("\nQuery was '$query'\n"); } return wantarray ? ($self->setErr(0, $dbh->errstr), undef) : $self->setErr(0, $dbh->errstr); } $r->setQuery($query); $r->setExecArgs($exec_args); $self->_runPreExecHook($r); $exec_args = $r->getExecArgs; my $rv = $sth->execute(@$exec_args); $r->setExecReturnValue($rv); $r->setErrorStr($rv ? '' : $dbh->errstr); $self->_runPostExecHook($r); $rv = $r->getExecReturnValue; $sth = $r->getStatementHandle; unless ($rv) { if ($self->_isDebugOn) { $self->_printDebug(Carp::longmess($dbh->errstr) . "\nQuery was '$query'\n"); } else { $self->_printDbiError("\nQuery was '$query'\n"); } return wantarray ? ($self->setErr(1, $dbh->errstr), undef) : $self->setErr(1, $dbh->errstr); } return wantarray ? ($sth, $rv, $r) : $sth; } sub prepare_no_hooks { my $self = shift; my $query = shift; my $dbi_obj = $self->getDBI; my $sth = $dbi_obj->prepare($query); return $sth; } *prepare_no_handlers = \&prepare_no_hooks; =pod =head2 nativeSelect($query, \@exec_args) Executes the query in $query and returns a single row result (as a hash ref). If there are multiple rows in the result, the rest get silently dropped. @exec_args are the same arguments you would pass to an execute() called on a DBI object. Returns undef on error. =cut sub nativeSelect { my ($self, $query, $exec_args) = @_; my ($sth, $rv, $r); if (scalar(@_) == 3) { ($sth, $rv, $r) = $self->_getStatementHandleForQuery($query, $exec_args); } else { ($sth, $rv, $r) = $self->_getStatementHandleForQuery($query); } return $sth unless $sth; $self->_runPreFetchHook($r); $sth = $r->getStatementHandle; my $result = $sth->fetchrow_hashref($self->getNameArg); $r->setReturnVal($result); $self->_runPostFetchHook($r); $result = $r->getReturnVal; $sth->finish; return $result; } *read = \&nativeSelect; *native_select = \&nativeSelect; =pod =head2 nativeSelectExecLoop($query) Like nativeSelect(), but returns a loop object that can be used to execute the same query over and over with different bind parameters. This does a single DBI prepare() instead of a new prepare() for select. E.g., my $loop = $db->nativeSelectExecLoop("SELECT * FROM mytable WHERE id=?"); foreach my $id (@ids) { my $row = $loop->next([ $id ]); } =cut # added for v 0.08 sub nativeSelectExecLoop { my ($self, $query) = @_; return DBIx::Wrapper::SelectExecLoop->new($self, $query); } *native_select_exec_loop = \&nativeSelectExecLoop; =pod =head2 nativeSelectWithArrayRef($query, \@exec_args) Like nativeSelect(), but return a reference to an array instead of a hash. Returns undef on error. If there are no results from the query, a reference to an empty array is returned. =cut sub nativeSelectWithArrayRef { my ($self, $query, $exec_args) = @_; my ($sth, $rv, $r); if (scalar(@_) == 3) { ($sth, $rv, $r) = $self->_getStatementHandleForQuery($query, $exec_args); } else { ($sth, $rv, $r) = $self->_getStatementHandleForQuery($query); } return $sth unless $sth; $self->_runPreFetchHook($r); $sth = $r->getStatementHandle; my $result = $sth->fetchrow_arrayref; $r->setReturnVal($result); $self->_runPostFetchHook($r); $result = $r->getReturnVal; $sth->finish; return [] unless $result and ref($result) =~ /ARRAY/; # have to make copy because recent version of DBI now # return the same array reference each time return [ @$result ]; } *native_select_with_array_ref = \&nativeSelectArrayWithArrayRef; =pod =head2 nativeSelectMulti($query, \@exec_args) Executes the query in $query and returns an array of rows, where each row is a hash representing a row of the result. Returns undef on error. If there are no results for the query, an empty array ref is returned. =cut sub nativeSelectMulti { my ($self, $query, $exec_args) = @_; my ($sth, $rv, $r); if (scalar(@_) == 3) { ($sth, $rv, $r) = $self->_getStatementHandleForQuery($query, $exec_args); } else { ($sth, $rv, $r) = $self->_getStatementHandleForQuery($query); } return $sth unless $sth; $self->_runPreFetchHook($r); $sth = $r->getStatementHandle; my $rows = []; my $row = $sth->fetchrow_hashref($self->getNameArg); while ($row) { $r->setReturnVal($row); $self->_runPostFetchHook($r); $row = $r->getReturnVal; push @$rows, $row; $self->_runPreFetchHook($r); $sth = $r->getStatementHandle; $row = $sth->fetchrow_hashref($self->getNameArg) } my $col_names = $sth->{$self->getNameArg}; $self->_set_i_val('_last_col_names', $col_names); $sth->finish; return $rows; } *readArray = \&nativeSelectMulti; *native_select_multi = \&nativeSelectMulti; =pod =head2 nativeSelectMultiExecLoop($query) Like nativeSelectExecLoop(), but returns an array of rows, where each row is a hash representing a row of the result. =cut sub nativeSelectMultiExecLoop { my ($self, $query) = @_; return DBIx::Wrapper::SelectExecLoop->new($self, $query, 1); } *native_select_multi_exec_loop = \&nativeSelectMultiExecLoop; =pod =head2 nativeSelectMultiWithArrayRef($query, \@exec_args) Like nativeSelectMulti(), but return a reference to an array of arrays instead of to an array of hashes. Returns undef on error. =cut sub nativeSelectMultiWithArrayRef { my ($self, $query, $exec_args) = @_; my ($sth, $rv, $r); if (scalar(@_) == 3) { ($sth, $rv, $r) = $self->_getStatementHandleForQuery($query, $exec_args); } else { ($sth, $rv, $r) = $self->_getStatementHandleForQuery($query); } return $sth unless $sth; $self->_runPreFetchHook($r); $sth = $r->getStatementHandle; my $list = []; my $result = $sth->fetchrow_arrayref; while ($result) { $r->setReturnVal($result); $self->_runPostFetchHook($r); $result = $r->getReturnVal; # have to make copy because recent version of DBI now # return the same array reference each time push @$list, [ @$result ]; $result = $sth->fetchrow_arrayref; } $sth->finish; return $list; } *native_select_multi_with_array_ref = \&nativeSelectMultiWithArrayRef; =pod =head2 nativeSelectMapping($query, \@exec_args) Executes the given query and returns a reference to a hash containing the first and second columns of the results as key/value pairs. =cut sub nativeSelectMapping { my ($self, $query, $exec_args) = @_; if (scalar(@_) == 3) { $self->nativeSelectDynaMapping($query, [ 0, 1 ], $exec_args); } else { $self->nativeSelectDynaMapping($query, [ 0, 1 ]); } } *native_select_mapping = \&nativeSelectMapping; =pod =head2 nativeSelectDynaMapping($query, \@cols, \@exec_args) Similar to nativeSelectMapping() except you specify which columns to use for the key/value pairs in the return hash. If the first element of @cols starts with a digit, then @cols is assumed to contain indexes for the two columns you wish to use. Otherwise, @cols is assumed to contain the field names for the two columns you wish to use. For example, nativeSelectMapping($query, \@exec_args) is equivalent (and in fact calls) to nativeSelectDynaMapping($query, [ 0, 1 ], $exec_args). =cut # FIXME: return undef on error sub nativeSelectDynaMapping { my ($self, $query, $cols, $exec_args) = @_; my ($first, $second) = @$cols; my $map = {}; if ($first =~ /^\d/) { my $rows; if (scalar(@_) == 4) { $rows = $self->nativeSelectMultiWithArrayRef($query, $exec_args); } else { $rows = $self->nativeSelectMultiWithArrayRef($query); } foreach my $row (@$rows) { $$map{$$row[$first]} = $$row[$second]; } } else { my $rows; if (scalar(@_) == 4) { $rows = $self->nativeSelectMulti($query, $exec_args); } else { $rows = $self->nativeSelectMulti($query); } foreach my $row (@$rows) { $$map{$$row{$first}} = $$row{$second}; } } return $map; } *native_select_dyna_mapping = \&nativeSelectDynaMapping; =pod =head2 nativeSelectRecordMapping($query, \@exec_args) Similar to nativeSelectMapping(), except the values in the hash are references to the corresponding record (as a hash). =cut sub nativeSelectRecordMapping { my ($self, $query, $exec_args) = @_; if (scalar(@_) == 3) { return $self->nativeSelectRecordDynaMapping($query, 0, $exec_args); } else { return $self->nativeSelectRecordDynaMapping($query, 0); } } *native_select_record_mapping = \&nativeSelectRecordMapping; =pod =head2 nativeSelectRecordDynaMapping($query, $col, \@exec_args) Similar to nativeSelectRecordMapping(), except you specify which column is the key in each key/value pair in the hash. If $col starts with a digit, then it is assumed to contain the index for the column you wish to use. Otherwise, $col is assumed to contain the field name for the two columns you wish to use. =cut # FIXME: return undef on error sub nativeSelectRecordDynaMapping { my ($self, $query, $col, $exec_args) = @_; my $map = {}; if ($col =~ /^\d/) { my $rows; if (scalar(@_) == 4) { $rows = $self->nativeSelectMulti($query, $exec_args); } else { $rows = $self->nativeSelectMulti($query); } my $names = $self->_get_i_val('_last_col_names'); my $col_name = $$names[$col]; foreach my $row (@$rows) { $$map{$$row{$col_name}} = $row; } } else { my $rows; if (scalar(@_) == 4) { $rows = $self->nativeSelectMulti($query, $exec_args); } else { $rows = $self->nativeSelectMulti($query); } foreach my $row (@$rows) { $$map{$$row{$col}} = $row; } } return $map; } *native_select_record_dyna_mapping = \&nativeSelectRecordDynaMapping; sub _getSqlObj { # return SQL::Abstract->new(case => 'textbook', cmp => '=', logic => 'and'); require SQL::Abstract; return SQL::Abstract->new(case => 'textbook', cmp => '='); } =pod =head2 nativeSelectValue($query, \@exec_args) Returns a single value, the first column from the first row of the result. Returns undef on error or if there are no rows in the result. Note this may be the same value returned for a NULL value in the result. =cut sub nativeSelectValue { my ($self, $query, $exec_args) = @_; my $row; if (scalar(@_) == 3) { $row = $self->nativeSelectWithArrayRef($query, $exec_args); } else { $row = $self->nativeSelectWithArrayRef($query); } if ($row and @$row) { return $row->[0]; } return undef; } *native_select_value = \&nativeSelectValue; =pod =head2 nativeSelectValuesArray($query, \@exec_args) Like nativeSelectValue(), but return multiple values, e.g., return an array of ids for the query "SELECT id FROM WHERE color_pref='red'". =cut sub nativeSelectValuesArray { my ($self, $query, $exec_args) = @_; my $rows; if (scalar(@_) == 3) { $rows = $self->nativeSelectMultiWithArrayRef($query, $exec_args); } else { $rows = $self->nativeSelectMultiWithArrayRef($query); } return undef unless $rows; return [ map { $_->[0] } @$rows ]; } *native_select_values_array = \&nativeSelectValuesArray; =pod =head2 abstractSelect($table, \@fields, \%where, \@order) Same as nativeSelect() except uses SQL::Abstract to generate the SQL. See the POD for SQL::Abstract for usage. You must have SQL::Abstract installed for this method to work. =cut sub abstractSelect { my ($self, $table, $fields, $where, $order) = @_; my $sql_obj = $self->_getSqlObj; my ($query, @bind) = $sql_obj->select($table, $fields, $where, $order); if (@bind) { return $self->nativeSelect($query, \@bind); } else { return $self->nativeSelect($query); } } *abstract_select = \&abstractSelect; =pod =head2 abstractSelectMulti($table, \@fields, \%where, \@order) Same as nativeSelectMulti() except uses SQL::Abstract to generate the SQL. See the POD for SQL::Abstract for usage. You must have SQL::Abstract installed for this method to work. =cut sub abstractSelectMulti { my ($self, $table, $fields, $where, $order) = @_; my $sql_obj = $self->_getSqlObj; my ($query, @bind) = $sql_obj->select($table, $fields, $where, $order); if (@bind) { return $self->nativeSelectMulti($query, \@bind); } else { return $self->nativeSelectMulti($query); } } *abstract_select_multi = \&abstractSelectMulti; =pod =head2 nativeSelectLoop($query, @exec_args) Executes the query in $query, then returns an object that allows you to loop through one result at a time, e.g., my $loop = $db->nativeSelectLoop("SELECT * FROM my_table"); while (my $row = $loop->next) { my $id = $$row{id}; } To get the number of rows selected, you can call the rowCountCurrent() method on the loop object, e.g., my $loop = $db->nativeSelectLoop("SELECT * FROM my_table"); my $rows_in_result = $loop->rowCountCurrent; The count() method is an alias for rowCountCurrent(). To get the number of rows returned by next() so far, use the rowCountTotal() method. =cut sub nativeSelectLoop { my ($self, $query, $exec_args) = @_; $self->_printDebug($query); if (scalar(@_) == 3) { return DBIx::Wrapper::SelectLoop->new($self, $query, $exec_args); } else { return DBIx::Wrapper::SelectLoop->new($self, $query); } } *readLoop = \&nativeSelectLoop; *native_select_loop = \&nativeSelectLoop; =pod =head2 nativeQuery($query, @exec_args) Executes the query in $query and returns true if successful. This is typically used for deletes and is a catchall for anything the methods provided by this module don't take into account. =cut sub nativeQuery { my ($self, $query, $exec_args) = @_; my ($sth, $rv, $r); if (scalar(@_) == 3) { ($sth, $rv, $r) = $self->_getStatementHandleForQuery($query, $exec_args); } else { ($sth, $rv, $r) = $self->_getStatementHandleForQuery($query); } return $sth unless $sth; return $rv; } *doQuery = \&nativeQuery; *native_query = \&nativeQuery; =pod =head2 nativeQueryLoop($query) A loop on nativeQuery, where any placeholders you have put in your query are bound each time you call next(). E.g., my $loop = $db->nativeQueryLoop("UPDATE my_table SET value=? WHERE id=?"); $loop->next([ 'one', 1]); $loop->next([ 'two', 2]); =cut sub nativeQueryLoop { my ($self, $query) = @_; $self->_printDebug($query); return DBIx::Wrapper::StatementLoop->new($self, $query); } *native_query_loop = \&nativeQueryLoop; =pod =head2 newCommand($cmd) This method is deprecated. Use $db->command($cmd_str) instead. This creates a literal SQL command for use in insert(), update(), and related methods, since if you simply put something like "CUR_DATE()" as a value in the %data parameter passed to insert, the function will get quoted, and so will not work as expected. Instead, do something like this: my $data = { file => 'my_document.txt', the_date => $db->newCommand('CUR_DATE()') }; $db->insert('my_doc_table', $data); This can also be done by passing a reference to a string with the SQL command, e.g., my $data = { file => 'my_document.txt', the_date => \'CUR_DATE()' }; $db->insert('my_doc_table', $data); =cut sub newCommand { my ($self, $contents) = @_; return DBIx::Wrapper::SQLCommand->new($contents); } *new_command = \&newCommand; =pod =head2 command($cmd_string), literal($cmd_string), sql_literal($cmd_string) This creates a literal SQL command for use in insert(), update(), and related methods, since if you simply put something like "CUR_DATE()" as a value in the %data parameter passed to insert, the function will get quoted, and so will not work as expected. Instead, do something like this: my $data = { file => 'my_document.txt', the_date => $db->command('CUR_DATE()') }; $db->insert('my_doc_table', $data); This can also be done by passing a reference to a string with the SQL command, e.g., my $data = { file => 'my_document.txt', the_date => \'CUR_DATE()' }; $db->insert('my_doc_table', $data); This is currently how command() is implemented. =cut sub command { my ($self, $str) = @_; return \$str; } *sql_literal = \&command; *literal = \&command; =pod =head2 debugOn(\*FILE_HANDLE) Turns on debugging output. Debugging information will be printed to the given filehandle. =cut # expects a reference to a filehandle to print debug info to sub debugOn { my $self = shift; my $fh = shift; $self->_set_i_val('_debug', 1); $self->_set_i_val('_debug_fh', $fh); return 1; } *debug_on = \&debugOn; =pod =head2 debugOff() Turns off debugging output. =cut sub debugOff { my $self = shift; $self->_delete_i_val('_debug'); $self->_delete_i_val('_debug_fh'); return 1; } *debug_off = \&debugOff; sub _isDebugOn { my ($self) = @_; if (($self->_get_i_val('_debug') and $self->_get_i_val('_debug_fh')) or $ENV{'DBIX_WRAPPER_DEBUG'}) { return 1; } return undef; } sub _printDbiError { my ($self, $extra) = @_; my $handler = $self->_getErrorHandler; $handler = [ $self, \&_default_error_handler ] unless $handler; if ($handler) { if (UNIVERSAL::isa($handler, 'ARRAY')) { my ($obj, $meth) = @$handler; return $obj->$meth($self, $extra); } else { return $handler->($self, $extra); } } return undef; } sub _default_error_handler { my ($self, $db, $extra) = @_; my $dbi_obj = $self->getDBI; return undef unless $dbi_obj->{PrintError}; return undef unless ($self->getDebugLevel | 2); my $fh = $self->_get_i_val('_debug_fh'); $fh = \*STDERR unless $fh; my $time = $self->_getCurDateTime; my ($package, $filename, $line, $subroutine, $hasargs, $wantarray, $evaltext, $is_require, $hints, $bitmask); my $frame = 1; my $this_pkg = __PACKAGE__; ($package, $filename, $line, $subroutine, $hasargs, $wantarray, $evaltext, $is_require, $hints, $bitmask) = caller($frame); while ($package eq $this_pkg) { $frame++; ($package, $filename, $line, $subroutine, $hasargs, $wantarray, $evaltext, $is_require, $hints, $bitmask) = caller($frame); # if we get more than 10 something must be wrong last if $frame >= 10; } local($Carp::CarpLevel) = $frame; my $str = Carp::longmess($DBI::errstr); $str .= $extra if defined($extra); my @one_more = caller($frame + 1); $subroutine = $one_more[3]; $subroutine = '' unless defined($subroutine); $subroutine .= '()' if $subroutine ne ''; print $fh '*' x 60, "\n", "$time:$filename:$line:$subroutine\n", $str, "\n"; } sub _default_debug_handler { my ($self, $db, $str, $fh) = @_; my $time = $self->_getCurDateTime; my ($package, $filename, $line, $subroutine, $hasargs, $wantarray, $evaltext, $is_require, $hints, $bitmask); my $frame = 1; my $this_pkg = __PACKAGE__; ($package, $filename, $line, $subroutine, $hasargs, $wantarray, $evaltext, $is_require, $hints, $bitmask) = caller($frame); while ($package eq $this_pkg) { $frame++; ($package, $filename, $line, $subroutine, $hasargs, $wantarray, $evaltext, $is_require, $hints, $bitmask) = caller($frame); # if we get more than 10 something must be wrong last if $frame >= 10; } my @one_more = caller($frame + 1); $subroutine = $one_more[3]; $subroutine = '' unless defined($subroutine); $subroutine .= '()' if $subroutine ne ''; print $fh '*' x 60, "\n", "$time:$filename:$line:$subroutine\n", $str, "\n"; } sub _printDebug { my ($self, $str) = @_; unless ($self->_isDebugOn) { return undef; } # FIXME: check perl version to see if should use \*STDERR or *STDERR my $fh = $self->_get_i_val('_debug_fh'); $fh = \*STDERR unless $fh; my $handler = $self->_getDebugHandler; $handler = [ $self, \&_default_debug_handler ] unless $handler; if ($handler) { if (UNIVERSAL::isa($handler, 'ARRAY')) { my ($obj, $meth) = @$handler; return $obj->$meth($self, $str, $fh); } else { return $handler->($self, $str, $fh); } } return undef; } sub _getCurDateTime { my ($self) = @_; my $time = time(); my ($sec,$min,$hour,$mday,$mon,$year,$wday,$yday,$isdst) = localtime($time); $mon += 1; $year += 1900; my $date = sprintf "%04d-%02d-%02d %02d:%02d:%02d", $year, $mon, $mday, $hour, $min, $sec; return $date; } sub escapeString { my ($self, $value) = @_; $value = "" unless defined($value); $value =~ s|\\|\\\\|g; $value =~ s|\'|''|g; $value =~ s|\?|\\\?|g; $value =~ s|\000|\\0|g; $value =~ s|\"|""|g; $value =~ s|\n|\\n|g; $value =~ s|\r|\\r|g; $value =~ s|\t|\\t|g; return $value; } *escape_string = \&escapeString; sub _moduleHasSub { my ($self, $module, $sub_name) = @_; } sub DESTROY { my ($self) = @_; return undef unless $self->_getDisconnect; my $dbh = $self->_getDatabaseHandle; $dbh->disconnect if $dbh; delete $i_data{ refaddr($self) }; # free up private data } ################# # getters/setters sub getNameArg { my ($self) = @_; my $arg = $self->_get_i_val('_name_arg'); $arg = 'NAME_lc' unless defined($arg) and $arg ne ''; return $arg; } =pod =head2 setNameArg($arg) This is the argument to pass to the fetchrow_hashref() call on the underlying DBI object. By default, this is 'NAME_lc', so that all field names returned are all lowercase to provide for portable code. If you want to make all the field names return be uppercase, call $db->setNameArg('NAME_uc') after the connect() call. And if you really want the case of the field names to be what the underlying database driveer returns them as, call $db->setNameArg('NAME'). =cut sub setNameArg { my $self = shift; $self->_set_i_val('_name_arg', shift()); } sub setErr { my ($self, $num, $str) = @_; $self->_set_i_val('_err_num', $num); $self->_set_i_val('_err_str', $str); return undef; } sub getErrorString { my $self = shift; return $self->_get_i_val('_err_str'); } sub getErrorNum { my $self = shift; return $self->_get_i_val('_err_num'); } =pod =head2 err() Calls err() on the underlying DBI object, which returns the native database engine error code from the last driver method called. =cut sub err { my ($self) = @_; my $dbh = $self->_getDatabaseHandle; return $dbh->err if $dbh; return 0; } =pod =head2 errstr() Calls errstr() on the underlying DBI object, which returns the native database engine error message from the last driver method called. =cut sub errstr { my $self = shift; my $dbh = $self->_getDatabaseHandle; return $dbh ? $dbh->errstr : undef; } sub _getAttr { my $self = shift; return $self->_get_i_val('_attr'); } sub _setAttr { my $self = shift; $self->_set_i_val('_attr', shift()); } sub _getAuth { my $self = shift; return $self->_get_i_val('_auth'); } sub _setAuth { my $self = shift; $self->_set_i_val('_auth', shift()); } sub _getUsername { my ($self) = @_; return $self->_get_i_val('_username'); } sub _setUsername { my $self = shift; my $username = shift; $self->_set_i_val('_username', $username); } sub _getDatabaseHandle { my $self = shift; return $self->_get_i_val('_dbh'); } sub _setDatabaseHandle { my $self = shift; my $dbh = shift; $self->_set_i_val('_dbh', $dbh); } sub _deleteDatabaseHandle { my $self = shift; my $data = $self->_get_i_data(); delete $data->{_dbh}; } sub getDataSourceAsString { return shift()->_getDataSourceStr; } sub _getDataSourceStr { my $self = shift; return $self->_get_i_val('_data_source_str'); } sub _setDataSourceStr { my $self = shift; $self->_set_i_val('_data_source_str', shift()); } sub _getDataSource { my $self = shift; return $self->_get_i_val('_data_source'); } sub _setDataSource { my $self = shift; $self->_set_i_val('_data_source', shift()); } sub _getDisconnect { my $self = shift; return $self->_get_i_val('_should_disconnect'); } sub _setErrorHandler { my $self = shift; $self->_set_i_val('_error_handler', shift()); } sub _getErrorHandler { return shift()->_get_i_val('_error_handler'); } sub _setDebugHandler { my $self = shift; $self->_set_i_val('_debug_handler', shift()); } sub _getDebugHandler { return shift()->_get_i_val('_debug_handler'); } sub _setDbStyle { my $self = shift; $self->_set_i_val('_db_style', shift()); } sub _getDbStyle { return shift()->_get_i_val('_db_style'); } sub _setDbdDriver { my $self = shift; $self->_set_i_val('_dbd_driver', shift()); } sub _getDbdDriver { return shift()->_get_i_val('_dbd_driver'); } # whether or not to disconnect when the Wrapper object is # DESTROYed sub _setDisconnect { my ($self, $val) = @_; $self->_set_i_val('_should_disconnect', 1); } sub _setHeavy { my $self = shift; $self->_set_i_val('_heavy', shift()); } sub _getHeavy { my $self = shift; return $self->_get_i_val('_heavy'); } sub shouldBeHeavy { my $self = shift; return 1 if $Heavy or $self->_getHeavy; return undef; } sub get_info { my ($self, $name) = @_; require DBI::Const::GetInfoType; my $dbh = $self->_getDatabaseHandle; return $dbh->get_info($DBI::Const::GetInfoType::GetInfoType{$name}); } =pod =head2 DBI methods The following method calls are just passed through to the underlying DBI object for convenience. See the documentation for DBI for details. =over 4 =item prepare This method may call hooks in the future. Use prepare_no_hooks() if you want to ensure that it will be a simple DBI call. =back =cut sub prepare { my $self = shift; my $query = shift; my $dbi_obj = $self->getDBI; my $sth = $dbi_obj->prepare($query); return $sth; } =pod =over 4 =item selectrow_arrayref =back =cut sub selectrow_arrayref { my $self = shift; my $dbh = $self->_getDatabaseHandle; return $dbh->selectrow_arrayref(@_); } =pod =over 4 =item selectrow_hashref =back =cut sub selectrow_hashref { my $self = shift; my $dbh = $self->_getDatabaseHandle; return $dbh->selectrow_hashref(@_); } =pod =over 4 =item selectall_arrayref =back =cut sub selectall_arrayref { my ($self, @args) = @_; my $dbh = $self->_getDatabaseHandle; return $dbh->selectall_arrayref(@args); } =pod =over 4 =item selectall_hashref =back =cut sub selectall_hashref { my ($self, @args) = @_; my $dbh = $self->_getDatabaseHandle; return $dbh->selectall_hashref(@args); } =pod =over 4 =item selectcol_arrayref =back =cut sub selectcol_arrayref { my ($self, @args) = @_; my $dbh = $self->_getDatabaseHandle; return $dbh->selectcol_arrayref(@args); } =pod =over 4 =item do =back =cut sub do { my ($self, @args) = @_; my $dbh = $self->_getDatabaseHandle; return $dbh->do(@args); } =pod =over 4 =item quote =back =cut sub quote { my ($self, @args) = @_; my $dbh = $self->_getDatabaseHandle; return $dbh->quote(@args); } =pod =over 4 =item commit =back =cut sub commit { my ($self) = @_; my $dbh = $self->_getDatabaseHandle; if ($dbh) { return $dbh->commit; } return undef; } =pod =over 4 =item begin_work =back =cut sub begin_work { my $self = shift; my $dbh = $self->_getDatabaseHandle; if ($dbh) { return $dbh->begin_work; } return undef; } =pod =over 4 =item rollback =back =cut sub rollback { my $self = shift; my $dbh = $self->_getDatabaseHandle; if ($dbh) { return $dbh->rollback; } return undef; } =pod =over 4 =item ping =cut sub ping { my ($self) =@_; my $dbh = $self->_getDatabaseHandle; return undef unless $dbh; return $dbh->ping; } # =pod # =head2 getLastInsertId($catalog, $schema, $table, $field, \%attr) # Returns a value identifying the row just inserted, if possible. # If using DBI version 1.38 or later, this method calls # last_insert_id() on the underlying DBI object. Otherwise, does a # "SELECT LAST_INSERT_ID()", which is MySQL specific. The # parameters passed to this method are driver-specific. See the # documentation on DBI for details. # get_last_insert_id() and last_insert_id() are aliases for this # method. # =cut # bah, DBI's last_insert_id is not working for me, so for # now this will be MySQL only =pod =head2 getLastInsertId(), get_last_insert_id(), last_insert_id() Returns the last_insert_id. The default is to be MySQL specific. It just runs the query "SELECT LAST_INSERT_ID()". However, it will also work with MSSQL with the right parameters (see the db_style parameter in the section explaining the connect() method). =cut sub getLastInsertId { my ($self, $catalog, $schema, $table, $field, $attr) = @_; if (0 and DBI->VERSION >= 1.38) { my $dbh = $self->_getDatabaseHandle; return $dbh->last_insert_id($catalog, $schema, $table, $field, $attr); } else { my $query; my $db_style = $self->_getDbStyle; my $dbd_driver = $self->_getDbdDriver; if (defined($db_style) and $db_style ne '') { $db_style = lc($db_style); if ($db_style eq 'mssql' or $db_style eq 'sybase' or $db_style eq 'asa' or $db_style eq 'asany') { $query = q{select @@IDENTITY}; } elsif ($db_style eq 'mysql') { $query = qq{SELECT LAST_INSERT_ID()}; } else { $query = qq{SELECT LAST_INSERT_ID()}; } } elsif (defined($dbd_driver) and $dbd_driver ne '') { if ($dbd_driver eq 'sybase' or $dbd_driver eq 'asany') { $query = q{SELECT @@IDENTITY}; } else { $query = qq{SELECT LAST_INSERT_ID()}; } } else { $query = qq{SELECT LAST_INSERT_ID()}; } my $row = $self->nativeSelectWithArrayRef($query); if ($row and @$row) { return $$row[0]; } return undef; } } *get_last_insert_id = \&getLastInsertId; *last_insert_id = \&getLastInsertId; sub debug_dump { my $self = shift; my $var = shift; my $data = $self->_get_i_data; require Data::Dumper; if (defined($var)) { return Data::Dumper->Dump([ $data ], [ $var ]); } else { return Data::Dumper::Dumper($data); } } *debugDump = \&debug_dump; =pod =head2 Hooks DBIx::Wrapper supports hooks that get called just before and just after various query operations. The add*Hook methods take a single argument that is either a code reference (e.g., anonymous subroutine reference), or an array whose first element is an object and whose second element is the name of a method to call on that object. The hooks will be called with a request object as the first argument. See DBIx::Wrapper::Request. The two expected return values are $request->OK and $request->DECLINED. The first tells DBIx::Wrapper that the current hook has done everything that needs to be done and doesn't call any other hooks in the stack for the current request. DECLINED tells DBIx::Wrapper to continue down the hook stack as if the current handler was never invoked. See DBIx::Wrapper::Request for example hooks. =cut =pod =head3 addPrePrepareHook($hook) Specifies a hook to be called just before any SQL statement is prepare()'d. =cut sub addPrePrepareHook { my $self = shift; my $handler = shift; push @{ $self->_get_i_val('_pre_prepare_hooks') }, [ $handler ]; } *add_pre_prepare_handler = \&addPrePrepareHook; *addPrePrepareHandler = \&addPrePrepareHook; *add_pre_prepare_hook = \&addPrePrepareHook; =pod =head3 addPostPrepareHook($hook) Specifies a hook to be called just after any SQL statement is prepare()'d. =cut sub addPostPrepareHook { my $self = shift; my $handler = shift; push @{ $self->_get_i_val('_post_prepare_hooks') }, [ $handler ]; } *add_post_prepare_hook = \&addPostPrepareHook; =pod =head3 addPreExecHook($hook) Specifies a hook to be called just before any SQL statement is execute()'d. =cut sub addPreExecHook { my $self = shift; my $handler = shift; push @{ $self->_get_i_val('_pre_exec_hooks') }, [ $handler ]; } *add_pre_exec_hook = \&addPreExecHook; =pod =head3 addPostExecHook($hook) Adds a hook to be called just after a statement is execute()'d. =cut sub addPostExecHook { my $self = shift; my $handler = shift; push @{ $self->_get_i_val('_post_exec_hooks') }, [ $handler ]; } *add_post_exec_handler = \&addPostExecHook; *addPostExecHandler = \&addPostExecHook; *add_post_exec_hook = \&addPostExecHook; =pod =head3 addPreFetchHook($hook) Adds a hook to be called just before data is fetch()'d from the server. =cut sub addPreFetchHook { my $self = shift; my $handler = shift; push @{ $self->_get_i_val('_pre_fetch_hooks') }, [ $handler ]; } *add_pre_fetch_hook = \&addPreFetchHook; *addPreFetchHandler = \&addPreFetchHook; =pod =head3 addPostFetchHook($hook) Adds a hook to be called just after data is fetch()'d from the server. =cut sub addPostFetchHook { my $self = shift; my $handler = shift; push @{ $self->_get_i_val('_post_fetch_hooks') }, [ $handler ]; } *addPostFetchHandler = \&addPostFetchHook; sub _do_benchmark { my $self = shift; require Benchmark; my $data = { _dbh => 'dummy' }; my $results = Benchmark::cmpthese(1000000, { 'Plain hash' => sub { my $val = $data->{_dbh} }, 'Indirect hash' => sub { my $val = $i_data{ refaddr($self) }{_dbh} }, } ); # print Data::Dumper->Dump([ $results ], [ 'results' ]) . "\n"; } sub AUTOLOAD { my $self = shift; (my $func = $AUTOLOAD) =~ s/^.*::([^:]+)$/$1/; no strict 'refs'; if (ref($self)) { my $dbh = $self->_getDatabaseHandle; return $dbh->$func(@_); } else { return DBI->$func(@_); } } =pod =head2 There are also underscore_separated versions of these methods. E.g., nativeSelectLoop() becomes native_select_loop() =head1 DEPENDENCIES DBI =head1 ACKNOWLEDGEMENTS Others who have contributed ideas and/or code for this module: Kevin Wilson Mark Stosberg David Bushong =head1 AUTHOR Don Owens =head1 LICENSE AND COPYRIGHT Copyright (c) 2003-2005 Don Owens (don@owensnet.com). All rights reserved. This free software; you can redistribute it and/or modify it under the same terms as Perl itself. See perlartistic. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. =head1 SEE ALSO DBI =head1 VERSION 0.20pre1 =cut 1; blib/lib/auto000755 001751 000000 00000000000 10326255706 013345 5ustar00donwheel000000 000000 blib/lib/auto/DBIx000755 001751 000000 00000000000 10326255706 014133 5ustar00donwheel000000 000000 blib/lib/auto/DBIx/Wrapper000755 001751 000000 00000000000 10326255706 015553 5ustar00donwheel000000 000000 blib/lib/auto/DBIx/Wrapper/.exists000644 001751 000000 00000000000 10224067271 017133 0ustar00donwheel000000 000000 blib/arch000755 001751 000000 00000000000 10326255704 012542 5ustar00donwheel000000 000000 blib/arch/auto000755 001751 000000 00000000000 10326255704 013512 5ustar00donwheel000000 000000 blib/arch/auto/DBIx000755 001751 000000 00000000000 10326255704 014300 5ustar00donwheel000000 000000 blib/arch/auto/DBIx/Wrapper000755 001751 000000 00000000000 10326255705 015721 5ustar00donwheel000000 000000 blib/arch/auto/DBIx/Wrapper/.exists000644 001751 000000 00000000000 10224067271 017302 0ustar00donwheel000000 000000