Merge 'trunk' into 'on_connect_call'
Rafael Kitover [Sun, 14 Jun 2009 16:57:41 +0000 (16:57 +0000)]
r5510@hlagh (orig r6661):  ribasushi | 2009-06-13 08:56:54 -0700
Test resultset serialization as well
r5511@hlagh (orig r6662):  ribasushi | 2009-06-13 09:08:14 -0700
local()ize sqla for in the right place
r5512@hlagh (orig r6663):  ribasushi | 2009-06-13 09:10:22 -0700
Do not use raw sources in {from} - proxy via source handles
r5514@hlagh (orig r6665):  ribasushi | 2009-06-13 10:31:55 -0700
really local()ize for as the author intended
r5515@hlagh (orig r6666):  michaelr | 2009-06-13 15:23:15 -0700
Added documentation for from => $rs->as_query

r5516@hlagh (orig r6667):  caelum | 2009-06-13 16:20:02 -0700
fix master debug output for ::Replicated
r5517@hlagh (orig r6668):  ribasushi | 2009-06-14 01:22:28 -0700
Release 0.08107
r5519@hlagh (orig r6670):  ribasushi | 2009-06-14 02:00:35 -0700
Lapse in copy() docs
r5521@hlagh (orig r6672):  ribasushi | 2009-06-14 02:33:22 -0700
Forgotten piece of as_query refactor

1  2 
lib/DBIx/Class/Storage/DBI.pm

@@@ -14,15 -14,13 +14,15 @@@ use List::Util()
  
  __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 savepoints/
 +       _conn_pid _conn_tid transaction_depth _dbh_autocommit _on_connect_do
 +       _on_disconnect_do _on_connect_do_store _on_disconnect_do_store
 +       savepoints/
  );
  
  # the values for these accessors are picked out (and deleted) from
  # the attribute hashref passed to connect_info
  my @storage_options = qw/
 -  on_connect_do on_disconnect_do disable_sth_caching unsafe auto_savepoint
 +  on_connect_call on_disconnect_call disable_sth_caching unsafe auto_savepoint
  /;
  __PACKAGE__->mk_group_accessors('simple' => @storage_options);
  
@@@ -179,91 -177,6 +179,91 @@@ immediately before disconnecting from t
  Note, this only runs if you explicitly call L</disconnect> on the
  storage object.
  
 +=item on_connect_call
 +
 +A more generalized form of L</on_connect_do> that calls the specified
 +C<connect_call_METHOD> methods in your storage driver.
 +
 +  on_connect_do => 'select 1'
 +
 +is equivalent to:
 +
 +  on_connect_call => [ [ do_sql => 'select 1' ] ]
 +
 +Its values may contain:
 +
 +=over
 +
 +=item a scalar
 +
 +Will call the C<connect_call_METHOD> method.
 +
 +=item a code reference
 +
 +Will execute C<< $code->($storage) >>
 +
 +=item an array reference
 +
 +Each value can be a method name or code reference.
 +
 +=item an array of arrays
 +
 +For each array, the first item is taken to be the C<connect_call_> method name
 +or code reference, and the rest are parameters to it.
 +
 +=back
 +
 +Some predefined storage methods you may use:
 +
 +=over
 +
 +=item do_sql
 +
 +Executes a SQL string or a code reference that returns a SQL string. This is
 +what L</on_connect_do> and L</on_disconnect_do> use.
 +
 +It can take:
 +
 +=over
 +
 +=item a scalar
 +
 +Will execute the scalar as SQL.
 +
 +=item an arrayref
 +
 +Taken to be arguments to L<DBI/do>, the SQL string optionally followed by the
 +attributes hashref and bind values.
 +
 +=item a code reference
 +
 +Will execute C<< $code->($storage) >> and execute the return array refs as
 +above.
 +
 +=back
 +
 +=item datetime_setup
 +
 +Execute any statements necessary to initialize the database session to return
 +and accept datetime/timestamp values used with
 +L<DBIx::Class::InflateColumn::DateTime>.
 +
 +Only necessary for some databases, see your specific storage driver for
 +implementation details.
 +
 +=back
 +
 +=item on_disconnect_call
 +
 +Takes arguments in the same form as L</on_connect_call> and executes them
 +immediately before disconnecting from the database.
 +
 +Calls the C<disconnect_call_METHOD> methods as opposed to the
 +C<connect_call_METHOD> methods called by L</on_connect_call>.
 +
 +Note, this only runs if you explicitly call L</disconnect> on the
 +storage object.
 +
  =item disable_sth_caching
  
  If set to a true value, this option will disable the caching of
@@@ -434,11 -347,6 +434,11 @@@ sub connect_info 
          $self->_sql_maker_opts->{$sql_maker_opt} = $opt_val;
        }
      }
 +    for my $connect_do_opt (qw/on_connect_do on_disconnect_do/) {
 +      if(my $opt_val = delete $attrs{$connect_do_opt}) {
 +        $self->$connect_do_opt($opt_val);
 +      }
 +    }
    }
  
    %attrs = () if (ref $args[0] eq 'CODE');  # _connect() never looks past $args[0] in this case
  
  This method is deprecated in favour of setting via L</connect_info>.
  
 +=cut
 +
 +sub on_connect_do {
 +  my $self = shift;
 +  $self->_setup_connect_do(on_connect_do => @_);
 +}
 +
 +=head2 on_disconnect_do
 +
 +This method is deprecated in favour of setting via L</connect_info>.
 +
 +=cut
 +
 +sub on_disconnect_do {
 +  my $self = shift;
 +  $self->_setup_connect_do(on_disconnect_do => @_);
 +}
 +
 +sub _setup_connect_do {
 +  my ($self, $opt) = (shift, shift);
 +
 +  my $accessor = "_$opt";
 +  my $store    = "_${opt}_store";
 +
 +  return $self->$accessor if not @_;
 +
 +  my $val = shift;
 +
 +  if (not defined $val) {
 +    $self->$accessor(undef);
 +    $self->$store(undef);
 +    return;
 +  }
 +
 +  my @store;
 +
 +  if (not ref($val)) {
 +    push @store, [ 'do_sql', $val ];
 +  } elsif (ref($val) eq 'CODE') {
 +    push @store, $val;
 +  } elsif (ref($val) eq 'ARRAY') {
 +    push @store, map [ 'do_sql', $_ ], @$val;
 +  } else {
 +    $self->throw_exception("Invalid type for $opt ".ref($val));
 +  }
 +
 +  $self->$store(\@store);
 +  $self->$accessor($val);
 +}
  
  =head2 dbh_do
  
@@@ -647,12 -506,8 +647,12 @@@ sub disconnect 
    my ($self) = @_;
  
    if( $self->connected ) {
 -    my $connection_do = $self->on_disconnect_do;
 -    $self->_do_connection_actions($connection_do) if ref($connection_do);
 +    if (my $connection_call = $self->on_disconnect_call) {
 +      $self->_do_connection_actions(disconnect_call_ => $connection_call)
 +    }
 +    if (my $connection_do   = $self->_on_disconnect_do_store) {
 +      $self->_do_connection_actions(disconnect_call_ => $connection_do)
 +    }
  
      $self->_dbh->rollback unless $self->_dbh_autocommit;
      $self->_dbh->disconnect;
@@@ -769,12 -624,8 +769,12 @@@ sub _populate_dbh 
    #  there is no transaction in progress by definition
    $self->{transaction_depth} = $self->_dbh_autocommit ? 0 : 1;
  
 -  my $connection_do = $self->on_connect_do;
 -  $self->_do_connection_actions($connection_do) if $connection_do;
 +  if (my $connection_call = $self->on_connect_call) {
 +    $self->_do_connection_actions(connect_call_ => $connection_call)
 +  }
 +  if (my $connection_do = $self->_on_connect_do_store) {
 +    $self->_do_connection_actions(connect_call_ => $connection_do)
 +  }
  }
  
  sub _determine_driver {
  }
  
  sub _do_connection_actions {
 -  my $self = shift;
 -  my $connection_do = shift;
 -
 -  if (!ref $connection_do) {
 -    $self->_do_query($connection_do);
 -  }
 -  elsif (ref $connection_do eq 'ARRAY') {
 -    $self->_do_query($_) foreach @$connection_do;
 -  }
 -  elsif (ref $connection_do eq 'CODE') {
 -    $connection_do->($self);
 -  }
 -  else {
 -    $self->throw_exception (sprintf ("Don't know how to process conection actions of type '%s'", ref $connection_do) );
 +  my $self          = shift;
 +  my $method_prefix = shift;
 +  my $call          = shift;
 +
 +  if (not ref($call)) {
 +    my $method = $method_prefix . $call;
 +    $self->$method(@_);
 +  } elsif (ref($call) eq 'CODE') {
 +    $self->$call(@_);
 +  } elsif (ref($call) eq 'ARRAY') {
 +    if (ref($call->[0]) ne 'ARRAY') {
 +      $self->_do_connection_actions($method_prefix, $_) for @$call;
 +    } else {
 +      $self->_do_connection_actions($method_prefix, @$_) for @$call;
 +    }
 +  } else {
 +    $self->throw_exception (sprintf ("Don't know how to process conection actions of type '%s'", ref($call)) );
    }
  
    return $self;
  }
  
 +sub connect_call_do_sql {
 +  my $self = shift;
 +  $self->_do_query(@_);
 +}
 +
 +sub disconnect_call_do_sql {
 +  my $self = shift;
 +  $self->_do_query(@_);
 +}
 +
 +# override in db-specific backend when necessary
 +sub connect_call_datetime_setup { 1 }
 +
  sub _do_query {
    my ($self, $action) = @_;
  
@@@ -1356,34 -1191,45 +1356,45 @@@ sub _per_row_update_delete 
  
  sub _select {
    my $self = shift;
+   # localization is neccessary as
+   # 1) there is no infrastructure to pass this around (easy to do, but will wait)
+   # 2) _select_args sets it and _prep_for_execute consumes it
    my $sql_maker = $self->sql_maker;
    local $sql_maker->{for};
    return $self->_execute($self->_select_args(@_));
  }
  
  sub _select_args_to_query {
    my $self = shift;
  
+   # localization is neccessary as
+   # 1) there is no infrastructure to pass this around (easy to do, but will wait)
+   # 2) _select_args sets it and _prep_for_execute consumes it
    my $sql_maker = $self->sql_maker;
    local $sql_maker->{for};
  
-   # my ($op, $bind, $ident, $bind_attrs, $select, $cond, $order, $rows, $offset) 
+   # my ($op, $bind, $ident, $bind_attrs, $select, $cond, $order, $rows, $offset)
    #  = $self->_select_args($ident, $select, $cond, $attrs);
    my ($op, $bind, $ident, $bind_attrs, @args) =
      $self->_select_args(@_);
  
-   # my ($sql, $bind) = $self->_prep_for_execute($op, $bind, $ident, [ $select, $cond, $order, $rows, $offset ]);
+   # my ($sql, $prepared_bind) = $self->_prep_for_execute($op, $bind, $ident, [ $select, $cond, $order, $rows, $offset ]);
    my ($sql, $prepared_bind) = $self->_prep_for_execute($op, $bind, $ident, \@args);
+   $prepared_bind ||= [];
  
-   return \[ "($sql)", @{ $prepared_bind || [] }];
+   return wantarray
+     ? ($sql, $prepared_bind, $bind_attrs)
+     : \[ "($sql)", @$prepared_bind ]
+   ;
  }
  
  sub _select_args {
    my ($self, $ident, $select, $condition, $attrs) = @_;
  
-   my $for = delete $attrs->{for};
    my $sql_maker = $self->sql_maker;
-   $sql_maker->{for} = $for;
+   $sql_maker->{for} = delete $attrs->{for};
  
    my $order = { map
      { $attrs->{$_} ? ( $_ => $attrs->{$_} ) : ()  }
@@@ -1452,8 -1298,8 +1463,8 @@@ sub _resolve_ident_sources 
          $tabinfo = $_->[0];
        }
  
-       $alias2source->{$tabinfo->{-alias}} = $tabinfo->{-result_source}
-         if ($tabinfo->{-result_source});
+       $alias2source->{$tabinfo->{-alias}} = $tabinfo->{-source_handle}->resolve
+         if ($tabinfo->{-source_handle});
      }
    }