rbhive 0.2.4 → 0.2.5

data/lib/rbhive/connection.rb

@@ -56,6 +56,13 @@ module RBHive
       execute_safe(query)
     end
     
+    def explain(query)
+      safe do
+        execute_unsafe("EXPLAIN "+ query)
+        ExplainResult.new(client.fetchAll)
+      end
+    end
+    
     def priority=(priority)
       set("mapred.job.priority", priority)
     end

data/lib/rbhive/explain_result.rb

@@ -0,0 +1,46 @@
+class ExplainResult
+  def initialize(rows)
+    @rows = rows
+  end
+  
+  def ast
+    by_section[:abstract_syntax_tree].first
+  end
+  
+  def stage_count
+    stage_dependencies.length
+  end
+  
+  def stage_dependencies
+    by_section[:stage_dependencies] || []
+  end
+  
+  def to_tsv
+    @rows.join("\n")
+  end
+  
+  def raw
+    @rows
+  end
+  
+  def to_s
+    to_tsv
+  end
+  
+  private
+  
+  def by_section
+    current_section = nil
+    @rows.inject({}) do |sections, row|
+      if row.match(/^[A-Z]/)
+        current_section = row.chomp(':').downcase.gsub(' ', '_').to_sym
+        sections[current_section] = []
+      elsif row.length == 0
+        next sections
+      else
+        sections[current_section] << row.strip
+      end
+      sections
+    end
+  end
+end

data/lib/rbhive.rb

@@ -1,4 +1,5 @@
 require File.join(File.dirname(__FILE__), 'rbhive', 'connection')
 require File.join(File.dirname(__FILE__), 'rbhive', 'table_schema')
 require File.join(File.dirname(__FILE__), 'rbhive', 'result_set')
+require File.join(File.dirname(__FILE__), 'rbhive', 'explain_result')
 require File.join(File.dirname(__FILE__), 'rbhive', 'schema_definition')

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: rbhive
 version: !ruby/object:Gem::Version 
-  hash: 31
+  hash: 29
   prerelease: 
   segments: 
   - 0
   - 2
-  - 4
-  version: 0.2.4
+  - 5
+  version: 0.2.5
 platform: ruby
 authors: 
 - Forward Internet Group
@@ -47,6 +47,7 @@ files:
 - lib/rbhive/connection.rb
 - lib/rbhive/table_schema.rb
 - lib/rbhive/result_set.rb
+- lib/rbhive/explain_result.rb
 - lib/rbhive/schema_definition.rb
 - lib/thrift/facebook_service.rb
 - lib/thrift/fb303_constants.rb