impala 0.3.1 → 0.4.0

data/lib/impala.rb

@@ -14,7 +14,6 @@ require 'impala/cursor'
 require 'impala/connection'
 
 module Impala
-  KNOWN_COMMANDS = ['select', 'insert', 'show', 'describe', 'use', 'explain', 'create', 'drop', 'invalidate', 'with']
   DEFAULT_HOST = 'localhost'
   DEFAULT_PORT = 21000
   class InvalidQueryError < StandardError; end

data/lib/impala/connection.rb

@@ -84,10 +84,6 @@ module Impala
       raise InvalidQueryError.new("Empty query") if words.empty?
 
       command = words.first.downcase
-      if !KNOWN_COMMANDS.include?(command)
-        raise InvalidQueryError.new("Unrecognized command: '#{words.first}'")
-      end
-
       ([command] + words[1..-1]).join(' ')
     end
 

data/lib/impala/cursor.rb

@@ -70,6 +70,10 @@ module Impala
       !@done || !@row_buffer.empty?
     end
 
+    def runtime_profile
+      @service.GetRuntimeProfile(@handle)
+    end
+
     private
 
     def metadata
@@ -121,7 +125,7 @@ module Impala
         else
           raise ParsingError.new("Invalid value for boolean: #{value}")
         end
-      when 'tinyint', 'int', 'bigint'
+      when 'tinyint', 'smallint', 'int', 'bigint'
         value.to_i
       when 'double', 'float'
         value.to_f

data/lib/impala/version.rb

@@ -1,3 +1,3 @@
 module Impala
-  VERSION = "0.3.1"
+  VERSION = "0.4.0"
 end

data/test/test_impala.rb

@@ -42,7 +42,6 @@ describe Impala::Connection do
 
     it 'should reject empty or invalid queries' do
       assert_raises(Impala::InvalidQueryError) { @connection.send(:sanitize_query, '')}
-      assert_raises(Impala::InvalidQueryError) { @connection.send(:sanitize_query, 'HERRO herro herro')}
     end
   end
 

data/test/test_impala_connected.rb

@@ -59,6 +59,11 @@ describe 'connected tests' do
       assert_equal([{:foo=>1.23}], ret, "the result should be a float")
     end
 
+    it 'can handle smallint values' do
+      ret = @connection.query('SELECT CAST(123 AS smallint) AS foo')
+      assert_equal([{:foo=>123}], ret, "the result should be an integer")
+    end
+
     it 'can handle float values' do
       ret = @connection.query('SELECT CAST(1.23 AS float) as foo')
       assert_instance_of(Float, ret.first[:foo], "the result should be a float")
@@ -77,6 +82,12 @@ describe 'connected tests' do
     it 'can successfully refresh the metadata store' do
       ret = @connection.refresh
     end
+
+    it 'can get the runtime profile from a cursor' do
+      cursor = @connection.execute('SELECT NOW() as foo')
+      ret = cursor.runtime_profile
+      assert_instance_of(String, ret, "the result should be a string")
+    end
   end
 
   describe 'with a test database' do
@@ -97,6 +108,7 @@ describe 'connected tests' do
     describe 'and a test table' do
       before do
         @table = "#{@database}.foobar"
+        @connection.query("DROP TABLE IF EXISTS #{@table}")
         @connection.query("CREATE TABLE #{@table} (i INT)")
       end
 
@@ -177,7 +189,11 @@ describe 'connected tests' do
         it 'can handle interspersed NULL values' do
           @connection.query("INSERT INTO #{@table} (i) SELECT NULL")
           res = @connection.query("SELECT * FROM #{@table} ORDER BY i DESC LIMIT 4")
-          assert_equal([{:i => 1}, {:i => 1}, {:i => 1}, {:i => nil}], res)
+          assert_equal([{:i => nil}, {:i => 1}, {:i => 1}, {:i => 1}], res)
+        end
+
+        it 'can alter table add column' do
+          @connection.query("ALTER TABLE #{@table} ADD COLUMNS (new INT)")
         end
       end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: impala
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.4.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-02-10 00:00:00.000000000 Z
+date: 2014-04-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thrift