Support identity columns in PostgreSQL v10

[dbsrgits/DBIx-Class-Schema-Loader.git] / lib / DBIx / Class / Schema / Loader / DBI / Pg.pm

diff --git a/lib/DBIx/Class/Schema/Loader/DBI/Pg.pm b/lib/DBIx/Class/Schema/Loader/DBI/Pg.pm
index b614919..aaa2c92 100644 (file)
--- a/lib/DBIx/Class/Schema/Loader/DBI/Pg.pm
+++ b/lib/DBIx/Class/Schema/Loader/DBI/Pg.pm
@@ -5,7 +5,7 @@ use warnings;
 use base 'DBIx::Class::Schema::Loader::DBI::Component::QuotedDefault';
 use mro 'c3';
 
-our $VERSION = '0.07047';
+our $VERSION = '0.07048_01';
 
 =head1 NAME
 
@@ -323,6 +323,24 @@ EOF
                 my $now = 'now()';
                 $info->{original}{default_value} = \$now;
             }
+            elsif (${ $info->{default_value} } =~ /\bCURRENT_TIMESTAMP\b/) {
+                # PostgreSQL v10 upcases current_timestamp in default values
+                ${ $info->{default_value} } =~ s/\b(CURRENT_TIMESTAMP)\b/lc $1/ge;
+            }
+        }
+
+        # Identity columns can't have default values
+        if ($self->dbh->{pg_server_version} >= 100000 && !exists $info->{default_value}) {
+            my $identity = $self->schema->storage->dbh->selectrow_array(<<'EOF', {}, $table->sql_name, $col);
+SELECT attidentity
+FROM pg_catalog.pg_attribute
+WHERE attrelid = ?::regclass
+AND attname = ?
+EOF
+            if ($identity) {
+                $info->{is_auto_increment} = 1;
+                $info->{extra}{generated_as_identity} = { a => 'always', d => 'by_default' }->{$identity};
+            }
         }
 
         # detect 0/1 for booleans and rewrite