Use escape_literal more places

Author: ankane

lib/pgslice/cli/add_partitions.rb

@@ -27,7 +27,7 @@ def add_partitions(table)
       queries = []
 
       if needs_comment
-        queries << "COMMENT ON TRIGGER #{quote_ident(trigger_name)} ON #{quote_table(table)} IS 'column:#{field},period:#{period},cast:#{cast}';"
+        queries << "COMMENT ON TRIGGER #{quote_ident(trigger_name)} ON #{quote_table(table)} IS #{escape_literal("column:#{field},period:#{period},cast:#{cast}")};"
       end
 
       # today = utc date

lib/pgslice/cli/fill.rb

@@ -74,7 +74,7 @@ def fill(table)
       end
 
       while starting_id < max_source_id
-        where = "#{quote_ident(primary_key)} > #{starting_id} AND #{quote_ident(primary_key)} <= #{starting_id + batch_size}"
+        where = "#{quote_ident(primary_key)} > #{escape_literal(starting_id)} AND #{quote_ident(primary_key)} <= #{escape_literal(starting_id + batch_size)}"
         if starting_time
           where << " AND #{quote_ident(field)} >= #{sql_date(starting_time, cast)} AND #{quote_ident(field)} < #{sql_date(ending_time, cast)}"
         end

lib/pgslice/helpers.rb

@@ -158,7 +158,11 @@ def quote_ident(value)
     end
 
     def escape_literal(value)
-      connection.escape_literal(value)
+      if value.is_a?(Integer)
+        value
+      else
+        connection.escape_literal(value)
+      end
     end
 
     def quote_table(table)