jruby_hive 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 6e6054c6faf5462d63e5b9f20f93440b0f786039
-  data.tar.gz: 2fec05149e6eb01d969d7254ad3ab27a0637819d
+  metadata.gz: 9f82466466b7be12a54b7847f84a1d3c442387f4
+  data.tar.gz: 5f8481d21b8d416f19a29a186eef7e6d511470b8
 SHA512:
-  metadata.gz: 16dc4ca2f8271bafe1d94629505575fba3451d2060f5446521dacc7b8e78ca4b01da3ad42e7e81bf2af124f810b84988f83fa19464676a32be2afef670b206de
-  data.tar.gz: fc8dcafab592d79b0003fd12d1d35efe5d3326c822593e4922b3bc7eef1e19940703f4f733ec393c3c719c5b7944560409afc5d7a6e3cad985990cf4eb24ce0c
+  metadata.gz: 2b7a5837bcbc0a10b8b2be9a7a64607260e7d2c10411257f628149f51e13232d5cb4c0f70ce3d3d9570e952db334fcfc9fa53c2e99e85aac77ad2f2b2d34bf9d
+  data.tar.gz: 726d81dc574a83d7a4f60caeb1c1a16f5931976b08cc32d33fef64eb3982aef92030063574d7935c99bb9fd5092ba303ded413d829d371327774fb976fa88357

data/jruby_hive.gemspec

@@ -4,7 +4,7 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 
 Gem::Specification.new do |spec|
   spec.name          = "jruby_hive"
-  spec.version       = "0.0.2"
+  spec.version       = "0.0.3"
   spec.authors       = ["Shinji Ikeda"]
   spec.email         = ["gm.ikeda@gmail.com"]
   spec.summary       = %q{jruby hive api}

data/lib/jruby_hive.rb

@@ -36,24 +36,45 @@ class Hive
     
   end
   
-  def initialize
+  def initialize(db="default")
+    @db = db
     @hconf = HiveConf.new
     
     @driver = Driver.new(@hconf)
     SessionState.start(CliSessionState.new(@hconf))
+    @driver.run(db)
   end
   
-  def run(cmd)
+  def use(db)
+    @db = db
+    @driver.run(@db)
+  end
+  
+  def run(cmd, max_rows=10000)
     r = @driver.run(cmd)
-    if r.getResponseCode != 0
-      raise HiveError, "hive error: " << r.getErrorMessage
-    end
-    
-    list = ArrayList.new
-    if @driver.getResults(list)
-      return list.to_ary
+    begin
+      if r.getResponseCode != 0
+        raise HiveError, "hive error: " << r.getErrorMessage
+      end
+      if block_given?
+        list = ArrayList.new
+        while @driver.getResults(list)
+          list.each do | n |
+            yield n
+          end
+          list.clear
+        end
+      else
+        @driver.setMaxRows(max_rows)
+        list = ArrayList.new
+        if @driver.getResults(list)
+          return list.to_ary
+        end
+        return nil
+      end
+    ensure
+      @driver.destroy
     end
-    return nil
   end
   
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: jruby_hive
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Shinji Ikeda