X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FStorage%2FDBI.pm;h=eac1dda76a6cc95f21b3f1be0f7cdc81dd21d6ed;hb=6a9765c19a33187ad323e0eb685d38c08a97deaf;hp=5f63e28977dc41e1a69a22fcf4b876819f6ffabe;hpb=34fbf45fab383d89143e3b4b54795c2c3ffc190a;p=dbsrgits%2FDBIx-Class.git diff --git a/lib/DBIx/Class/Storage/DBI.pm b/lib/DBIx/Class/Storage/DBI.pm index 5f63e28..eac1dda 100644 --- a/lib/DBIx/Class/Storage/DBI.pm +++ b/lib/DBIx/Class/Storage/DBI.pm @@ -14,6 +14,11 @@ use DBIx::Class::Storage::Statistics; use Scalar::Util(); use List::Util(); +# what version of sqlt do we require if deploy() without a ddl_dir is invoked +# when changing also adjust the corresponding author_require in Makefile.PL +my $minimum_sqlt_version = '0.11002'; + + __PACKAGE__->mk_group_accessors('simple' => qw/_connect_info _dbi_connect_info _dbh _sql_maker _sql_maker_opts _conn_pid _conn_tid transaction_depth _dbh_autocommit _driver_determined savepoints/ @@ -1002,6 +1007,8 @@ sub _connect { $weak_self->throw_exception("DBI Exception: $_[0]"); } else { + # the handler may be invoked by something totally out of + # the scope of DBIC croak ("DBI Exception: $_[0]"); } }; @@ -1337,15 +1344,56 @@ sub insert_bulk { } my %colvalues; - my $table = $source->from; @colvalues{@$cols} = (0..$#$cols); -# XXX some bulk APIs require column list in database order - my ($sql, @bind) = $self->sql_maker->insert($table, \%colvalues); + + # pass scalarref to SQLA for literal sql if it's the same in all slices + for my $i (0..$#$cols) { + my $first_val = $data->[0][$i]; + next unless (Scalar::Util::reftype($first_val)||'') eq 'SCALAR'; + + $colvalues{ $cols->[$i] } = $first_val + if (grep { + (Scalar::Util::reftype($_)||'') eq 'SCALAR' && + $$_ eq $$first_val + } map $data->[$_][$i], (1..$#$data)) == (@$data - 1); + } + + my ($sql, $bind) = $self->_prep_for_execute ( + 'insert', undef, $source, [\%colvalues] + ); + my @bind = @$bind; + + my $empty_bind = 1 if (not @bind) && + (grep { (Scalar::Util::reftype($_)||'') eq 'SCALAR' } values %colvalues) + == @$cols; + + if ((not @bind) && (not $empty_bind)) { + $self->throw_exception( + 'Cannot insert_bulk without support for placeholders' + ); + } $self->_query_start( $sql, @bind ); my $sth = $self->sth($sql, 'insert', $sth_attr); -# @bind = map { ref $_ ? ''.$_ : $_ } @bind; # stringify args + my $rv = do { + if ($empty_bind) { + # bind_param_array doesn't work if there are no binds + $self->_execute_array_empty( $sth, scalar @$data ); + } + else { +# @bind = map { ref $_ ? ''.$_ : $_ } @bind; # stringify args + $self->_execute_array( $source, $sth, \@bind, $cols, $data ); + } + }; + + $self->_query_end( $sql, @bind ); + + return (wantarray ? ($rv, $sth, @bind) : $rv); +} + +sub _execute_array { + my ($self, $source, $sth, $bind, $cols, $data, $guard) = @_; ## This must be an arrayref, else nothing works! my $tuple_status = []; @@ -1356,7 +1404,7 @@ sub insert_bulk { ## Bind the values and execute my $placeholder_index = 1; - foreach my $bound (@bind) { + foreach my $bound (@$bind) { my $attributes = {}; my ($column_name, $data_index) = @$bound; @@ -1371,7 +1419,14 @@ sub insert_bulk { $sth->bind_param_array( $placeholder_index, [@data], $attributes ); $placeholder_index++; } + my $rv = eval { $sth->execute_array({ArrayTupleStatus => $tuple_status}) }; + +# only needed for Sybase, it requires a commit before the $sth->finish + $guard->commit if $guard; + + $sth->finish; + if (my $err = $@ || $sth->errstr) { my $i = 0; ++$i while $i <= $#$tuple_status && !ref $tuple_status->[$i]; @@ -1379,24 +1434,35 @@ sub insert_bulk { $self->throw_exception("Unexpected populate error: $err") if ($i > $#$tuple_status); - require Data::Dumper; - local $Data::Dumper::Terse = 1; - local $Data::Dumper::Indent = 1; - local $Data::Dumper::Useqq = 1; - local $Data::Dumper::Quotekeys = 0; - local $Data::Dumper::Sortkeys = 1; - $self->throw_exception(sprintf "%s for populate slice:\n%s", ($tuple_status->[$i][1] || $err), - Data::Dumper::Dumper( - { map { $cols->[$_] => $data->[$i][$_] } (0 .. $#$cols) } - ), + $self->_pretty_print ({ + map { $cols->[$_] => $data->[$i][$_] } (0 .. $#$cols) + }), ); } + return $rv; +} + +sub _execute_array_empty { + my ($self, $sth, $count) = @_; + eval { + my $dbh = $self->_get_dbh; + local $dbh->{RaiseError} = 1; + local $dbh->{PrintError} = 0; + foreach (1..$count) { + $sth->execute; +# In case of a multi-statement with a select, some DBDs (namely Sybase) require +# the cursor to be exhausted. + $sth->fetchall_arrayref; + } + }; + my $exception = $@; $sth->finish; - $self->_query_end( $sql, @bind ); - return (wantarray ? ($rv, $sth, @bind) : $rv); + $self->throw_exception($exception) if $exception; + + return $count; } sub update { @@ -1987,15 +2053,23 @@ sub _subq_count_select { return @pcols ? \@pcols : [ 1 ]; } -# -# Returns an ordered list of column names before they are used -# in a SELECT statement. By default simply returns the list -# passed in. -# -# This may be overridden in a specific storage when there are -# requirements such as moving BLOB columns to the end of the -# SELECT list. -sub _order_select_columns { +=head2 order_select_columns + +=over 4 + +=item Arguments: $source, \@columns + +=back + +Returns an ordered list of column names to be used in a SELECT statement. By +default simply returns the list that was passed in. + +This may be overridden in a specific storage when there are requirements such as +moving BLOB columns to the end of the SELECT list. + +=cut + +sub order_select_columns { #my ($self, $source, $columns) = @_; return @{$_[2]}; } @@ -2620,6 +2694,33 @@ sub lag_behind_master { return; } +# SQLT version handling +{ + my $_sqlt_version_ok; # private + my $_sqlt_version_error; # private + + sub _sqlt_version_ok { + if (!defined $_sqlt_version_ok) { + eval "use SQL::Translator $minimum_sqlt_version"; + if ($@) { + $_sqlt_version_ok = 0; + $_sqlt_version_error = $@; + } + else { + $_sqlt_version_ok = 1; + } + } + return $_sqlt_version_ok; + } + + sub _sqlt_version_error { + shift->_sqlt_version_ok unless defined $_sqlt_version_ok; + return $_sqlt_version_error; + } + + sub _sqlt_minimum_version { $minimum_sqlt_version }; +} + sub DESTROY { my $self = shift;