Sha256: a5f1644463617e0ca019223df290a43fa535302fec496de9b8f96c62789c0490
Contents?: true
Size: 1.79 KB
Versions: 70
Compression:
Stored size: 1.79 KB
Contents
module Embulk
org.embulk.spi.util.dynamic.AbstractDynamicColumnSetter.module_eval do
alias_method(:set, :setRubyObject)
end
class PageBuilder
def initialize(schema, java_page_output)
require 'msgpack' # used at DynamicPageBuilder.set(Value)
# TODO get task as an argument
task = Java::SPI::Exec.newConfigSource.load_config(Java::DynamicPageBuilder::BuilderTask.java_class)
@page_builder = Java::DynamicPageBuilder.new(task, Java::Injected::BufferAllocator, schema.to_java, java_page_output)
@schema = schema
end
def add(record)
i = 0
m = record.size
while i < m
@page_builder.column(i).set(record[i])
i += 1
end
@page_builder.addRecord
nil
end
def [](index_or_column)
case index_or_column
when Integer
@page_builder.column_or_null(index_or_column)
when Column
@page_builder.column_or_null(index_or_column.index)
else
@page_builder.column_or_null(index_or_column)
end
end
def column(index_or_column)
case index_or_column
when Integer
@page_builder.column(index_or_column)
when Column
@page_builder.column(index_or_column.index)
else
@page_builder.lookupColumn(index_or_column)
end
end
def column_or_skip(index_or_column)
case index_or_column
when Integer
@page_builder.column_or_skip(index_or_column)
when Column
@page_builder.column_or_skip(index_or_column.index)
else
@page_builder.column_or_skip(index_or_column)
end
end
def add!
@page_builder.add_record
end
def flush
@page_builder.flush
end
def finish
@page_builder.finish
end
def close
@page_builder.close
end
end
end
Version data entries
70 entries across 70 versions & 1 rubygems