sql-parser2 0.0.3 → 0.0.4

data/lib/sql-parser/parser.rex

@@ -77,6 +77,7 @@ rule
             CURRENT_USER  { [:CURRENT_USER, text] }
             VALUES        { [:VALUES, text] }
             SET           { [:SET, text] }
+            LIMIT         { [:LIMIT, text] }
 
 # tokens
             E             { [:E, text] }

data/lib/sql-parser/parser.rex.rb

@@ -198,6 +198,9 @@ class SQLParser::Parser < Racc::Parser
       when (text = @ss.scan(/SET/i))
          action { [:SET, text] }
 
+      when (text = @ss.scan(/LIMIT/i))
+         action { [:LIMIT, text] }
+
       when (text = @ss.scan(/E/i))
          action { [:E, text] }
 

data/lib/sql-parser/sql_visitor.rb

@@ -43,7 +43,8 @@ module SQLParser
     def visit_DirectSelect(o)
       [
         o.query_expression,
-        o.order_by
+        o.order_by,
+        o.limit_clause
       ].compact.collect { |e| visit(e) }.join(' ')
     end
 
@@ -51,6 +52,10 @@ module SQLParser
       "ORDER BY #{arrayize(o.sort_specification)}"
     end
 
+    def visit_Limit(o)
+      "LIMIT #{o.limit_count}"
+    end
+
     def visit_Subquery(o)
       "(#{visit(o.query_specification)})"
     end

data/lib/sql-parser/statement.rb

@@ -80,13 +80,15 @@ module SQLParser
     
     class DirectSelect < Node
       
-      def initialize(query_expression, order_by)
+      def initialize(query_expression, order_by, limit_clause = nil)
         @query_expression = query_expression
         @order_by = order_by
+        @limit_clause = limit_clause
       end
 
       attr_reader :query_expression
       attr_reader :order_by
+      attr_reader :limit_clause
       
     end
 
@@ -100,6 +102,16 @@ module SQLParser
       
     end
 
+    class Limit < Node
+
+      def initialize(limit_count)
+        @limit_count = limit_count
+      end
+
+      attr_reader :limit_count
+
+    end
+
     class Subquery < Node
       
       def initialize(query_specification)

data/lib/sql-parser/version.rb

@@ -1,5 +1,5 @@
 module SQLParser
   
-  VERSION = '0.0.3'
+  VERSION = '0.0.4'
   
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sql-parser2
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Dray Lacy