Contents

class Cranium::ExternalTable

  def initialize(source, db_connection)
    @source, @connection = source, db_connection
  end



  def create
    @connection.run <<-sql
      CREATE EXTERNAL TABLE "#{name}" (
          #{field_definitions}
      )
      LOCATION (#{external_location})
      FORMAT 'CSV' (DELIMITER '#{quote @source.delimiter}' ESCAPE '#{quote @source.escape}' QUOTE '#{quote @source.quote}' HEADER)
      ENCODING 'UTF8'
    sql
  end



  def destroy
    @connection.run %Q[DROP EXTERNAL TABLE "#{name}"]
  end



  def name
    :"external_#{@source.name}"
  end



  private

  def field_definitions
    @source.fields.map do |name, type|
      %Q("#{name}" #{sql_type_for_ruby_type(type)})
    end.join ",\n          "
  end



  def sql_type_for_ruby_type(type)
    case type.to_s
      when "Integer" then
        "INTEGER"
      when "Float" then
        "NUMERIC"
      when "Date" then
        "DATE"
      when "Time" then
        "TIMESTAMP WITHOUT TIME ZONE"
      when "TrueClass", "FalseClass" then
        "BOOLEAN"
      else
        "TEXT"
    end
  end



  def quote(text)
    text.gsub "'", "''"
  end



  def external_location
    @source.files.map do |file_name|
      "'gpfdist://#{Cranium.configuration.gpfdist_url}/#{Cranium.configuration.upload_directory}/#{file_name}'"
    end.join(', ')
  end

end

Version data entries

9 entries across 9 versions & 1 rubygems

Version	Path
cranium-0.5	lib/cranium/external_table.rb
cranium-0.4.3	lib/cranium/external_table.rb
cranium-0.4.2	lib/cranium/external_table.rb
cranium-0.4.1	lib/cranium/external_table.rb
cranium-0.4	lib/cranium/external_table.rb
cranium-0.3.1	lib/cranium/external_table.rb
cranium-0.3.0	lib/cranium/external_table.rb
cranium-0.2.1	lib/cranium/external_table.rb
cranium-0.2.0	lib/cranium/external_table.rb