pg_query 0.8.0 → 0.9.0

data/lib/pg_query/truncate.rb

@@ -1,10 +1,12 @@
 class PgQuery
   PossibleTruncation = Struct.new(:location, :node_type, :length, :is_array)
 
+  A_TRUNCATED = 'A_Truncated'.freeze
+
   # Truncates the query string to be below the specified length, first trying to
   # omit less important parts of the query, and only then cutting off the end.
   def truncate(max_length)
-    output = deparse(parsetree)
+    output = deparse(@tree)
 
     # Early exit if we're already below the max length
     return output if output.size <= max_length
@@ -14,12 +16,12 @@ class PgQuery
     # Truncate the deepest possible truncation that is the longest first
     truncations.sort_by! { |t| [-t.location.size, -t.length] }
 
-    tree = deep_dup(parsetree)
+    tree = deep_dup(@tree)
     truncations.each do |truncation|
       next if truncation.length < 3
 
       find_tree_location(tree, truncation.location) do |expr, k|
-        expr[k] = { 'A_TRUNCATED' => nil }
+        expr[k] = { A_TRUNCATED => nil }
         expr[k] = [expr[k]] if truncation.is_array
       end
 
@@ -36,14 +38,14 @@ class PgQuery
   def find_possible_truncations
     truncations = []
 
-    treewalker! parsetree do |_expr, k, v, location|
+    treewalker! @tree do |_expr, k, v, location|
       case k
-      when 'targetList'
-        length = deparse([{ 'SELECT' => { k => v } }]).size - 'SELECT '.size
+      when TARGET_LIST_FIELD
+        length = deparse([{ SELECT_STMT => { k => v } }]).size - 'SELECT '.size
 
-        truncations << PossibleTruncation.new(location, 'targetList', length, true)
+        truncations << PossibleTruncation.new(location, TARGET_LIST_FIELD, length, true)
       when 'whereClause'
-        length = deparse([{ 'SELECT' => { k => v } }]).size
+        length = deparse([{ SELECT_STMT => { k => v } }]).size
 
         truncations << PossibleTruncation.new(location, 'whereClause', length, false)
       when 'ctequery'

data/lib/pg_query/version.rb

@@ -1,3 +1,3 @@
 class PgQuery
-  VERSION = '0.8.0'
+  VERSION = '0.9.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pg_query
 version: !ruby/object:Gem::Version
-  version: 0.8.0
+  version: 0.9.0
 platform: ruby
 authors:
 - Lukas Fittl
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-03-07 00:00:00.000000000 Z
+date: 2016-04-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake-compiler
@@ -98,11 +98,14 @@ files:
 - ext/pg_query/pg_query_ruby.h
 - ext/pg_query/pg_query_ruby.sym
 - lib/pg_query.rb
+- lib/pg_query/deep_dup.rb
 - lib/pg_query/deparse.rb
 - lib/pg_query/deparse/alter_table.rb
 - lib/pg_query/deparse/interval.rb
 - lib/pg_query/filter_columns.rb
 - lib/pg_query/fingerprint.rb
+- lib/pg_query/legacy_parsetree.rb
+- lib/pg_query/node_types.rb
 - lib/pg_query/param_refs.rb
 - lib/pg_query/parse.rb
 - lib/pg_query/parse_error.rb