RubyGems - arel_extensions - Versions diffs - 0.8.0 - Mend

arel_extensions 0.8.0

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