Added has_column and column_info methods

[dbsrgits/DBIx-Class.git] / lib / DBIx / Class / PK / Auto.pm

diff --git a/lib/DBIx/Class/PK/Auto.pm b/lib/DBIx/Class/PK/Auto.pm
index f4ee81c..c2027cc 100644 (file)
--- a/lib/DBIx/Class/PK/Auto.pm
+++ b/lib/DBIx/Class/PK/Auto.pm
@@ -1,5 +1,6 @@
 package DBIx::Class::PK::Auto;
 
+use base qw/Class::Data::Inheritable/;
 use strict;
 use warnings;
 
@@ -14,6 +15,9 @@ DBIx::Class::PK::Auto - Automatic Primary Key class
 This class overrides the insert method to get automatically
 incremented primary keys.
 
+You don't want to be using this directly - instead load the appropriate
+one for your database, e.g. PK::Auto::SQLite
+
 =head1 METHODS
 
 =over 4
@@ -28,10 +32,16 @@ primary keys.
 sub insert {
   my ($self, @rest) = @_;
   my $ret = $self->NEXT::ACTUAL::insert(@rest);
+
+  # if all primaries are already populated, skip auto-inc
+  my $populated = 0;
+  map { $populated++ if $self->$_ } $self->primary_columns;
+  return $ret if ( $populated == scalar $self->primary_columns );
+
   my ($pri, $too_many) =
-    (grep { $self->_primaries->{$_}{'auto_increment'} }
-       keys %{ $self->_primaries })
-    || (keys %{ $self->_primaries });
+    (grep { $self->column_info($_)->{'auto_increment'} }
+       $self->primary_columns)
+    || $self->primary_columns;
   $self->throw( "More than one possible key found for auto-inc on ".ref $self )
     if $too_many;
   unless (defined $self->get_column($pri)) {
@@ -44,13 +54,22 @@ sub insert {
   return $ret;
 }
 
+=item sequence
+
+Manually define the correct sequence for your table, to avoid the overhead
+associated with looking up the sequence automatically.
+
+=cut
+
+__PACKAGE__->mk_classdata('sequence');
+
 1;
 
 =back
 
 =head1 AUTHORS
 
-Matt S. Trout <perl-stuff@trout.me.uk>
+Matt S. Trout <mst@shadowcatsystems.co.uk>
 
 =head1 LICENSE