From 93555c672e3edd5705eb381f2bf3d4f93cffa96f Mon Sep 17 00:00:00 2001
From: Pratik Naik <pratiknaik@gmail.com>
Date: Fri, 1 Jan 2010 00:56:49 +0530
Subject: Add Relation#table to get the relevant Arel::Table

---
 activerecord/lib/active_record/relation.rb | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'activerecord/lib/active_record/relation.rb')

diff --git a/activerecord/lib/active_record/relation.rb b/activerecord/lib/active_record/relation.rb
index 502aa909ab..6a1237cdd1 100644
--- a/activerecord/lib/active_record/relation.rb
+++ b/activerecord/lib/active_record/relation.rb
@@ -6,7 +6,7 @@ module ActiveRecord
     delegate :length, :collect, :map, :each, :all?, :to => :to_a
 
     attr_reader :relation, :klass, :preload_associations, :eager_load_associations
-    attr_writer :readonly, :preload_associations, :eager_load_associations
+    attr_writer :readonly, :preload_associations, :eager_load_associations, :table
 
     def initialize(klass, relation)
       @klass, @relation = klass, relation
@@ -133,9 +133,18 @@ module ActiveRecord
       relation.readonly = @readonly
       relation.preload_associations = @preload_associations
       relation.eager_load_associations = @eager_load_associations
+      relation.table = table
       relation
     end
 
+    def table
+      @table ||= Arel::Table.new(@klass.table_name, Arel::Sql::Engine.new(@klass))
+    end
+
+    def primary_key
+      @primary_key ||= table[@klass.primary_key]
+    end
+
     protected
 
     def method_missing(method, *args, &block)
-- 
cgit v1.2.3