class Wukong::Load::LoadRunner

Runs the wu-load command.

Public Instance Methods

data_store_name() click to toggle source

The name of the data store

@return [String]

# File lib/wukong-load/load_runner.rb, line 46
def data_store_name
  args.first
end
processor() click to toggle source

The name of the processor that should handle the data store

@return [String]

# File lib/wukong-load/load_runner.rb, line 53
def processor
  case data_store_name
  when 'elasticsearch'   then :elasticsearch_loader
  when 'kafka'           then :kafka_loader
  when 'mongo','mongodb' then :mongodb_loader
  when 'sql', 'mysql'    then :sql_loader
  end
end
validate() click to toggle source

Ensure that we were passed a data store name that we know about.

@raise [Wukong::Error] if the data store is missing or unknown @return [true]

# File lib/wukong-load/load_runner.rb, line 33
def validate
  case
  when data_store_name.nil?
    raise Error.new("Must provide the name of a data store as the first argument")
  when processor.nil?
    raise Error.new("No loader defined for data store <#{data_store_name}>")
  end
  true
end