Contents

class Both
  include FromHash
  fattr(:objs) { [] }
  def method_missing(sym,*args,&b)
    objs.each { |x| x.send(sym,*args,&b) }
  end
end

class TableLoaderDSL
  fattr(:table_name) { loader.table_name }
  fattr(:loader) { TableLoader.new }
  fattr(:manager) { TableManager.new }
  fattr(:both) { Both.new(:objs => [loader,manager]) }
  def master
    MasterLoader.instance
  end
  def initialize(&b)
    @blk = b
    instance_eval(&@blk)
    master.add(self)
  end
  def column(name,type,&blk)
    real_name = name
    if name.is_a?(Hash)
      raise "block given with hash" if block_given?
      blk = lambda { |x| x.send(name.values.first) }
      real_name = name.keys.first
    end
    blk ||= lambda { |x| x.send(real_name) }
    loader.columns << Column.new(:target_name => real_name, :blk => blk, :column_type => type)
  end
  def method_missing(sym,*args,&b)
    if [:string, :text, :integer, :float, :decimal, :datetime, :timestamp, :time, :date, :binary, :boolean].include?(sym)
      args.flatten.each { |x| column(x,sym,&b) }
    else
      super(sym,*args,&b)
    end
  end
  def source(file)
    loader.source_filename = file
  end
  def table(name)
    both.table_name = name
  end
  def run!
    manager.delete_rows! if @delete_existing_rows
    loader.load!
  end
  def delimiter(x)
    loader.delimiter = x
  end
end

def table_dataload(&b)
  handle_errors do
    TableLoaderDSL.new(&b)
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
GFunk911-dataload-0.9.4	lib/dataload/dsl/table_loader_dsl.rb