forward-rbase 0.2.3 → 0.2.7

data/lib/rbase.rb

@@ -1,6 +1,10 @@
+# encoding: utf-8
+
 require File.join(File.dirname(__FILE__), 'rbase', 'thrift', 'hbase')
 require File.join(File.dirname(__FILE__), 'rbase', 'thrift', 'hbase_constants')
 require File.join(File.dirname(__FILE__), 'rbase', 'thrift', 'hbase_types')
 
+require File.join(File.dirname(__FILE__), 'rbase', 'thrift', 'hbase_extension')
+
 require File.join(File.dirname(__FILE__), 'rbase', 'client')
 require File.join(File.dirname(__FILE__), 'rbase', 'table')

data/lib/rbase/client.rb

@@ -44,7 +44,9 @@ module Rbase
     end
     
     def [](table_name)
-      @tables[table_name] ||= Rbase::Table.new(@client, table_name) if table_exists?(table_name)
+      @tables[table_name] ||= begin
+        Rbase::Table.new(@client, table_name) if table_exists?(table_name)
+      end
       @tables[table_name]
     end
   end

data/lib/rbase/table.rb

@@ -4,16 +4,16 @@ module Rbase
       @client = client
       @table_name = table_name.to_s
     end
-    
+
     def find(row)
       @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|
         mutations = value.map do |column,val|
@@ -23,7 +23,7 @@ module Rbase
       end
     end
     alias_method :[]=, :insert
-    
+
     def multi_insert(rows)
       row_mutations = rows.map do |row, hash|
         hash.map do |family,value|
@@ -35,7 +35,7 @@ module Rbase
       end
       @client.mutateRows(@table_name, row_mutations.flatten)
     end
-    
+
     def each_batch(batch_size=100, columns=[])
       columns = [columns] unless columns.is_a?(Array)
       scanner_id = @client.scannerOpen(@table_name, '', columns)
@@ -46,7 +46,7 @@ module Rbase
       end
       @client.scannerClose(scanner_id)
     end
-    
+
     def each(batch_size=100, columns=[])
       each_batch(batch_size, columns) do |results|
         results.each do |result|
@@ -54,13 +54,21 @@ module Rbase
         end
       end
     end
+
+    def set_auto_flush(enabled)
+      @client.setAutoFlush(@table_name, enabled)
+    end
     
+    def increment(row, family, column, value)
+      @client.atomicIncrement(@table_name, row, "#{family}:#{column}", value)
+    end
+
     private
-    
+
     def row_to_hash(row)
       ret_val = {}
       row.columns.each do |column,val|
-        family, key = *column.split(":")
+        family, key = *column.split(":", 2)
         ret_val[family] ||= {}
         ret_val[family][key] = val.value
       end

data/lib/rbase/thrift/hbase_extension.rb

@@ -0,0 +1,69 @@
+module Apache
+  module Hadoop
+    module Hbase
+      module Thrift
+        module Hbase
+          class Client
+            def setAutoFlush(tableName, enabled)
+              send_setAutoFlush(tableName, enabled)
+              return recv_setAutoFlush()
+            end
+
+            def send_setAutoFlush(tableName, enabled)
+              send_message('setAutoFlush', SetAutoFlush_args, :tableName => tableName, :enabled => enabled)
+            end
+
+            def recv_setAutoFlush()
+              receive_message(SetAutoFlush_result)
+            end
+          end
+
+          class SetAutoFlush_args
+            include ::Thrift::Struct, ::Thrift::Struct_Union
+            TABLENAME = 1
+            ENABLED = 2
+
+            FIELDS = {
+              TABLENAME => {:type => ::Thrift::Types::STRING, :name => 'tableName', :binary => true},
+              ENABLED => {:type => ::Thrift::Types::BOOL, :name => 'enabled'}
+            }
+
+            def struct_fields; FIELDS; end
+
+            def validate
+            end
+
+            ::Thrift::Struct.generate_accessors self
+          end
+
+          class SetAutoFlush_result
+            include ::Thrift::Struct, ::Thrift::Struct_Union
+
+            FIELDS = {}
+
+            def struct_fields; FIELDS; end
+
+            def validate
+            end
+
+            ::Thrift::Struct.generate_accessors self
+          end
+          
+          class Processor
+            def process_setAutoFlush(seqid, iprot, oprot)
+              args = read_args(iprot, SetAutoFlush_args)
+              result = SetAutoFlush_result.new()
+              begin
+                result.success = @handler.setAutoFlush()
+              rescue Apache::Hadoop::Hbase::Thrift::IOError => io
+                result.io = io
+              end
+              write_result(result, oprot, 'setAutoFlush', seqid)
+            end
+            
+          end
+        end
+      end
+    end
+  end
+end

metadata

@@ -1,83 +1,62 @@
---- !ruby/object:Gem::Specification 
+--- !ruby/object:Gem::Specification
 name: forward-rbase
-version: !ruby/object:Gem::Version 
-  prerelease: false
-  segments: 
-  - 0
-  - 2
-  - 3
-  version: 0.2.3
+version: !ruby/object:Gem::Version
+  version: 0.2.7
+  prerelease: 
 platform: ruby
-authors: 
+authors:
 - Forward Internet Group
 autorequire: 
 bindir: bin
 cert_chain: []
-
-date: 2011-02-17 00:00:00 +00:00
-default_executable: 
-dependencies: 
-- !ruby/object:Gem::Dependency 
+date: 2011-02-17 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
   name: thrift
-  prerelease: false
-  requirement: &id001 !ruby/object:Gem::Requirement 
+  requirement: &70106085781280 !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        segments: 
-        - 0
-        - 4
-        - 0
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
         version: 0.4.0
   type: :runtime
-  version_requirements: *id001
+  prerelease: false
+  version_requirements: *70106085781280
 description: Simple lib for executing hbase lookups
 email: andy@forward.co.uk
 executables: []
-
 extensions: []
-
 extra_rdoc_files: []
-
-files: 
+files:
 - lib/rbase.rb
 - lib/rbase/client.rb
 - lib/rbase/table.rb
 - lib/rbase/thrift/hbase.rb
 - lib/rbase/thrift/hbase_constants.rb
 - lib/rbase/thrift/hbase_types.rb
-has_rdoc: true
+- lib/rbase/thrift/hbase_extension.rb
 homepage: http://github.com/forward/rbase
 licenses: []
-
 post_install_message: 
 rdoc_options: []
-
-require_paths: 
+require_paths:
 - lib
-required_ruby_version: !ruby/object:Gem::Requirement 
+required_ruby_version: !ruby/object:Gem::Requirement
   none: false
-  requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      segments: 
-      - 0
-      version: "0"
-required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
-  requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      segments: 
-      - 0
-      version: "0"
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
 requirements: []
-
 rubyforge_project: 
-rubygems_version: 1.3.7
+rubygems_version: 1.8.11
 signing_key: 
 specification_version: 3
 summary: Simple lib for executing hbase lookups
 test_files: []
-