forward-rbase 0.1.0 → 0.2.0

data/lib/rbase/client.rb

@@ -16,6 +16,10 @@ module Rbase
       client.getTableNames
     end
     
+    def table_exists?(table_name)
+      table_names.include?(table_name)
+    end
+    
     def create_table(table_name, *column_family_names)
       column_families = column_family_names.map do |family_name|
         Apache::Hadoop::Hbase::Thrift::ColumnDescriptor.new(:name => family_name)

data/lib/rbase/table.rb

@@ -6,16 +6,13 @@ module Rbase
     end
     
     def find(row)
-      @client.getRow(@table_name,row).map do |row|
-        ret_val = {}
-        row.columns.each do |column,val|
-          family, key = *column.split(":")
-          ret_val[family] ||= {}
-          ret_val[family][key] = val.value
-        end
-        ret_val
-      end
+      @client.getRow(@table_name,row).map { |row| row_to_hash(row) }
+    end
+    
+    def first(row)
+      find(row).first
     end
+    alias_method :[], :first
     
     def insert(row, hash)
       hash.each do |family,value|
@@ -25,5 +22,37 @@ module Rbase
         @client.mutateRow(@table_name,row,mutations)
       end
     end
+    alias_method :[]=, :insert
+    
+    def each_batch(batch_size=100, columns=[])
+      columns = [columns] unless columns.is_a?(Array)
+      scanner_id = @client.scannerOpen(@table_name, '', columns)
+      loop do
+        results = @client.scannerGetList(scanner_id, batch_size)
+        break if results.length == 0
+        yield results.map {|r| row_to_hash(r) }
+      end
+      @client.scannerClose(scanner_id)
+    end
+    
+    def each(batch_size=100, columns=[])
+      each_batch(batch_size, columns) do |results|
+        results.each do |result|
+          yield(result)
+        end
+      end
+    end
+    
+    private
+    
+    def row_to_hash(row)
+      ret_val = {}
+      row.columns.each do |column,val|
+        family, key = *column.split(":")
+        ret_val[family] ||= {}
+        ret_val[family][key] = val.value
+      end
+      ret_val
+    end
   end
 end

data/lib/rbase/thrift/hbase.rb

@@ -2043,9 +2043,8 @@ require File.join(File.dirname(__FILE__), 'hbase_types')
                               IO => {:type => ::Thrift::Types::STRUCT, :name => 'io', :class => Apache::Hadoop::Hbase::Thrift::IOError}
                             }
 
-                            def struct_fields; FIELDS; __END__
+                            def struct_fields; FIELDS; end
                             
-
                             def validate
                             end
 
@@ -2308,5 +2307,4 @@ require File.join(File.dirname(__FILE__), 'hbase_types')
                       end
                     end
                   end
-                end
-              end
+                end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: forward-rbase
 version: !ruby/object:Gem::Version 
-  hash: 27
+  hash: 23
   prerelease: false
   segments: 
   - 0
-  - 1
+  - 2
   - 0
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Forward Internet Group