Support :no_inherit option for NOT NULL and CHECK constraints on PostgreSQL

Author: jeremyevans

CHANGELOG

@@ -1,5 +1,9 @@
 === master
 
+* Support :no_inherit option for CHECK constraints on PostgreSQL 9.2+ (jeremyevans)
+
+* Support :no_inherit option for NOT NULL constraints on PostgreSQL 18+ (jeremyevans)
+
 * Support not_null: {name: "constaint_name"} column option for naming NOT NULL constraints on PostgreSQL 18+ (jeremyevans)
 
 * Support :period foreign_key option for PERIOD on PostgreSQL 18+ (jeremyevans)

lib/sequel/adapters/shared/postgres.rb

@@ -1388,6 +1388,9 @@ def constraint_definition_sql(constraint)
           sql
         else # when :foreign_key, :check
           sql = super
+          if constraint[:no_inherit]
+            sql << " NO INHERIT"
+          end
           if constraint[:not_enforced]
             sql << " NOT ENFORCED"
           end
@@ -1406,10 +1409,15 @@ def column_definition_add_references_sql(sql, column)
       end
 
       def column_definition_null_sql(sql, column)
-        if column[:not_null].is_a?(Hash) && (name = column[:not_null][:name])
+        constraint = column[:not_null]
+        constraint = nil unless constraint.is_a?(Hash)
+        if constraint && (name = constraint[:name])
           sql << " CONSTRAINT #{quote_identifier(name)}"
         end
         super
+        if constraint && constraint[:no_inherit]
+          sql << " NO INHERIT"
+        end
       end
 
       # Handle :period option

lib/sequel/database/schema_generator.rb

@@ -121,6 +121,8 @@ def check(*args, &block)
       # :not_null :: Similar to setting <tt>null: false</tt>, but you can provide a hash value
       #              to specify options for the NOT NULL constraint on PostgreSQL 18+:
       #              :name :: The name of the NOT NULL constraint
+      #              :no_inherit :: Set NO INHERIT on the constraint, so it will not propogate to
+      #                             child tables.
       # :null :: Mark the column as allowing NULL values (if true),
       #          or not allowing NULL values (if false).  The default is to allow NULL values.
       # :on_delete :: Specify the behavior of this column when being deleted
@@ -191,6 +193,8 @@ def column(name, type, opts = OPTS)
       # :deferrable :: Whether the CHECK constraint should be marked DEFERRABLE.
       #
       # PostgreSQL specific options:
+      # :no_inherit :: Set NO INHERIT on the constraint, so it will not propogate to
+      #                child tables.
       # :not_enforced :: Whether the CHECK constraint should be marked NOT ENFORCED.
       # :not_valid :: Whether the CHECK constraint should be marked NOT VALID.
       def constraint(name, *args, &block)

spec/adapters/postgres_spec.rb

@@ -1324,6 +1324,30 @@ def before_create
     @db[:atest].select_map(:a).must_equal [nil]
   end if DB.server_version >= 180000
 
+  it 'supports :not_null column option with :no_inherit option' do
+    begin
+      @db.create_table(:atest){Integer :a, :not_null => {:no_inherit => true}}
+      proc{@db[:atest].insert}.must_raise Sequel::NotNullConstraintViolation
+      @db.create_table(:atest2, :inherits=>:atest)
+      @db[:atest2].insert
+      @db[:atest2].select_map(:a).must_equal [nil]
+    ensure
+      @db.drop_table?(:atest2)
+    end
+  end if DB.server_version >= 180000
+
+  it 'supports check constraint :no_inherit option' do
+    begin
+      @db.create_table(:atest){Integer :a; constraint(:no_inherit => true){a > 10}}
+      proc{@db[:atest].insert(9)}.must_raise Sequel::CheckConstraintViolation
+      @db.create_table(:atest2, :inherits=>:atest)
+      @db[:atest2].insert(9)
+      @db[:atest2].select_map(:a).must_equal [9]
+    ensure
+      @db.drop_table?(:atest2)
+    end
+  end if DB.server_version >= 90200
+
   it 'supports using PERIOD in composite foreign key constraints' do
     @db.create_table!(:atest) do
       int4range :id