Sha256: 25354ca6643b15075c45d9a4dc697b1a794c450fc6dd31e8c43ac7664a9014cd

Contents?: true

Size: 1.56 KB

Versions: 9

Compression:

Stored size: 1.56 KB

Contents

module ActiveRecord
  module ConnectionAdapters
    class PostgreSQLColumn
      # This overide is needed to ensure ActiveRecord::Dirty behaves as expected
      def type_cast(value)
        return value if value.kind_of?(Array)
        super
      end
    end
    
    class PostgreSQLAdapter
      def insert_sql(sql, name = nil, pk = nil, id_value = nil, sequence_name = nil)
        unless pk
          # Extract the table from the insert sql. Yuck.
          table_ref = extract_table_ref_from_insert_sql(sql)
          pk = primary_key(table_ref) if table_ref
        end

        if pk
          # CPK
          # select_value("#{sql} RETURNING #{quote_column_name(pk)}")
          select_value("#{sql} RETURNING #{quote_column_names(pk)}")
        else
          super
        end
      end
      alias :create :insert

      def sql_for_insert(sql, pk, id_value, sequence_name, binds)
        unless pk
          # Extract the table from the insert sql. Yuck.
          table_ref = extract_table_ref_from_insert_sql(sql)
          pk = primary_key(table_ref) if table_ref
        end

        # CPK
        # sql = "#{sql} RETURNING #{quote_column_name(pk)}" if pk
        sql = "#{sql} RETURNING #{quote_column_names(pk)}" if pk

        [sql, binds]
      end

      # Returns a single value if query returns a single element
      # otherwise returns an array coresponding to the composite keys
      #
      def last_inserted_id(result)
        row = result && result.rows.first
        if Array === row
          row.size == 1 ? row[0] : row
        end
      end
    end
  end
end

Version data entries

9 entries across 9 versions & 1 rubygems

Version Path
composite_primary_keys-6.0.8 lib/composite_primary_keys/connection_adapters/postgresql_adapter.rb
composite_primary_keys-6.0.7 lib/composite_primary_keys/connection_adapters/postgresql_adapter.rb
composite_primary_keys-7.0.4 lib/composite_primary_keys/connection_adapters/postgresql_adapter.rb
composite_primary_keys-7.0.3 lib/composite_primary_keys/connection_adapters/postgresql_adapter.rb
composite_primary_keys-7.0.2 lib/composite_primary_keys/connection_adapters/postgresql_adapter.rb
composite_primary_keys-7.0.1 lib/composite_primary_keys/connection_adapters/postgresql_adapter.rb
composite_primary_keys-7.0.0 lib/composite_primary_keys/connection_adapters/postgresql_adapter.rb
composite_primary_keys-6.0.6 lib/composite_primary_keys/connection_adapters/postgresql_adapter.rb
composite_primary_keys-6.0.5 lib/composite_primary_keys/connection_adapters/postgresql_adapter.rb