activerecord-jdbc-adapter 50.8-java → 51.1-java

data/lib/arjdbc/postgresql/connection_methods.rb

@@ -1,7 +1,6 @@
 # frozen_string_literal: true
 ArJdbc::ConnectionMethods.module_eval do
   def postgresql_connection(config)
-    config = config.deep_dup
     # NOTE: this isn't "really" necessary but Rails (in tests) assumes being able to :
     #   ActiveRecord::Base.postgresql_connection ActiveRecord::Base.configurations['arunit'].merge(:insert_returning => false)
     # ... while using symbols by default but than configurations returning string keys ;(
@@ -17,8 +16,7 @@ ArJdbc::ConnectionMethods.module_eval do
       ::Jdbc::Postgres.load_driver(:require) if defined?(::Jdbc::Postgres.load_driver)
     rescue LoadError # assuming driver.jar is on the class-path
     end
-    driver = (config[:driver] ||=
-      defined?(::Jdbc::Postgres.driver_name) ? ::Jdbc::Postgres.driver_name : 'org.postgresql.Driver')
+    driver = config[:driver] ||= 'org.postgresql.Driver'
 
     host = config[:host] ||= ( config[:hostaddr] || ENV['PGHOST'] || 'localhost' )
     port = config[:port] ||= ( ENV['PGPORT'] || 5432 )
@@ -53,15 +51,8 @@ ArJdbc::ConnectionMethods.module_eval do
     properties['tcpKeepAlive'] ||= config[:keepalives] if config.key?(:keepalives)
     properties['kerberosServerName'] ||= config[:krbsrvname] if config[:krbsrvname]
 
-    prepared_statements = config.fetch(:prepared_statements) { true }
-    prepared_statements = false if prepared_statements == 'false'
-    if prepared_statements
-      # this makes the pgjdbc driver handle hot compatibility internally
-      properties['autosave'] ||= 'conservative'
-    else
-      # If prepared statements are off, lets make sure they are really *off*
-      properties['prepareThreshold'] = 0
-    end
+    # If prepared statements are off, lets make sure they are really *off*
+    properties['prepareThreshold'] ||= 0 unless config[:prepared_statements]
 
     jdbc_connection(config)
   end

data/lib/arjdbc/postgresql/oid_types.rb

@@ -9,61 +9,6 @@ module ArJdbc
     # @private
     OID = ::ActiveRecord::ConnectionAdapters::PostgreSQL::OID
 
-    # this version makes sure to register the types by name as well
-    # we still need to version with OID since it's used from SchemaStatements as well
-    class ArjdbcTypeMapInitializer < OID::TypeMapInitializer
-      private
-
-      def name_with_ns(row)
-        if row['in_ns']
-          row['typname']
-        else
-          %Q("#{row['nspname']}"."#{row['typname']}")
-        end
-      end
-
-      def register_enum_type(row)
-        super
-        register name_with_ns(row), OID::Enum.new
-      end
-
-      def register_array_type(row)
-        super
-        register_with_subtype(name_with_ns(row), row['typelem'].to_i) do |subtype|
-          OID::Array.new(subtype, row['typdelim'])
-        end
-      end
-
-      def register_range_type(row)
-        super
-        name = name_with_ns(row)
-        register_with_subtype(name, row['rngsubtype'].to_i) do |subtype|
-          OID::Range.new(subtype, name.to_sym)
-        end
-      end
-
-      def register_domain_type(row)
-        if base_type = @store.lookup(row['typbasetype'].to_i)
-          register row['oid'], base_type
-          register name_with_ns(row), base_type
-        else
-          warn "unknown base type (OID: #{row['typbasetype']}) for domain #{row['typname']}."
-        end
-      end
-
-      def register_composite_type(row)
-        if subtype = @store.lookup(row['typelem'].to_i)
-          register row['oid'], OID::Vector.new(row['typdelim'], subtype)
-          register name_with_ns(row), OID::Vector.new(row['typdelim'], subtype)
-        end
-      end
-
-      def assert_valid_registration(oid, oid_type)
-        ret = super
-        ret == 0 ? oid : ret
-      end
-    end
-
     # @private
     module OIDTypes
 
@@ -96,7 +41,7 @@ module ArJdbc
 
       def get_oid_type(oid, fmod, column_name, sql_type = '') # :nodoc:
         if !type_map.key?(oid)
-          load_additional_types(type_map, oid)
+          load_additional_types(type_map, [oid])
         end
 
         type_map.fetch(oid, fmod, sql_type) {
@@ -124,7 +69,7 @@ module ArJdbc
         register_class_with_limit m, 'int2', Type::Integer
         register_class_with_limit m, 'int4', Type::Integer
         register_class_with_limit m, 'int8', Type::Integer
-        m.alias_type 'oid', 'int2'
+        m.register_type 'oid', OID::Oid.new
         m.register_type 'float4', Type::Float.new
         m.alias_type 'float8', 'float4'
         m.register_type 'text', Type::Text.new
@@ -159,15 +104,10 @@ module ArJdbc
         m.register_type 'polygon', OID::SpecializedString.new(:polygon)
         m.register_type 'circle', OID::SpecializedString.new(:circle)
 
-        #m.alias_type 'interval', 'varchar' # in Rails 5.0
-        # This is how Rails 5.1 handles it.
-        # In 5.0 SpecializedString doesn't take a precision option  5.0 actually leaves it as a regular String
-        # but we need it specialized to support prepared statements
-        # m.register_type 'interval' do |_, _, sql_type|
-        #   precision = extract_precision(sql_type)
-        #   OID::SpecializedString.new(:interval, precision: precision)
-        # end
-        m.register_type 'interval', OID::SpecializedString.new(:interval)
+        m.register_type 'interval' do |_, _, sql_type|
+          precision = extract_precision(sql_type)
+          OID::SpecializedString.new(:interval, precision: precision)
+        end
 
         register_class_with_precision m, 'time', Type::Time
         register_class_with_precision m, 'timestamp', OID::DateTime
@@ -193,45 +133,26 @@ module ArJdbc
         end
 
         load_additional_types(m)
-
-        # pgjdbc returns these if the column is auto-incrmenting
-        m.alias_type 'serial', 'int4'
-        m.alias_type 'bigserial', 'int8'
       end
 
-      def load_additional_types(type_map, oid = nil) # :nodoc:
-        initializer = ArjdbcTypeMapInitializer.new(type_map)
+      def load_additional_types(type_map, oids = nil) # :nodoc:
+        initializer = OID::TypeMapInitializer.new(type_map)
 
         if supports_ranges?
           query = <<-SQL
-              SELECT t.oid, t.typname, t.typelem, t.typdelim, t.typinput, r.rngsubtype, t.typtype, t.typbasetype,
-                ns.nspname, ns.nspname = ANY(current_schemas(true)) in_ns
+              SELECT t.oid, t.typname, t.typelem, t.typdelim, t.typinput, r.rngsubtype, t.typtype, t.typbasetype
               FROM pg_type as t
               LEFT JOIN pg_range as r ON oid = rngtypid
-              JOIN pg_namespace AS ns ON t.typnamespace = ns.oid
           SQL
         else
           query = <<-SQL
-              SELECT t.oid, t.typname, t.typelem, t.typdelim, t.typinput, t.typtype, t.typbasetype,
-                ns.nspname, ns.nspname = ANY(current_schemas(true)) in_ns
+              SELECT t.oid, t.typname, t.typelem, t.typdelim, t.typinput, t.typtype, t.typbasetype
               FROM pg_type as t
-              JOIN pg_namespace AS ns ON t.typnamespace = ns.oid
           SQL
         end
 
-        if oid
-          if oid.is_a? Numeric || oid.match(/^\d+$/)
-            # numeric OID
-            query += "WHERE t.oid::integer = %s" % oid
-
-          elsif m = oid.match(/"?(\w+)"?\."?(\w+)"?/)
-            # namespace and type name
-            query += "WHERE ns.nspname = '%s' AND t.typname = '%s'" % [m[1], m[2]]
-
-          else
-            # only type name
-            query += "WHERE t.typname = '%s' AND ns.nspname = ANY(current_schemas(true))" % oid
-          end
+        if oids
+          query += "WHERE t.oid::integer IN (%s)" % oids.join(", ")
         else
           query += initializer.query_conditions_for_initial_load(type_map)
         end
@@ -255,8 +176,8 @@ module ArJdbc
       ActiveRecord::Type.register(:json, OID::Json, adapter: :postgresql)
       ActiveRecord::Type.register(:jsonb, OID::Jsonb, adapter: :postgresql)
       ActiveRecord::Type.register(:money, OID::Money, adapter: :postgresql)
-      ActiveRecord::Type.register(:point, OID::Rails51Point, adapter: :postgresql)
-      ActiveRecord::Type.register(:legacy_point, OID::Point, adapter: :postgresql)
+      ActiveRecord::Type.register(:point, OID::Point, adapter: :postgresql)
+      ActiveRecord::Type.register(:legacy_point, OID::LegacyPoint, adapter: :postgresql)
       ActiveRecord::Type.register(:uuid, OID::Uuid, adapter: :postgresql)
       ActiveRecord::Type.register(:vector, OID::Vector, adapter: :postgresql)
       ActiveRecord::Type.register(:xml, OID::Xml, adapter: :postgresql)

data/lib/arjdbc/sqlite3/adapter.rb

@@ -4,11 +4,14 @@ require "arjdbc/abstract/core"
 require "arjdbc/abstract/database_statements"
 require 'arjdbc/abstract/statement_cache'
 require "arjdbc/abstract/transaction_support"
+require "active_record/connection_adapters/abstract_adapter"
 require "active_record/connection_adapters/statement_pool"
-require "active_record/connection_adapters/abstract/database_statements"
 require "active_record/connection_adapters/sqlite3/explain_pretty_printer"
 require "active_record/connection_adapters/sqlite3/quoting"
 require "active_record/connection_adapters/sqlite3/schema_creation"
+require "active_record/connection_adapters/sqlite3/schema_definitions"
+require "active_record/connection_adapters/sqlite3/schema_dumper"
+require "active_record/connection_adapters/sqlite3/schema_statements"
 
 module ArJdbc
   # All the code in this module is a copy of ConnectionAdapters::SQLite3Adapter from active_record 5.
@@ -28,7 +31,10 @@ module ArJdbc
 
     ADAPTER_NAME = 'SQLite'.freeze
 
-    include Quoting
+    # DIFFERENCE: FQN
+    include ::ActiveRecord::ConnectionAdapters::SQLite3::Quoting
+    include ::ActiveRecord::ConnectionAdapters::SQLite3::ColumnDumper
+    include ::ActiveRecord::ConnectionAdapters::SQLite3::SchemaStatements
 
     NATIVE_DATABASE_TYPES = {
         primary_key:  "INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL",
@@ -52,19 +58,28 @@ module ArJdbc
       end
     end
 
+    def update_table_definition(table_name, base) # :nodoc:
+      # DIFFERENCE: FQN
+      ::ActiveRecord::ConnectionAdapters::SQLite3::Table.new(table_name, base)
+    end
+
     def schema_creation # :nodoc:
-      SQLite3::SchemaCreation.new self
+      # DIFFERENCE: FQN
+      ::ActiveRecord::ConnectionAdapters::SQLite3::SchemaCreation.new self
     end
 
     def arel_visitor # :nodoc:
       Arel::Visitors::SQLite.new(self)
     end
 
-    def initialize(connection, logger, connection_options, config)
+    # DIFFERENCE: we remove connection_options because we are not using it.
+    def initialize(connection, logger, config)
       super(connection, logger, config)
 
       @active     = nil
       @statements = StatementPool.new(self.class.type_cast_config_to_integer(config[:statement_limit]))
+
+      configure_connection
     end
 
     def supports_ddl_transactions?
@@ -85,17 +100,12 @@ module ArJdbc
       true
     end
 
-    # Returns true, since this connection adapter supports migrations.
-    def supports_migrations? #:nodoc:
-      true
-    end
-
-    def supports_primary_key? #:nodoc:
+    def requires_reloading?
       true
     end
 
-    def requires_reloading?
-      true
+    def supports_foreign_keys_in_create?
+      sqlite_version >= "3.6.19"
     end
 
     def supports_views?
@@ -131,10 +141,6 @@ module ArJdbc
       true
     end
 
-    def valid_type?(type)
-      true
-    end
-
     # Returns 62. SQLite supports index names up to 64
     # characters. The rest is used by Rails internally to perform
     # temporary rename operations
@@ -155,43 +161,59 @@ module ArJdbc
       true
     end
 
+    # REFERENTIAL INTEGRITY ====================================
+
+    def disable_referential_integrity # :nodoc:
+      old = query_value("PRAGMA foreign_keys")
+
+      begin
+        execute("PRAGMA foreign_keys = OFF")
+        yield
+      ensure
+        execute("PRAGMA foreign_keys = #{old}")
+      end
+    end
+    
     #--
     # DATABASE STATEMENTS ======================================
     #++
 
     def explain(arel, binds = [])
       sql = "EXPLAIN QUERY PLAN #{to_sql(arel, binds)}"
+      # DIFFERENCE: FQN
       ::ActiveRecord::ConnectionAdapters::SQLite3::ExplainPrettyPrinter.new.pp(exec_query(sql, "EXPLAIN", []))
     end
 
     def exec_query(sql, name = nil, binds = [], prepare: false)
-      type_casted_binds = binds.map { |attr| type_cast(attr.value_for_database) }
-
-      log(sql, name, binds) do
-        # Don't cache statements if they are not prepared
-        unless prepare
-          stmt    = @connection.prepare(sql)
-          begin
-            cols    = stmt.columns
-            unless without_prepared_statement?(binds)
-              stmt.bind_params(type_casted_binds)
+      type_casted_binds = type_casted_binds(binds)
+
+      log(sql, name, binds, type_casted_binds) do
+        ActiveSupport::Dependencies.interlock.permit_concurrent_loads do
+          # Don't cache statements if they are not prepared
+          unless prepare
+            stmt = @connection.prepare(sql)
+            begin
+              cols = stmt.columns
+              unless without_prepared_statement?(binds)
+                stmt.bind_params(type_casted_binds)
+              end
+              records = stmt.to_a
+            ensure
+              stmt.close
             end
+          else
+            cache = @statements[sql] ||= {
+                stmt: @connection.prepare(sql)
+            }
+            stmt = cache[:stmt]
+            cols = cache[:cols] ||= stmt.columns
+            stmt.reset!
+            stmt.bind_params(type_casted_binds)
             records = stmt.to_a
-          ensure
-            stmt.close
           end
-          stmt = records
-        else
-          cache = @statements[sql] ||= {
-              :stmt => @connection.prepare(sql)
-          }
-          stmt = cache[:stmt]
-          cols = cache[:cols] ||= stmt.columns
-          stmt.reset!
-          stmt.bind_params(type_casted_binds)
-        end
 
-        ActiveRecord::Result.new(cols, stmt.to_a)
+          ActiveRecord::Result.new(cols, records)
+        end
       end
     end
 
@@ -206,7 +228,11 @@ module ArJdbc
     end
 
     def execute(sql, name = nil) #:nodoc:
-      log(sql, name) { @connection.execute(sql) }
+      log(sql, name) do
+        ActiveSupport::Dependencies.interlock.permit_concurrent_loads do
+          @connection.execute(sql)
+        end
+      end
     end
 
     def begin_db_transaction #:nodoc:
@@ -223,80 +249,30 @@ module ArJdbc
 
     # SCHEMA STATEMENTS ========================================
 
-    def tables(name = nil) # :nodoc:
-      ActiveSupport::Deprecation.warn(<<-MSG.squish)
-          #tables currently returns both tables and views.
-          This behavior is deprecated and will be changed with Rails 5.1 to only return tables.
-          Use #data_sources instead.
-      MSG
-
-      if name
-        ActiveSupport::Deprecation.warn(<<-MSG.squish)
-            Passing arguments to #tables is deprecated without replacement.
-        MSG
+    def new_column_from_field(table_name, field) # :nondoc:
+      case field["dflt_value"]
+      when /^null$/i
+        field["dflt_value"] = nil
+      when /^'(.*)'$/m
+        field["dflt_value"] = $1.gsub("''", "'")
+      when /^"(.*)"$/m
+        field["dflt_value"] = $1.gsub('""', '"')
       end
 
-      data_sources
-    end
-
-    def data_sources
-      select_values("SELECT name FROM sqlite_master WHERE type IN ('table','view') AND name <> 'sqlite_sequence'", "SCHEMA")
-    end
-
-    def table_exists?(table_name)
-      ActiveSupport::Deprecation.warn(<<-MSG.squish)
-          #table_exists? currently checks both tables and views.
-          This behavior is deprecated and will be changed with Rails 5.1 to only check tables.
-          Use #data_source_exists? instead.
-      MSG
-
-      data_source_exists?(table_name)
-    end
-
-    def data_source_exists?(table_name)
-      return false unless table_name.present?
-
-      sql = "SELECT name FROM sqlite_master WHERE type IN ('table','view') AND name <> 'sqlite_sequence'"
-      sql << " AND name = #{quote(table_name)}"
-
-      select_values(sql, "SCHEMA").any?
-    end
-
-    def views # :nodoc:
-      select_values("SELECT name FROM sqlite_master WHERE type = 'view' AND name <> 'sqlite_sequence'", "SCHEMA")
-    end
-
-    def view_exists?(view_name) # :nodoc:
-      return false unless view_name.present?
-
-      sql = "SELECT name FROM sqlite_master WHERE type = 'view' AND name <> 'sqlite_sequence'"
-      sql << " AND name = #{quote(view_name)}"
-
-      select_values(sql, "SCHEMA").any?
-    end
-
-    # Returns an array of +Column+ objects for the table specified by +table_name+.
-    def columns(table_name) # :nodoc:
-      table_name = table_name.to_s
-      table_structure(table_name).map do |field|
-        case field["dflt_value"]
-          when /^null$/i
-            field["dflt_value"] = nil
-          when /^'(.*)'$/m
-            field["dflt_value"] = $1.gsub("''", "'")
-          when /^"(.*)"$/m
-            field["dflt_value"] = $1.gsub('""', '"')
-        end
-
-        collation = field["collation"]
-        sql_type = field["type"]
-        type_metadata = fetch_type_metadata(sql_type)
-        new_column(field["name"], field["dflt_value"], type_metadata, field["notnull"].to_i == 0, table_name, nil, collation)
-      end
+      collation = field["collation"]
+      sql_type = field["type"]
+      type_metadata = fetch_type_metadata(sql_type)
+      new_column(field["name"], field["dflt_value"], type_metadata, field["notnull"].to_i == 0, table_name, nil, collation)
     end
 
     # Returns an array of indexes for the given table.
     def indexes(table_name, name = nil) #:nodoc:
+      if name
+        ActiveSupport::Deprecation.warn(<<-MSG.squish)
+            Passing name to #indexes is deprecated without replacement.
+          MSG
+      end
+
       exec_query("PRAGMA index_list(#{quote_table_name(table_name)})", "SCHEMA").map do |row|
         sql = <<-SQL
             SELECT sql
@@ -306,17 +282,17 @@ module ArJdbc
             SELECT sql
             FROM sqlite_temp_master
             WHERE name=#{quote(row['name'])} AND type='index'
-        SQL
+          SQL
         index_sql = exec_query(sql).first["sql"]
         match = /\sWHERE\s+(.+)$/i.match(index_sql)
         where = match[1] if match
         IndexDefinition.new(
-            table_name,
-            row["name"],
-            row["unique"] != 0,
-            exec_query("PRAGMA index_info('#{row['name']}')", "SCHEMA").map { |col|
-              col["name"]
-            }, nil, nil, where)
+          table_name,
+          row["name"],
+          row["unique"] != 0,
+          exec_query("PRAGMA index_info('#{row['name']}')", "SCHEMA").map { |col|
+            col["name"]
+          }, nil, nil, where)
       end
     end
 
@@ -346,7 +322,7 @@ module ArJdbc
     end
 
     def add_column(table_name, column_name, type, options = {}) #:nodoc:
-      if valid_alter_table_type?(type)
+      if valid_alter_table_type?(type) && !options[:primary_key]
         super(table_name, column_name, type, options)
       else
         alter_table(table_name) do |definition|
@@ -380,11 +356,10 @@ module ArJdbc
 
     def change_column(table_name, column_name, type, options = {}) #:nodoc:
       alter_table(table_name) do |definition|
-        include_default = options_include_default?(options)
         definition[column_name].instance_eval do
           self.type    = type
           self.limit   = options[:limit] if options.include?(:limit)
-          self.default = options[:default] if include_default
+          self.default = options[:default] if options.include?(:default)
           self.null    = options[:null] if options.include?(:null)
           self.precision = options[:precision] if options.include?(:precision)
           self.scale   = options[:scale] if options.include?(:scale)
@@ -399,14 +374,34 @@ module ArJdbc
       rename_column_indexes(table_name, column.name, new_column_name)
     end
 
-    protected
+    def add_reference(table_name, ref_name, **options) # :nodoc:
+      super(table_name, ref_name, type: :integer, **options)
+    end
+    alias :add_belongs_to :add_reference
+
+    def foreign_keys(table_name)
+      fk_info = exec_query("PRAGMA foreign_key_list(#{quote(table_name)})", "SCHEMA")
+      fk_info.map do |row|
+        options = {
+          column: row["from"],
+          primary_key: row["to"],
+          on_delete: extract_foreign_key_action(row["on_delete"]),
+          on_update: extract_foreign_key_action(row["on_update"])
+        }
+        # DIFFERENCE: FQN
+        ::ActiveRecord::ConnectionAdapters::ForeignKeyDefinition.new(table_name, row["table"], options)
+      end
+    end
+    
+    private
 
     def table_structure(table_name)
       structure = exec_query("PRAGMA table_info(#{quote_table_name(table_name)})", "SCHEMA")
       raise(ActiveRecord::StatementInvalid, "Could not find table '#{table_name}'") if structure.empty?
       table_structure_with_collation(table_name, structure)
     end
-
+    alias column_definitions table_structure
+    
     def alter_table(table_name, options = {}) #:nodoc:
       altered_table_name = "a#{table_name}"
       caller = lambda { |definition| yield definition if block_given? }
@@ -418,28 +413,31 @@ module ArJdbc
       end
     end
 
-    def move_table(from, to, options = {}, &block) #:nodoc:
+    def move_table(from, to, options = {}, &block)
       copy_table(from, to, options, &block)
       drop_table(from)
     end
 
-    def copy_table(from, to, options = {}) #:nodoc:
+    def copy_table(from, to, options = {})
       from_primary_key = primary_key(from)
       options[:id] = false
       create_table(to, options) do |definition|
         @definition = definition
-        @definition.primary_key(from_primary_key) if from_primary_key.present?
+        if from_primary_key.is_a?(Array)
+          @definition.primary_keys from_primary_key
+        end
         columns(from).each do |column|
           column_name = options[:rename] ?
               (options[:rename][column.name] ||
                   options[:rename][column.name.to_sym] ||
                   column.name) : column.name
-          next if column_name == from_primary_key
 
           @definition.column(column_name, column.type,
                              limit: column.limit, default: column.default,
                              precision: column.precision, scale: column.scale,
-                             null: column.null, collation: column.collation)
+                             null: column.null, collation: column.collation,
+                             primary_key: column_name == from_primary_key
+          )
         end
         yield @definition if block_given?
       end
@@ -449,9 +447,12 @@ module ArJdbc
                           options[:rename] || {})
     end
 
-    def copy_table_indexes(from, to, rename = {}) #:nodoc:
+    def copy_table_indexes(from, to, rename = {})
       indexes(from).each do |index|
         name = index.name
+        # indexes sqlite creates for internal use start with `sqlite_` and
+        # don't need to be copied
+        next if name.starts_with?("sqlite_")
         if to == "a#{from}"
           name = "t#{name}"
         elsif from == "a#{to}"
@@ -467,12 +468,13 @@ module ArJdbc
           # index name can't be the same
           opts = { name: name.gsub(/(^|_)(#{from})_/, "\\1#{to}_"), internal: true }
           opts[:unique] = true if index.unique
+          opts[:where] = index.where if index.where
           add_index(to, columns, opts)
         end
       end
     end
 
-    def copy_table_contents(from, to, columns, rename = {}) #:nodoc:
+    def copy_table_contents(from, to, columns, rename = {})
       column_mappings = Hash[columns.map { |name| [name, name] }]
       rename.each { |a| column_mappings[a.last] = a.first }
       from_columns = columns(from).collect(&:name)
@@ -496,43 +498,51 @@ module ArJdbc
         # Older versions of SQLite return:
         #   column *column_name* is not unique
         when /column(s)? .* (is|are) not unique/, /UNIQUE constraint failed: .*/
-          RecordNotUnique.new(message)
+          # DIFFERENCE: FQN
+          ::ActiveRecord::RecordNotUnique.new(message)
+        when /.* may not be NULL/, /NOT NULL constraint failed: .*/
+          # DIFFERENCE: FQN
+          ::ActiveRecord::NotNullViolation.new(message)
+        when /FOREIGN KEY constraint failed/i
+          # DIFFERENCE: FQN
+          ::ActiveRecord::InvalidForeignKey.new(message)
         else
           super
       end
     end
 
-    private
     COLLATE_REGEX = /.*\"(\w+)\".*collate\s+\"(\w+)\".*/i.freeze
 
     def table_structure_with_collation(table_name, basic_structure)
       collation_hash = {}
-      sql            = "SELECT sql FROM
-                              (SELECT * FROM sqlite_master UNION ALL
-                               SELECT * FROM sqlite_temp_master)
-                            WHERE type='table' and name='#{ table_name }' \;"
+      sql = <<-SQL
+            SELECT sql FROM
+              (SELECT * FROM sqlite_master UNION ALL
+               SELECT * FROM sqlite_temp_master)
+            WHERE type = 'table' AND name = #{quote(table_name)}
+      SQL
 
       # Result will have following sample string
       # CREATE TABLE "users" ("id" INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
       #                       "password_digest" varchar COLLATE "NOCASE");
-      result = exec_query(sql, 'SCHEMA').first
+      result = exec_query(sql, "SCHEMA").first
 
       if result
         # Splitting with left parentheses and picking up last will return all
         # columns separated with comma(,).
-        columns_string = result["sql"].split('(').last
+        columns_string = result["sql"].split("(").last
 
-        columns_string.split(',').each do |column_string|
+        columns_string.split(",").each do |column_string|
           # This regex will match the column name and collation type and will save
           # the value in $1 and $2 respectively.
-          collation_hash[$1] = $2 if (COLLATE_REGEX =~ column_string)
+          collation_hash[$1] = $2 if COLLATE_REGEX =~ column_string
         end
 
         basic_structure.map! do |column|
-          column_name = column['name']
+          column_name = column["name"]
 
           if collation_hash.has_key? column_name
-            column['collation'] = collation_hash[column_name]
+            column["collation"] = collation_hash[column_name]
           end
 
           column
@@ -541,6 +551,23 @@ module ArJdbc
         basic_structure.to_hash
       end
     end
+
+    def create_table_definition(*args)
+      # DIFFERENCE: FQN
+      ::ActiveRecord::ConnectionAdapters::SQLite3::TableDefinition.new(*args)
+    end
+
+    def extract_foreign_key_action(specifier)
+      case specifier
+      when "CASCADE"; :cascade
+      when "SET NULL"; :nullify
+      when "RESTRICT"; :restrict
+      end
+    end
+
+    def configure_connection
+      execute("PRAGMA foreign_keys = ON", "SCHEMA")
+    end
   end
 end
 
@@ -651,10 +678,14 @@ module ActiveRecord::ConnectionAdapters
     include ArJdbc::Abstract::StatementCache
     include ArJdbc::Abstract::TransactionSupport
 
+    def supports_transaction_isolation?
+      false
+    end
+
     def begin_isolated_db_transaction(isolation)
       raise ActiveRecord::TransactionIsolationError, 'adapter does not support setting transaction isolation'
     end
-
+    
     # SQLite driver doesn't support all types of insert statements with executeUpdate so
     # make it act like a regular query and the ids will be returned from #last_inserted_id
     # example: INSERT INTO "aircraft" DEFAULT VALUES