RubyGems - arel_extensions - Versions diffs - 0.8.0 - Mend

arel_extensions 0.8.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (70) hide show

checksums.yaml +7 -0
data/.gitignore +9 -0
data/.travis.yml +46 -0
data/Gemfile +10 -0
data/MIT-LICENSE.txt +20 -0
data/README.md +101 -0
data/Rakefile +14 -0
data/arel_extensions.gemspec +32 -0
data/functions.html +344 -0
data/gemfiles/Gemfile.rails3 +10 -0
data/gemfiles/Gemfile.rails5 +10 -0
data/init/mssql.sql +6 -0
data/init/mysql.sql +0 -0
data/init/oracle.sql +31 -0
data/init/postgresql.sql +12 -0
data/init/sqlite.sql +1 -0
data/lib/arel_extensions.rb +84 -0
data/lib/arel_extensions/attributes.rb +26 -0
data/lib/arel_extensions/comparators.rb +59 -0
data/lib/arel_extensions/date_duration.rb +28 -0
data/lib/arel_extensions/insert_manager.rb +33 -0
data/lib/arel_extensions/math.rb +48 -0
data/lib/arel_extensions/math_functions.rb +35 -0
data/lib/arel_extensions/nodes.rb +27 -0
data/lib/arel_extensions/nodes/abs.rb +6 -0
data/lib/arel_extensions/nodes/ceil.rb +6 -0
data/lib/arel_extensions/nodes/coalesce.rb +22 -0
data/lib/arel_extensions/nodes/concat.rb +33 -0
data/lib/arel_extensions/nodes/date_diff.rb +106 -0
data/lib/arel_extensions/nodes/duration.rb +30 -0
data/lib/arel_extensions/nodes/find_in_set.rb +16 -0
data/lib/arel_extensions/nodes/floor.rb +6 -0
data/lib/arel_extensions/nodes/function.rb +17 -0
data/lib/arel_extensions/nodes/isnull.rb +30 -0
data/lib/arel_extensions/nodes/length.rb +6 -0
data/lib/arel_extensions/nodes/locate.rb +33 -0
data/lib/arel_extensions/nodes/ltrim.rb +28 -0
data/lib/arel_extensions/nodes/matches.rb +22 -0
data/lib/arel_extensions/nodes/rand.rb +23 -0
data/lib/arel_extensions/nodes/replace.rb +36 -0
data/lib/arel_extensions/nodes/round.rb +15 -0
data/lib/arel_extensions/nodes/rtrim.rb +29 -0
data/lib/arel_extensions/nodes/soundex.rb +23 -0
data/lib/arel_extensions/nodes/sum.rb +23 -0
data/lib/arel_extensions/nodes/trim.rb +26 -0
data/lib/arel_extensions/nodes/wday.rb +23 -0
data/lib/arel_extensions/null_functions.rb +16 -0
data/lib/arel_extensions/string_functions.rb +68 -0
data/lib/arel_extensions/version.rb +4 -0
data/lib/arel_extensions/visitors.rb +6 -0
data/lib/arel_extensions/visitors/ibm_db.rb +206 -0
data/lib/arel_extensions/visitors/mssql.rb +213 -0
data/lib/arel_extensions/visitors/mysql.rb +184 -0
data/lib/arel_extensions/visitors/oracle.rb +267 -0
data/lib/arel_extensions/visitors/postgresql.rb +258 -0
data/lib/arel_extensions/visitors/sqlite.rb +218 -0
data/lib/arel_extensions/visitors/to_sql.rb +199 -0
data/test/helper.rb +18 -0
data/test/real_db_test.rb +251 -0
data/test/support/fake_record.rb +137 -0
data/test/test_comparators.rb +49 -0
data/test/visitors/test_bulk_insert_oracle.rb +30 -0
data/test/visitors/test_bulk_insert_sqlite.rb +31 -0
data/test/visitors/test_bulk_insert_to_sql.rb +32 -0
data/test/visitors/test_oracle.rb +105 -0
data/test/visitors/test_to_sql.rb +148 -0
data/test/with_ar/test_bulk_sqlite.rb +44 -0
data/test/with_ar/test_math_sqlite.rb +59 -0
data/test/with_ar/test_string_sqlite.rb +69 -0
metadata +230 -0

data/lib/arel_extensions/visitors/mysql.rb ADDED Viewed

@@ -0,0 +1,184 @@
+module ArelExtensions
+  module Visitors
+    Arel::Visitors::MySQL.class_eval do
+      def visit_ArelExtensions_Nodes_DateDiff o, collector
+        collector << "DATEDIFF("
+        collector = visit o.left, collector
+        collector << ","
+        collector = visit o.right, collector
+        collector << ")"
+        collector
+      end
+      def visit_ArelExtensions_Nodes_Duration o, collector
+        #visit left for period
+        if(o.left == "d")
+          collector << "DAY("
+        elsif(o.left == "m")
+          collector << "MONTH("
+        elsif (o.left == "w")
+          collector << "WEEK("
+        elsif (o.left == "y")
+          collector << "YEAR("
+        end
+        #visit right
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.right, collector
+        else
+          collector << "#{o.right}"
+        end
+        collector << ")"
+        collector
+      end
+     #****************************************************************#
+      def visit_ArelExtensions_Nodes_Concat o, collector
+        arg = o.left.relation.engine.columns.find{|c| c.name == o.left.name.to_s}.type
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector << "CONCAT("
+          collector = visit o.left, collector
+          collector << ","
+          collector = visit o.right, collector
+          collector << ")"
+          collector
+        elsif (arg == :date || arg == :datetime)
+          collector << "DATE_ADD("
+          collector = visit o.left, collector
+          collector << ",INTERVAL #{o.right} DAY)"
+          collector
+        else
+          collector << "CONCAT("
+          collector = visit o.left, collector
+          collector << ",'"
+          collector << "#{o.right}')"
+          collector
+        end
+      end
+      def visit_ArelExtensions_Nodes_Coalesce o, collector
+        collector << "COALESCE("
+        if(o.left.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.left, collector
+        else
+          collector << "#{o.left}"
+        end
+           o.other.each { |a|
+          collector << ","
+          if(a.is_a?(Arel::Attributes::Attribute))
+            collector = visit a, collector
+          else
+            collector << "#{a}"
+          end
+        }
+        collector << ")"
+        collector
+      end
+      def visit_ArelExtensions_Nodes_Isnull o, collector
+          collector << "IFNULL("
+         collector = visit o.left, collector
+         collector << ","
+         if(o.right.is_a?(Arel::Attributes::Attribute))
+         collector = visit o.right, collector
+         else
+         collector << "'#{o.right}'"
+         end
+         collector << ")"
+         collector
+      end
+       def visit_ArelExtensions_Nodes_Replace o, collector
+         collector << "REPLACE("
+         collector = visit o.expr,collector
+         collector << ","
+         if(o.left.is_a?(Arel::Attributes::Attribute))
+           collector = visit o.left, collector
+         else
+           collector << "'#{o.left}'"
+         end
+         collector << ","
+         if(o.right.is_a?(Arel::Attributes::Attribute))
+           collector = visit o.right, collector
+         else
+           collector << "'#{o.right}'"
+         end
+         collector << ")"
+         collector
+       end
+      def visit_ArelExtensions_Nodes_Soundex o, collector
+        collector << "SOUNDEX("
+       if((o.expr).is_a?(Arel::Attributes::Attribute))
+        collector = visit o.expr, collector
+      else
+          collector << "'#{o.expr}'"
+       end
+        collector << ")"
+        collector
+      end
+      def visit_ArelExtensions_Nodes_Trim o , collector
+          collector << 'TRIM(BOTH '
+          if(o.right.is_a?(Arel::Attributes::Attribute))
+            collector = visit o.right, collector
+          else
+            collector << "'#{o.right}'"
+          end
+          collector << " FROM "
+          collector = visit o.left, collector
+          collector << ")"
+          collector
+      end
+      def visit_ArelExtensions_Nodes_Ltrim o , collector
+          collector << 'TRIM(LEADING '
+          if(o.right.is_a?(Arel::Attributes::Attribute))
+            collector = visit o.right, collector
+          else
+            collector << "'#{o.right}'"
+          end
+          collector << " FROM "
+          collector = visit o.left, collector
+          collector << ")"
+          collector
+      end
+      def visit_ArelExtensions_Nodes_Rtrim o , collector
+          collector << 'TRIM(TRAILING '
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.right, collector
+        else
+          collector << "'#{o.right}'"
+        end
+        collector << " FROM "
+        collector = visit o.left, collector
+        collector << ")"
+        collector
+      end
+      def visit_ArelExtensions_Nodes_Wday o, collector
+        collector << "(WEEKDAY("
+        if((o.date).is_a?(Arel::Attributes::Attribute))
+          collector = visit o.date, collector
+        else
+          collector << "'#{o.date}'"
+        end
+        collector << ") + 1) % 7"
+        collector
+      end
+    end
+  end
+end

data/lib/arel_extensions/visitors/oracle.rb ADDED Viewed

@@ -0,0 +1,267 @@
+module ArelExtensions
+  module Visitors
+    Arel::Visitors::Oracle.class_eval do
+      def visit_ArelExtensions_Nodes_IMatches o, collector
+        collector = visit o.left.lower, collector
+        collector << ' LIKE '
+        collector = visit o.right.lower(o.right), collector
+        if o.escape
+          collector << ' ESCAPE '
+          visit o.escape, collector
+        else
+          collector
+        end
+      end
+      def visit_ArelExtensions_Nodes_IDoesNotMatch o, collector
+        collector = visit o.left.lower, collector
+        collector << ' NOT LIKE '
+        collector = visit o.right.lower(o.right), collector
+        if o.escape
+          collector << ' ESCAPE '
+          visit o.escape, collector
+        else
+          collector
+        end
+      end
+  def visit_ArelExtensions_Nodes_Concat o, collector
+    arg = o.left.relation.engine.columns.find{|c| c.name == o.left.name.to_s}.type
+    if(o.right.is_a?(Arel::Attributes::Attribute))
+      collector << "CONCAT("
+      collector = visit o.left, collector
+      collector << ","
+      collector = visit o.right, collector
+      collector << ")"
+    elsif ( arg === :date || arg === :datetime)
+      collector << "DATEADD("
+      collector = visit o.left, collector
+      collector << ", + interval '#{o.right}' DAY)"
+    else
+      collector << "CONCAT("
+      collector = visit o.left, collector
+      collector << ","
+      collector << "#{o.right})"
+    end
+    collector
+  end
+  def visit_ArelExtensions_Nodes_Coalesce o, collector
+    collector << "COALESCE("
+    if(o.left.is_a?(Arel::Attributes::Attribute))
+      collector = visit o.left, collector
+    else
+      collector << "#{o.left}"
+    end
+       o.other.each { |a|
+      collector << ","
+      if(a.is_a?(Arel::Attributes::Attribute))
+        collector = visit a, collector
+      else
+        collector << "#{a}"
+      end
+    }
+    collector << ")"
+    collector
+  end
+  def visit_ArelExtensions_Nodes_DateDiff o, collector
+    collector << '('
+    collector = visit o.left, collector
+    collector << " - "
+    collector = visit o.right, collector
+    collector << ')'
+    collector
+  end
+  def visit_ArelExtensions_Nodes_Duration o, collector
+    #visit left for period
+    if(o.left == "d")
+      collector << "DAY("
+    elsif(o.left == "m")
+      collector << "MONTH("
+    elsif (o.left == "w")
+      collector << "WEEK"
+    elsif (o.left == "y")
+      collector << "YEAR("
+    end
+    #visit right
+    if(o.right.is_a?(Arel::Attributes::Attribute))
+      collector = visit o.right, collector
+    else
+      collector << "'#{o.right}'"
+    end
+    collector << ")"
+    collector
+  end
+  def visit_ArelExtensions_Nodes_Findis o,collector
+  end
+  def visit_ArelExtensions_Nodes_Length o, collector
+    collector << "LENGTH("
+    collector = visit o.expr, collector
+    collector << ")"
+    collector
+  end
+  def visit_ArelExtensions_Nodes_Locate o, collector
+    collector << "LOCATE("
+    if(o.val.is_a?(Arel::Attributes::Attribute))
+      collector = visit o.val, collector
+    else
+      collector << "'#{o.val}'"
+    end
+    collector << ","
+    collector = visit o.expr, collector
+    collector << ")"
+    collector
+  end
+  def visit_ArelExtensions_Nodes_Isnull o, collector
+    collector << "NVL("
+    collector = visit o.left, collector
+    collector << ","
+    if(o.right.is_a?(Arel::Attributes::Attribute))
+      collector = visit o.right, collector
+    else
+      collector << "'#{o.right}'"
+    end
+    collector << ")"
+    collector
+  end
+  def visit_ArelExtensions_Nodes_Rand o, collector
+    collector << "dbms_random.value("
+    if(o.left != nil && o.right != nil)
+      collector << "'#{o.left}'"
+      collector << ","
+      collector << "'#{o.right}'"
+    end
+    collector << ")"
+    collector
+  end
+  def visit_ArelExtensions_Nodes_Replace o, collector
+    collector << "REPLACE("
+    collector = visit o.expr,collector
+    collector << ","
+    if(o.left.is_a?(Arel::Attributes::Attribute))
+      collector = visit o.left, collector
+    else
+      collector << "'#{o.left}'"
+    end
+    collector << ","
+    if(o.right.is_a?(Arel::Attributes::Attribute))
+      collector = visit o.right, collector
+    else
+      collector << "'#{o.right}'"
+    end
+    collector << ")"
+    collector
+  end
+    def visit_Arel_Nodes_Regexp o, collector
+      collector << " REGEXP_LIKE("
+      collector = visit o.left, collector
+      collector << Arel::Visitors::Oracle::COMMA
+      collector = visit o.right, collector
+      collector << ')'
+      collector
+    end
+    def visit_Arel_Nodes_NotRegexp o, collector
+      collector << " NOT REGEXP_LIKE("
+      collector = visit o.left, collector
+      collector << Arel::Visitors::Oracle::COMMA
+      collector = visit o.right, collector
+      collector << ')'
+      collector
+    end
+    def visit_ArelExtensions_Nodes_Soundex o, collector
+      collector << "SOUNDEX("
+      if((o.expr).is_a?(Arel::Attributes::Attribute))
+        collector = visit o.expr, collector
+      else
+        collector << "'#{o.expr}'"
+      end
+      collector << ")"
+      collector
+    end
+    def visit_ArelExtensions_Nodes_Trim o , collector
+      collector << 'TRIM("BOTH"'
+      collector = visit o.left, collector
+      collector << ","
+      if(o.right.is_a?(Arel::Attributes::Attribute))
+        collector = visit o.right, collector
+      else
+        collector << "'#{o.right}'"
+      end
+      collector << ")"
+      collector
+    end
+    def visit_ArelExtensions_Nodes_Ltrim o , collector
+      collector << 'TRIM("LEADING",'
+      collector = visit o.left, collector
+      collector << ","
+      if(o.right.is_a?(Arel::Attributes::Attribute))
+        collector = visit o.right, collector
+      else
+        collector << "'#{o.right}'"
+      end
+      collector << ")"
+      collector
+    end
+    def visit_ArelExtensions_Nodes_Rtrim o , collector
+      collector << 'TRIM("TRAILING",'
+      collector = visit o.left, collector
+      collector << ","
+      if(o.right.is_a?(Arel::Attributes::Attribute))
+        collector = visit o.right, collector
+      else
+        collector << "'#{o.right}'"
+      end
+      collector << ")"
+      collector
+    end
+      def visit_ArelExtensions_InsertManager_BulkValues o, collector
+        table = collector.value.sub(/\AINSERT INTO/, '')
+        into = " INTO#{table}"
+        collector = Arel::Collectors::SQLString.new
+        collector << "INSERT ALL\n"
+  #      row_nb = o.left.length
+        o.left.each_with_index do |row, idx|
+          collector << "#{into} VALUES ("
+          v = Arel::Nodes::Values.new(row, o.cols)
+          len = v.expressions.length - 1
+          v.expressions.zip(v.columns).each_with_index { |(value, attr), i|
+              case value
+              when Arel::Nodes::SqlLiteral, Arel::Nodes::BindParam
+                collector = visit value, collector
+              else
+                collector << quote(value, attr && column_for(attr)).to_s
+              end
+              collector << Arel::Visitors::Oracle::COMMA unless i == len
+          }
+          collector << ')'
+        end
+        collector << ' SELECT 1 FROM dual'
+        collector
+      end
+    end
+  end
+end