forward-rbase 0.1.0

data/lib/rbase/client.rb

@@ -0,0 +1,37 @@
+module Rbase
+  class Client
+    attr_reader :client
+    
+    def initialize(server,port=9090)
+      socket = Thrift::Socket.new(server, port.to_s)
+      transport = Thrift::BufferedTransport.new(socket)
+      transport.open
+
+      protocol = Thrift::BinaryProtocol.new(transport)
+      @tables = {}
+      @client = Apache::Hadoop::Hbase::Thrift::Hbase::Client.new(protocol)
+    end
+    
+    def table_names
+      client.getTableNames
+    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)
+      end
+      client.createTable(table_name, column_families)
+    end
+    
+    def delete_table(table_name)
+      @client.disableTable(table_name)
+      @client.deleteTable(table_name)
+    end
+    
+    def [](table_name)
+      table_name = table_name.to_sym
+      @tables[table_name] ||= Rbase::Table.new(client, table_name)
+      @tables[table_name]
+    end
+  end
+end

data/lib/rbase/table.rb

@@ -0,0 +1,29 @@
+module Rbase
+  class Table
+    def initialize(client, table_name)
+      @client = client
+      @table_name = table_name.to_s
+    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
+    end
+    
+    def insert(row, hash)
+      hash.each do |family,value|
+        mutations = value.map do |column,val|
+          Apache::Hadoop::Hbase::Thrift::Mutation.new(:column => "#{family}:#{column}", :value => val)
+        end
+        @client.mutateRow(@table_name,row,mutations)
+      end
+    end
+  end
+end