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/postgresql.rb ADDED Viewed

@@ -0,0 +1,258 @@
+module ArelExtensions
+  module Visitors
+    Arel::Visitors::PostgreSQL.class_eval do
+      def visit_ArelExtensions_Nodes_Concat o, collector
+        arg = o.left.relation.engine.columns.find{|c| c.name == o.left.name.to_s}.type
+        collector = visit o.left, collector
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector << '||'
+          collector = visit o.right, collector
+        elsif ( arg == :date || arg == :datetime)
+          collector << " + INTERVAL '#{o.right} day'"
+          collector
+        else
+          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
+          if(a.is_a?(Integer))
+            collector << "#{a}"
+          else
+            collector << "'#{a}'"
+          end
+        end
+      }
+      collector << ")"
+      collector
+    end
+      def visit_ArelExtensions_Nodes_DateDiff o, collector
+        collector = visit o.left, collector
+        collector << " -"
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.left, collector
+        else
+          collector << " date '#{o.right}'"
+        end
+        collector
+      end
+    def visit_ArelExtensions_Nodes_Duration o, collector
+      #visit left for period
+      if(o.left == "d")
+        collector << "EXTRACT(DAY FROM"
+      elsif(o.left == "m")
+        collector << "EXTRACT(MONTH FROM "
+      elsif (o.left == "w")
+        collector << "EXTRACT(WEEK FROM"
+      elsif (o.left == "y")
+        collector << "EXTRACT(YEAR FROM"
+      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_Floor o, collector
+      collector << "FLOOR("
+     if((o.expr).is_a?(Arel::Attributes::Attribute))
+      collector = visit o.expr, collector
+     end
+      collector << ")"
+      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 << "position("
+      if(o.val.is_a?(Arel::Attributes::Attribute))
+        collector = visit o.val, collector
+      else
+        collector << "'#{o.val}'"
+      end
+      collector << " IN "
+      collector = visit o.expr, collector
+      collector << ")"
+      collector
+    end
+    def visit_ArelExtensions_Nodes_Isnull o, collector
+      collector << "coalesce("
+      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 << "random("
+       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
+    remove_method :visit_Arel_Nodes_Regexp
+    def visit_Arel_Nodes_Regexp o, collector
+      collector = visit o.left, collector
+      collector << " ~ "
+      collector << "'#{o.right}'"
+      collector
+    end
+    remove_method :visit_Arel_Nodes_NotRegexp
+    def visit_Arel_Nodes_NotRegexp o, collector
+      collector = visit o.left, collector
+      collector << " !~ "
+      collector << "'#{o.right}'"
+      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_Sum o, collector
+      collector << "sum("
+     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 << "date_part('dow', "
+        if((o.date).is_a?(Arel::Attributes::Attribute))
+          collector = visit o.date, collector
+        else
+          collector << "'#{o.date}'"
+        end
+        collector << ")"
+        collector
+      end
+    end
+  end
+end

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

@@ -0,0 +1,218 @@
+module ArelExtensions
+  module Visitors
+    Arel::Visitors::SQLite.class_eval do
+      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
+      # Date operations
+      def visit_ArelExtensions_Nodes_DateAdd o, collector
+        collector << "date("
+        collector = visit o.expressions.first, collector
+        collector << Arel::Visitors::ToSql::COMMA
+        collector = visit o.sqlite_value, collector
+        collector << ")"
+        collector
+      end
+      def visit_ArelExtensions_Nodes_DateDiff o, collector
+        collector << " julianday("
+        collector = visit o.left, collector
+        collector << ")- julianday("
+        collector = visit o.right, collector
+        collector << ")"
+        collector
+      end
+      def visit_ArelExtensions_Nodes_Duration o, collector
+        #visit left for period
+        if(o.left == "d")
+          collector << "strftime('%d',"
+        elsif(o.left == "m")
+          collector << "strftime('%m',"
+        elsif (o.left == "w")
+          collector << "strftime('%W',"
+        elsif (o.left == "y")
+          collector << "strftime('%Y',"
+        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_Locate o, collector
+        collector << "instr("
+        collector = visit o.expr, collector
+        collector << ","
+        if(o.val.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.val, collector
+        else
+          collector << "'#{o.val}'"
+        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_Rand o, collector
+        collector << "RANDOM("
+        if(o.left != nil && o.right != nil)
+          if(o.left.is_a?(Arel::Attributes::Attribute))
+            collector = visit o.left, collector
+          else
+            collector << "'#{o.left}'"
+          end
+          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 = visit o.left, collector
+        collector << " REGEXP '#{o.right}'"
+        collector
+      end
+      def visit_Arel_Nodes_NotRegexp o, collector
+        collector = visit o.left, collector
+        collector << " NOT REGEXP "
+        collector << "'#{o.right}'"
+        collector
+      end
+      def visit_ArelExtensions_Nodes_Round o, collector
+        collector << "ROUND("
+        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_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("
+        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 << "LTRIM("
+        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 << "RTRIM("
+        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_Wday o, collector
+        collector << "strftime('%w',"
+        if((o.date).is_a?(Arel::Attributes::Attribute))
+          collector = visit o.date, collector
+        else
+          collector << "'#{o.date}'"
+        end
+        collector << ")"
+        collector
+      end
+    end
+  end
+end