rbhive 0.1.4 → 0.1.5

data/lib/rbhive/connection.rb

@@ -24,6 +24,7 @@ module RBHive
   
   class Connection
     attr_reader :client
+    
     def initialize(server, port=10_000, logger=StdOutLogger.new)
       @socket = Thrift::Socket.new(server, port)
       @transport = Thrift::BufferedTransport.new(@socket)
@@ -59,6 +60,19 @@ module RBHive
       ResultSet.new([client.fetchOne])
     end
     
+    def create_table(schema)
+      execute(schema.create_table_statement)
+    end
+    
+    def drop_table(name)
+      name = name.name if name.is_a?(TableSchema)
+      execute("DROP TABLE `#{name}`")
+    end
+    
+    def replace_columns(schema)
+      execute(schema.replace_columns_statement)
+    end
+    
     def method_missing(meth, *args)
       client.send(meth, *args)
     end

data/lib/rbhive/schema.rb

@@ -0,0 +1,63 @@
+class TableSchema
+  attr_accessor :name
+  attr_reader :columns, :partitions
+  def initialize(name, comment=nil, field_sep='\t', line_sep='\n', &blk)
+    @name, @comment, @field_sep, @line_sep = name, comment, field_sep, line_sep
+    @columns = []
+    @partitions = []
+    instance_eval(&blk) if blk
+  end
+  
+  def column(name, type, comment=nil)
+    @columns << Column.new(name, type, comment)
+  end
+  
+  def partition(name, type, comment=nil)
+    @partitions << Column.new(name, type, comment)
+  end
+  
+  def create_table_statement()
+    %[CREATE TABLE #{table_statement}
+ROW FORMAT DELIMITED
+FIELDS TERMINATED BY '#{@field_sep}'
+LINES TERMINATED BY '#{@line_sep}'
+STORED AS TEXTFILE]
+  end
+  
+  def replace_columns_statement()
+    %[ALTER TABLE `#{name}` REPLACE COLUMNS #{column_statement}]
+  end
+  
+  def to_s
+    table_statement
+  end
+  
+  private
+  
+  def table_statement
+    comment_string = (@comment.nil? ? '' : " COMMENT '#{@comment}'")
+    %[`#{@name}` #{column_statement}#{comment_string}\n#{partition_statement}]
+  end
+  
+  def column_statement
+    cols = @columns.join(",\n")
+    "(\n#{cols}\n)"
+  end
+  
+  def partition_statement
+    cols = @partitions.join(",\n")
+    "PARTITIONED BY (\n#{cols}\n)"
+  end
+  
+  class Column
+    attr_reader :name, :type, :comment
+    def initialize(name, type, comment=nil)
+      @name, @type, @comment = name, type, comment
+    end
+    
+    def to_s
+      comment_string = @comment.nil? ? '' : " COMMENT '#{@comment}'"
+      "`#{@name}` #{@type.to_s.upcase}#{comment_string}"
+    end
+  end
+end

data/lib/rbhive.rb

@@ -1 +1,2 @@
-require File.join(File.dirname(__FILE__), 'rbhive', 'connection')
+require File.join(File.dirname(__FILE__), 'rbhive', 'connection')
+require File.join(File.dirname(__FILE__), 'rbhive', 'schema')

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: rbhive
 version: !ruby/object:Gem::Version 
-  hash: 19
+  hash: 17
   prerelease: false
   segments: 
   - 0
   - 1
-  - 4
-  version: 0.1.4
+  - 5
+  version: 0.1.5
 platform: ruby
 authors: 
 - Forward Internet Group
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-09-29 00:00:00 +01:00
+date: 2010-10-01 00:00:00 +01:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -45,6 +45,7 @@ extra_rdoc_files: []
 files: 
 - lib/rbhive.rb
 - lib/rbhive/connection.rb
+- lib/rbhive/schema.rb
 - lib/thrift/facebook_service.rb
 - lib/thrift/fb303_constants.rb
 - lib/thrift/fb303_types.rb