From abffef945a64a32ac959ee11d41593d575bcccab Mon Sep 17 00:00:00 2001
From: Aaron Patterson <aaron.patterson@gmail.com>
Date: Fri, 25 Feb 2011 15:19:11 -0800
Subject: Adding SelectManager#limit= and removing limit nodes when nil is
 assigned to limit

---
 lib/arel/select_manager.rb | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

(limited to 'lib/arel')

diff --git a/lib/arel/select_manager.rb b/lib/arel/select_manager.rb
index adffcca764..37405b4596 100644
--- a/lib/arel/select_manager.rb
+++ b/lib/arel/select_manager.rb
@@ -172,10 +172,16 @@ module Arel
     end
 
     def take limit
-      @ast.limit = Nodes::Limit.new(limit)
-      @ctx.top   = Nodes::Top.new(limit)
+      if limit
+        @ast.limit = Nodes::Limit.new(limit)
+        @ctx.top   = Nodes::Top.new(limit)
+      else
+        @ast.limit = nil
+        @ctx.top   = nil
+      end
       self
     end
+    alias limit= take
 
     def join_sql
       return nil if @ctx.source.right.empty?
-- 
cgit v1.2.3