hbaserb 0.0.1

data/lib/hbaserb/client.rb

@@ -0,0 +1,37 @@
+module HBaseRb
+  
+  class Client
+    def initialize(server, port=9090)
+      socket = Thrift::Socket.new(server, port)
+      @transport = Thrift::BufferedTransport.new(socket)
+      @transport.open
+      protocol = Thrift::BinaryProtocol.new(@transport)
+      @client = Apache::Hadoop::Hbase::Thrift::Hbase::Client.new(protocol)
+    end
+
+    def table_names
+      @client.getTableNames
+    end
+
+    def has_table?(name)
+      table_names.include? name
+    end
+
+    def get_table(tablename)
+      raise HBaseRb::NoSuchTable if not has_table? tablename
+      HBaseRb::Table.new @client, tablename
+    end
+
+    def create_table(tablename, *column_family_names)
+      column_family_names.map! { |name| Apache::Hadoop::Hbase::Thrift::ColumnDescriptor.new(:name => name) }
+      @client.createTable tablename, column_family_names
+      get_table tablename
+    end
+
+    def close
+      @transport.close
+    end
+
+  end
+
+end

data/lib/hbaserb/exceptions.rb

@@ -0,0 +1,3 @@
+module HBaseRb
+  class NoSuchTable < StandardError; end
+end

data/lib/hbaserb/extensions.rb

@@ -0,0 +1,39 @@
+module Apache
+  module Hadoop
+    module Hbase
+      module Thrift
+
+
+        class TCell
+          def to_i16
+            val, = value.unpack('n')
+            (val > 0x7fff) ? (0 - ((val - 1) ^ 0xffff)) : val
+          end
+
+          def to_i32
+            val, = value.unpack('N')
+            (val > 0x7fffffff) ? (0 - ((val - 1) ^ 0xffffffff)) : val
+          end
+          
+          def to_i64
+            hi, lo = value.unpack('N2')
+            if (hi > 0x7fffffff)
+              hi ^= 0xffffffff
+              lo ^= 0xffffffff
+              0 - (hi << 32) - lo - 1
+            else
+              (hi << 32) + lo
+            end
+          end
+          
+          def to_double
+            value.unpack('G').first
+          end
+          
+        end
+
+
+      end
+    end
+  end
+end

data/lib/hbaserb/scanner.rb

@@ -0,0 +1,39 @@
+module HBaseRb
+  
+  class Scanner
+    def initialize(client, scanner_id)
+      @client = client
+      @sid = scanner_id
+      if block_given?
+        n = next_row
+        while n.length > 0
+          yield n.first
+          n = next_row
+        end        
+        close
+      end
+    end
+
+    def next_row
+      call :scannerGet
+    end
+
+    def close
+      call :scannerClose
+    end
+
+    def each 
+      n = next_row
+      while n.length > 0
+        yield n.first
+        n = next_row
+      end
+    end
+
+    private
+    def call(method, *args)
+      @client.send method, @sid, *args
+    end
+  end
+
+end

data/lib/hbaserb/table.rb

@@ -0,0 +1,78 @@
+module HBaseRb
+  
+  class Table
+    def initialize(client, name)
+      @client = client
+      @name = name
+    end
+
+    def column_families
+      call :getColumnDescriptors
+    end
+    
+    def regions
+      call :getTableRegions
+    end
+
+    def get(row, column)
+      call :get, row.to_s, column
+    end
+
+    # get the last value for the given row / column
+    def get_last(row, column, default=nil)
+      r = get(row, column)
+      (r.length > 0) ? r.first.value : default
+    end
+
+    def get_row(row)
+      call :getRow, row.to_s
+    end
+    
+    def delete
+      call :disableTable
+      call :deleteTable
+    end
+
+    def delete_row(row)
+      call :deleteAllRow, row.to_s
+    end
+
+    def delete_cells(row, column)
+      call :deleteAll, row.to_s, column
+    end
+
+    def atomic_increment(row, column, value=1)
+      call :atomicIncrement, row.to_s, column, value
+    end
+
+    # pass in no params to scan whole table
+    def create_scanner(row=nil, *columns, &block)
+      row ||= ""
+      columns = (columns.length > 0) ? columns : column_families.keys 
+      sid = call :scannerOpen, row.to_s, columns
+      Scanner.new @client, sid, &block
+    end
+
+    # mutations is a key / value pair to insert / update for the given row
+    # the keys are in the form "family:column"
+    def mutate_row(row, mutations)
+      mutations = mutations.map { |k,v| Apache::Hadoop::Hbase::Thrift::Mutation.new(:column => k, :value => v) }
+      call :mutateRow, row, mutations
+    end
+
+    def to_s
+      s = ""
+      create_scanner { |r|
+        cols = r.columns.map { |k,v| "#{k}: #{v.value}" }.join(", ")
+        s += "#{r.row}: #{cols}\n"
+      }
+      s
+    end
+
+    private
+    def call(method, *args)
+      @client.send method, @name, *args
+    end
+  end
+
+end

data/lib/hbaserb.rb

@@ -0,0 +1,10 @@
+require 'hbaserb/client'
+require 'hbaserb/exceptions'
+require 'hbaserb/table'
+require 'hbaserb/scanner'
+require 'hbaserb/extensions'
+
+$:.unshift File.join(File.dirname(__FILE__), 'thrift')
+require 'hbase'
+require 'hbase_constants'
+