bmg 0.9.1 → 0.10.0

data/lib/bmg/sql/nodes/set_operator.rb

@@ -0,0 +1,68 @@
+module Bmg
+  module Sql
+    module SetOperator
+      include Expr
+
+      def left
+        self[2]
+      end
+
+      def right
+        self[3]
+      end
+
+      def set_quantifier
+        self[1]
+      end
+
+      def with_exp?
+        false
+      end
+
+      def set_operator?
+        true
+      end
+
+      def is_table_dee?
+        false
+      end
+
+      def distinct?
+        set_quantifier.distinct?
+      end
+
+      def all?
+        set_quantifier.all?
+      end
+
+      def should_be_reused?
+        true
+      end
+
+      def order_by_clause
+        nil
+      end
+
+      def to_attr_list
+        self.last.to_attr_list
+      end
+
+      def to_sql(buffer, dialect, parenthesize = !buffer.empty?)
+        if parenthesize
+          sql_parenthesized(buffer){|b| to_sql(b, dialect, false) }
+        else
+          left.to_sql(buffer, dialect, true)
+          buffer << SPACE << keyword
+          unless distinct?
+            buffer << SPACE
+            set_quantifier.to_sql(buffer, dialect)
+          end
+          buffer << SPACE
+          right.to_sql(buffer, dialect, true)
+          buffer
+        end
+      end
+
+    end # module SetOperator
+  end # module Sql
+end # module Bmg

data/lib/bmg/sql/nodes/set_quantifier.rb

@@ -0,0 +1,20 @@
+module Bmg
+  module Sql
+    module SetQuantifier
+      include Expr
+
+      def all?
+        last == "all"
+      end
+
+      def distinct?
+        last == "distinct"
+      end
+
+      def to_sql(buffer, dialect)
+        buffer << self.last
+      end
+
+    end # module SetQuantifier
+  end # module Sql
+end # module Bmg

data/lib/bmg/sql/nodes/subquery_as.rb

@@ -0,0 +1,28 @@
+module Bmg
+  module Sql
+    module SubqueryAs
+      include Expr
+
+      def left
+        self[1]
+      end
+      alias :subquery :left
+
+      def right
+        self[2]
+      end
+
+      def as_name
+        self[2].last
+      end
+
+      def to_sql(buffer, dialect)
+        left.to_sql(buffer, dialect)
+        buffer << SPACE << AS << SPACE
+        right.to_sql(buffer, dialect)
+        buffer
+      end
+
+    end # module SubqueryAs
+  end # module Sql
+end # module Bmg

data/lib/bmg/sql/nodes/table_as.rb

@@ -0,0 +1,31 @@
+module Bmg
+  module Sql
+    module TableAs
+      include Expr
+
+      def left
+        self[1]
+      end
+
+      def right
+        self[2]
+      end
+
+      def table_name
+        self[1].last
+      end
+
+      def as_name
+        self[2].last
+      end
+
+      def to_sql(buffer, dialect)
+        self[1].to_sql(buffer, dialect)
+        buffer << SPACE << AS << SPACE
+        self[2].to_sql(buffer, dialect)
+        buffer
+      end
+
+    end # module TableAs
+  end # module Sql
+end # module Bmg

data/lib/bmg/sql/nodes/table_name.rb

@@ -0,0 +1,17 @@
+module Bmg
+  module Sql
+    module TableName
+      include Expr
+
+      def value
+        last
+      end
+
+      def to_sql(buffer, dialect)
+        buffer << dialect.quote_identifier(last.to_s)
+        buffer
+      end
+
+    end # module TableName
+  end # module Sql
+end # module Bmg

data/lib/bmg/sql/nodes/union.rb

@@ -0,0 +1,14 @@
+module Bmg
+  module Sql
+    module Union
+      include SetOperator
+
+      UNION = "UNION".freeze
+
+      def keyword
+        UNION
+      end
+
+    end # module Union
+  end # module Sql
+end # module Bmg

data/lib/bmg/sql/nodes/where_clause.rb

@@ -0,0 +1,20 @@
+module Bmg
+  module Sql
+    module WhereClause
+      include Expr
+
+      WHERE = "WHERE".freeze
+
+      def predicate
+        last
+      end
+
+      def to_sql(buffer, dialect)
+        buffer << WHERE << SPACE
+        predicate.to_sql(buffer, dialect)
+        buffer
+      end
+
+    end # module WhereClause
+  end # module Sql
+end # module Bmg

data/lib/bmg/sql/nodes/with_exp.rb

@@ -0,0 +1,51 @@
+module Bmg
+  module Sql
+    module WithExp
+      include Expr
+      extend Forwardable
+
+      WITH = "WITH".freeze
+
+      def with_exp?
+        true
+      end
+
+      def with_spec
+        self[1]
+      end
+
+      def select_exp
+        last
+      end
+      def_delegators :select_exp, :select_list,
+                                  :where_clause,
+                                  :predicate,
+                                  :from_clause,
+                                  :table_spec,
+                                  :order_by_clause,
+                                  :limit_clause,
+                                  :offset_clause,
+                                  :desaliaser,
+                                  :to_attr_list,
+                                  :to_ordering,
+                                  :all?,
+                                  :distinct?,
+                                  :set_operator?,
+                                  :limit_or_offset?,
+                                  :join?,
+                                  :should_be_reused?,
+                                  :is_table_dee?
+
+    # to_xxx
+
+      def to_sql(buffer, dialect)
+        buffer << WITH << SPACE
+        self[1].to_sql(buffer, dialect)
+        buffer << SPACE
+        self[2].to_sql(buffer, dialect, false)
+        buffer
+      end
+
+    end # module WithExp
+  end # module Sql
+end # module Bmg

data/lib/bmg/sql/nodes/with_spec.rb

@@ -0,0 +1,24 @@
+module Bmg
+  module Sql
+    module WithSpec
+      include Expr
+
+      def to_sql(buffer, dialect)
+        each_child do |child,index|
+          buffer << COMMA << SPACE unless index==0
+          child.to_sql(buffer, dialect)
+        end
+        buffer
+      end
+
+      def to_hash
+        hash = {}
+        each_child do |child|
+          hash[child.table_name.value] = child.subquery
+        end
+        hash
+      end
+
+    end # module WithSpec
+  end # module Sql
+end # module Bmg

data/lib/bmg/sql/processor.rb

@@ -0,0 +1,85 @@
+module Bmg
+  module Sql
+    class Processor < Sexpr::Rewriter
+      grammar Sql::Grammar
+
+      UnexpectedError = Class.new(Bmg::Error)
+
+      def initialize(builder)
+        @builder = builder
+      end
+      attr_reader :builder
+
+      def on_with_exp(sexpr)
+        applied = apply(sexpr.select_exp)
+        if applied.with_exp?
+          merge_with_exps(sexpr, applied, applied.select_exp)
+        else
+          sexpr.with_update(-1, applied)
+        end
+      end
+
+      def on_set_operator(sexpr)
+        sexpr.each_with_index.map{|child,index|
+          index <= 1 ? child : apply(child)
+        }
+      end
+      alias :on_union     :on_set_operator
+      alias :on_except    :on_set_operator
+      alias :on_intersect :on_set_operator
+
+      def on_select_exp(sexpr)
+        sexpr.with_update(2, apply(sexpr[2]))
+      end
+
+    private
+
+      def merge_with_exps(left, right, main)
+        if left.with_exp? and right.with_exp?
+          [ :with_exp,
+            merge_with_specs(left.with_spec, right.with_spec),
+            main ]
+        elsif left.with_exp?
+          left.with_update(-1, main)
+        elsif right.with_exp?
+          right.with_update(-1, main)
+        else
+          main
+        end
+      end
+
+      def merge_with_specs(left, right)
+        hash = left.to_hash.merge(right.to_hash){|k,v1,v2|
+          unless v1 == v2
+            msg = "Unexpected different SQL expr: "
+            msg << "`#{v1.inspect}` vs. `#{v2.inspect}`"
+            raise UnexpectedError, msg
+          end
+          v1
+        }
+        hash.map{|(k,v)| builder.name_intro(k,v) }.unshift(:with_spec)
+      end
+
+      def tautology
+        Predicate::Factory.tautology
+      end
+
+    end
+  end
+end
+require_relative 'processor/distinct'
+require_relative 'processor/all'
+require_relative 'processor/clip'
+require_relative 'processor/star'
+require_relative 'processor/rename'
+require_relative 'processor/order_by'
+require_relative 'processor/limit_offset'
+require_relative 'processor/from_self'
+require_relative 'processor/reorder'
+require_relative 'processor/merge'
+require_relative 'processor/where'
+require_relative 'processor/join_support'
+require_relative 'processor/join'
+require_relative 'processor/semi_join'
+require_relative 'processor/flatten'
+require_relative 'processor/requalify'

data/lib/bmg/sql/processor/all.rb

@@ -0,0 +1,17 @@
+module Bmg
+  module Sql
+    class Processor
+      class All < Processor
+
+        def on_set_quantified(sexpr)
+          sexpr.with_update(1, builder.all)
+        end
+        alias :on_union      :on_set_quantified
+        alias :on_except     :on_set_quantified
+        alias :on_intersect  :on_set_quantified
+        alias :on_select_exp :on_set_quantified
+
+      end # class All
+    end # class Processor
+  end # module Sql
+end # module Bmg

data/lib/bmg/sql/processor/clip.rb

@@ -0,0 +1,57 @@
+module Bmg
+  module Sql
+    class Processor
+      class Clip < Processor
+
+        def initialize(attributes, allbut = false, on_empty = :is_table_dee, builder)
+          super(builder)
+          @attributes = attributes
+          @allbut = allbut
+          @on_empty = on_empty
+        end
+        attr_reader :attributes, :on_empty
+
+        def allbut?
+          @allbut
+        end
+
+        def on_set_operator(sexpr)
+          apply(builder.from_self(sexpr))
+        end
+        alias :on_union     :on_set_operator
+        alias :on_except    :on_set_operator
+        alias :on_intersect :on_set_operator
+
+        def on_select_exp(sexpr)
+          catch(:empty){ return super(sexpr) }
+          send("select_#{on_empty}", sexpr)
+        end
+
+        def on_select_star(sexpr)
+          raise NotImplementedError, "Allbut on * is not supported" if allbut?
+          builder.select_list(attributes, builder.last_qualifier)
+        end
+
+        def on_select_list(sexpr)
+          allbut = self.allbut?
+          result = sexpr.select{|child|
+            (child == :select_list) or
+            attributes.include?(child.as_name.to_sym) == !allbut
+          }
+          (result.size==1) ? throw(:empty) : result
+        end
+
+      private
+
+        def select_is_table_dee(sexpr)
+          builder.select_is_table_dee(select_star(sexpr))
+        end
+
+        def select_star(sexpr)
+          All.new(builder).call(Star.new(builder).call(sexpr))
+        end
+
+      end # class Clip
+    end # class Processor
+  end # module Sql
+end # module Bmg