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 78e8a9f..aaa2c92 100644 (file)
--- a/lib/DBIx/Class/Schema/Loader/DBI/Pg.pm
+++ b/lib/DBIx/Class/Schema/Loader/DBI/Pg.pm
@@ -329,6 +329,20 @@ EOF
             }
         }
 
+        # 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
         if ($data_type =~ /^bool/i && exists $info->{default_value}) {
             if ($info->{default_value} eq '0') {