From c66fcc2b151337d337e16ead970db11491d74bbe Mon Sep 17 00:00:00 2001 From: Raimonds Simanovskis Date: Tue, 28 Sep 2010 16:15:35 +0300 Subject: pass primary key name and value to ActiveRecord adapter insert method necessary for Oracle adapter for RETURNING ... INTO ... clause generation --- lib/arel/select_manager.rb | 12 ++++++++++-- lib/arel/table.rb | 6 +++++- 2 files changed, 15 insertions(+), 3 deletions(-) diff --git a/lib/arel/select_manager.rb b/lib/arel/select_manager.rb index 21b9395120..77930a6bb1 100644 --- a/lib/arel/select_manager.rb +++ b/lib/arel/select_manager.rb @@ -163,9 +163,17 @@ module Arel # FIXME: this method should go away def insert values im = InsertManager.new @engine - im.into @ctx.froms + table = @ctx.froms + primary_key_name = (primary_key = table.primary_key) && primary_key.name + # FIXME: in AR tests values sometimes were Array and not Hash therefore is_a?(Hash) check is added + primary_key_value = primary_key && values.is_a?(Hash) && values[primary_key] + im.into table im.insert values - @engine.connection.insert im.to_sql + # Oracle adapter needs primary key name to generate RETURNING ... INTO ... clause + # for tables which assign primary key value using trigger. + # RETURNING ... INTO ... clause will be added only if primary_key_value is nil + # therefore it is necessary to pass primary key value as well + @engine.connection.insert im.to_sql, 'AREL', primary_key_name, primary_key_value end private diff --git a/lib/arel/table.rb b/lib/arel/table.rb index ba31e3183b..3285037154 100644 --- a/lib/arel/table.rb +++ b/lib/arel/table.rb @@ -27,7 +27,11 @@ module Arel end def primary_key - @primary_key ||= self[@engine.connection.primary_key(name)] + @primary_key ||= begin + primary_key_name = @engine.connection.primary_key(name) + # some tables might be without primary key + primary_key_name && self[primary_key_name] + end end def alias -- cgit v1.2.3 From 6f8c4f183d47bcc36fc30230bb55f90e59a98a37 Mon Sep 17 00:00:00 2001 From: Raimonds Simanovskis Date: Tue, 28 Sep 2010 17:05:34 +0300 Subject: added missing method visit_Arel_Attributes_Decimal as alias for visit_Arel_Attributes_Attribute --- lib/arel/visitors/to_sql.rb | 1 + 1 file changed, 1 insertion(+) diff --git a/lib/arel/visitors/to_sql.rb b/lib/arel/visitors/to_sql.rb index 274a491f78..fc821a0ddb 100644 --- a/lib/arel/visitors/to_sql.rb +++ b/lib/arel/visitors/to_sql.rb @@ -234,6 +234,7 @@ module Arel "#{quote_table_name join_name}.#{quote_column_name o.name}" end alias :visit_Arel_Attributes_Integer :visit_Arel_Attributes_Attribute + alias :visit_Arel_Attributes_Decimal :visit_Arel_Attributes_Attribute alias :visit_Arel_Attributes_String :visit_Arel_Attributes_Attribute alias :visit_Arel_Attributes_Time :visit_Arel_Attributes_Attribute alias :visit_Arel_Attributes_Boolean :visit_Arel_Attributes_Attribute -- cgit v1.2.3