X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FStorage%2FDBI.pm;h=1ce71fc0187019445ea77d6448b49cdd398cf67d;hb=2baff5dae873d8de453383486f304721dd26653c;hp=4990e78c4a99cbb16c21ffecf1ec7f6d9869aae7;hpb=84f97107323065f3be493552cd70ae9d976a4f80;p=dbsrgits%2FDBIx-Class.git diff --git a/lib/DBIx/Class/Storage/DBI.pm b/lib/DBIx/Class/Storage/DBI.pm index 4990e78..1ce71fc 100644 --- a/lib/DBIx/Class/Storage/DBI.pm +++ b/lib/DBIx/Class/Storage/DBI.pm @@ -1328,7 +1328,7 @@ sub insert { ## scalar refs, or at least, all the same type as the first set, the statement is ## only prepped once. sub insert_bulk { - my ($self, $source, $cols, $data) = @_; + my ($self, $source, $cols, $data, $sth_attr) = @_; # redispatch to insert_bulk method of storage we reblessed into, if necessary if (not $self->_driver_determined) { @@ -1337,12 +1337,51 @@ sub insert_bulk { } my %colvalues; - my $table = $source->from; @colvalues{@$cols} = (0..$#$cols); - my ($sql, @bind) = $self->sql_maker->insert($table, \%colvalues); + + # bind 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)) { + croak 'Cannot insert_bulk without support for placeholders'; + } $self->_query_start( $sql, @bind ); - my $sth = $self->sth($sql); + my $sth = $self->sth($sql, 'insert', $sth_attr); + + if ($empty_bind) { + # bind_param_array doesn't work if there are no binds + eval { + local $self->_get_dbh->{RaiseError} = 1; + local $self->_get_dbh->{PrintError} = 0; + foreach (0..$#$data) { + $sth->execute; + $sth->fetchall_arrayref; + } + }; + my $exception = $@; + $sth->finish; + $self->throw_exception($exception) if $exception; + return; + } # @bind = map { ref $_ ? ''.$_ : $_ } @bind; # stringify args @@ -1370,12 +1409,14 @@ sub insert_bulk { $sth->bind_param_array( $placeholder_index, [@data], $attributes ); $placeholder_index++; } + my $rv = eval { $sth->execute_array({ArrayTupleStatus => $tuple_status}) }; - if (my $err = $@) { + $sth->finish; + if (my $err = $@ || $sth->errstr) { my $i = 0; ++$i while $i <= $#$tuple_status && !ref $tuple_status->[$i]; - $self->throw_exception($sth->errstr || "Unexpected populate error: $err") + $self->throw_exception("Unexpected populate error: $err") if ($i > $#$tuple_status); require Data::Dumper; @@ -1386,13 +1427,12 @@ sub insert_bulk { local $Data::Dumper::Sortkeys = 1; $self->throw_exception(sprintf "%s for populate slice:\n%s", - $tuple_status->[$i][1], + ($tuple_status->[$i][1] || $err), Data::Dumper::Dumper( { map { $cols->[$_] => $data->[$i][$_] } (0 .. $#$cols) } ), ); } - $self->throw_exception($sth->errstr) if !$rv; $self->_query_end( $sql, @bind ); return (wantarray ? ($rv, $sth, @bind) : $rv); @@ -2057,12 +2097,15 @@ Returns a L sth (statement handle) for the supplied SQL. =cut sub _dbh_sth { - my ($self, $dbh, $sql) = @_; + my ($self, $dbh, $sql, $op, $sth_attr) = @_; +# $op is ignored right now + + $sth_attr ||= {}; # 3 is the if_active parameter which avoids active sth re-use my $sth = $self->disable_sth_caching - ? $dbh->prepare($sql) - : $dbh->prepare_cached($sql, {}, 3); + ? $dbh->prepare($sql, $sth_attr) + : $dbh->prepare_cached($sql, $sth_attr, 3); # XXX You would think RaiseError would make this impossible, # but apparently that's not true :( @@ -2072,8 +2115,8 @@ sub _dbh_sth { } sub sth { - my ($self, $sql) = @_; - $self->dbh_do('_dbh_sth', $sql); # retry over disconnects + my ($self, $sql, $op, $sth_attr) = @_; + $self->dbh_do('_dbh_sth', $sql, $op, $sth_attr); # retry over disconnects } sub _dbh_columns_info_for {