@Override public void timestampColumn(Column inputColumn) { int index = indexMapping[inputColumn.getIndex()]; if (index >= 0) { if (pageReader.isNull(inputColumn)) { pageBuilder.setNull(index); } else { pageBuilder.setTimestamp(index, pageReader.getTimestamp(inputColumn)); } } }
@Override public void booleanColumn(Column inputColumn) { int index = indexMapping[inputColumn.getIndex()]; if (index >= 0) { if (pageReader.isNull(inputColumn)) { pageBuilder.setNull(index); } else { pageBuilder.setBoolean(index, pageReader.getBoolean(inputColumn)); } } }
@Test public void testRepeatableClose() { MockPageOutput output = new MockPageOutput(); this.builder = new PageBuilder(bufferAllocator, Schema.builder().add("col1", STRING).build(), output); builder.close(); builder.close(); }
@Test public void testRepeatableFlush() { MockPageOutput output = new MockPageOutput(); this.builder = new PageBuilder(bufferAllocator, Schema.builder().add("col1", STRING).build(), output); builder.flush(); builder.flush(); } }
private static Row newRow(Schema schema) { ColumnValue[] values = new ColumnValue[schema.getColumnCount()]; for (Column column : schema.getColumns()) { values[column.getIndex()] = newValue(column); } return new Row(values); }
@Override public ConfigDiff resume(TaskSource taskSource, Schema schema, int taskCount, InputPlugin.Control control) { control.run(taskSource, schema, taskCount); return Exec.newConfigDiff(); }
@Override public ConfigDiff resume(TaskSource taskSource, int taskCount, FileOutputPlugin.Control control) { control.run(taskSource); return Exec.newConfigDiff(); }
@Override public ConfigDiff resume(TaskSource taskSource, Schema schema, int taskCount, OutputPlugin.Control control) { control.run(taskSource); return Exec.newConfigDiff(); }
public void openOutputs(OutputPlugin outputPlugin, Schema outputSchema, TaskSource outputTaskSource) { for (int i = 0; i < scatterCount; i++) { int outputTaskIndex = taskIndex * scatterCount + i; if (!state.getOutputTaskState(outputTaskIndex).isCommitted()) { TransactionalPageOutput tran = PluginWrappers.transactionalPageOutput( outputPlugin.open(outputTaskSource, outputSchema, outputTaskIndex)); trans[i] = tran; closeThese[i].closeThis(tran); } } }
public void write(PageBuilder pageBuilder) { if (!isNull) { writeNotNull(pageBuilder); } else { pageBuilder.writeNull(column.getIndex()); } }
private boolean isAllScatterOutputFinished(ProcessState state, int taskIndex) { for (int i = 0; i < scatterCount; i++) { int outputTaskIndex = taskIndex * scatterCount + i; if (!state.getOutputTaskState(outputTaskIndex).isCommitted()) { return false; } } return true; }
@Override public ConfigDiff resume(TaskSource taskSource, Schema schema, int taskCount, Control control) { control.run(taskSource, schema, taskCount); return Exec.newConfigDiff(); }
public ConfigDiff resume(TaskSource taskSource, Schema schema, int taskCount, OutputPlugin.Control control) { control.run(taskSource); return Exec.newConfigDiff(); }