td-client 0.8.52 → 0.8.53

data/ChangeLog

@@ -1,3 +1,8 @@
+== 2012-07-22 version 0.8.53
+
+* Add normalized_msgpack method to serialize Bignum type
+
+
 == 2012-06-24 version 0.8.52
 
 * Add last_log_timestamp to Table model

data/lib/td/client.rb

@@ -124,7 +124,7 @@ class Client
   def query(db_name, q, result_url=nil, priority=nil, retry_limit=nil, opts={})
     # for compatibility, assume type is hive unless specifically specified
     type = opts[:type] || opts['type'] || :hive
-    raise ArgumentError, "only 'hive' and 'pig' types are supported" unless [:hive, :pig].include? type
+    raise ArgumentError, "The specified query type is not supported: #{type}" unless [:hive, :pig, :impala].include? type
     job_id = @api.query(q, type, db_name, result_url, priority, retry_limit, opts)
     Job.new(self, job_id, type, q)
   end

data/lib/td/client/api.rb

@@ -78,6 +78,16 @@ class API
 
   attr_reader :apikey
 
+  def self.normalized_msgpack(record, out = nil)
+    record.keys.each { |k|
+      v = record[k]
+      if v.kind_of?(Bignum)
+        record[k] = v.to_s
+      end
+    }
+    record.to_msgpack(out)
+  end
+
   def self.validate_database_name(name)
     name = name.to_s
     if name.empty?

data/lib/td/client/version.rb

@@ -1,5 +1,5 @@
 module TreasureData
   class Client
-    VERSION = '0.8.52'
+    VERSION = '0.8.53'
   end
 end

data/spec/spec_helper.rb

@@ -13,6 +13,7 @@ include WebMock::API
 
 $LOAD_PATH << File.dirname(__FILE__)+"../lib"
 require 'td-client'
+require 'msgpack'
 
 include TreasureData
 

data/spec/td/client/api_spec.rb

@@ -24,6 +24,20 @@ describe API do
   }
 
   describe 'normalizer' do
+    it 'normalized_msgpack should convert Bignum into String' do
+      h = {'key' => 1111111111111111111111111111111111}
+      unpacked = MessagePack.unpack(API.normalized_msgpack(h))
+      expect(unpacked['key']).to eq(h['key'].to_s)
+    end
+
+    it 'normalized_msgpack with out argument should convert Bignum into String' do
+      h = {'key' => 1111111111111111111111111111111111}
+      out = ''
+      API.normalized_msgpack(h, out)
+      unpacked = MessagePack.unpack(out)
+      expect(unpacked['key']).to eq(h['key'].to_s)
+    end
+
     it 'normalize_database_name should return normalized data' do
       INVALID_NAMES.each_pair {|ng,ok|
         API.normalize_database_name(ng).should == ok

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: td-client
 version: !ruby/object:Gem::Version
-  version: 0.8.52
+  version: 0.8.53
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-06-24 00:00:00.000000000 Z
+date: 2013-07-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: msgpack