How to use

org.apache.druid.segment.realtime.appenderator.StreamAppenderatorDriver

constructor

@Before
public void setUp() throws Exception
{
 appenderatorTester = new AppenderatorTester(MAX_ROWS_IN_MEMORY);
 allocator = new TestSegmentAllocator(DATA_SOURCE, Granularities.HOUR);
 segmentHandoffNotifierFactory = new TestSegmentHandoffNotifierFactory();
 dataSegmentKiller = createStrictMock(DataSegmentKiller.class);
 driver = new StreamAppenderatorDriver(
   appenderatorTester.getAppenderator(),
   allocator,
   segmentHandoffNotifierFactory,
   new TestUsedSegmentChecker(appenderatorTester),
   dataSegmentKiller,
   OBJECT_MAPPER,
   new FireDepartmentMetrics()
 );
 EasyMock.replay(dataSegmentKiller);
}

);
driver = new StreamAppenderatorDriver(
  createDropFailAppenderator(),
  allocator,

                + "foo_2000-01-01T01:00:00.000Z_2000-01-01T02:00:00.000Z_abc123]]");
driver = new StreamAppenderatorDriver(
  createPushFailAppenderator(),
  allocator,

private void testFailDuringPublishInternal(boolean failWithException) throws Exception
 driver = new StreamAppenderatorDriver(
   new FailableAppenderator(),
   allocator,

                + "foo_2000-01-01T01:00:00.000Z_2000-01-01T02:00:00.000Z_abc123]]");
driver = new StreamAppenderatorDriver(
  createPersistFailAppenderator(),
  allocator,

public StreamAppenderatorDriver newDriver(
  final Appenderator appenderator,
  final TaskToolbox toolbox,
  final FireDepartmentMetrics metrics
)
{
 return new StreamAppenderatorDriver(
   appenderator,
   new ActionBasedSegmentAllocator(
     toolbox.getTaskActionClient(),
     dataSchema,
     (schema, row, sequenceName, previousSegmentId, skipSegmentLineageCheck) -> new SegmentAllocateAction(
       schema.getDataSource(),
       row.getTimestamp(),
       schema.getGranularitySpec().getQueryGranularity(),
       schema.getGranularitySpec().getSegmentGranularity(),
       sequenceName,
       previousSegmentId,
       skipSegmentLineageCheck
     )
   ),
   toolbox.getSegmentHandoffNotifierFactory(),
   new ActionBasedUsedSegmentChecker(toolbox.getTaskActionClient()),
   toolbox.getDataSegmentKiller(),
   toolbox.getObjectMapper(),
   metrics
 );
}

private static StreamAppenderatorDriver newDriver(
  final DataSchema dataSchema,
  final Appenderator appenderator,
  final TaskToolbox toolbox,
  final FireDepartmentMetrics metrics
)
{
 return new StreamAppenderatorDriver(
   appenderator,
   new ActionBasedSegmentAllocator(
     toolbox.getTaskActionClient(),
     dataSchema,
     (schema, row, sequenceName, previousSegmentId, skipSegmentLineageCheck) -> new SegmentAllocateAction(
       schema.getDataSource(),
       row.getTimestamp(),
       schema.getGranularitySpec().getQueryGranularity(),
       schema.getGranularitySpec().getSegmentGranularity(),
       sequenceName,
       previousSegmentId,
       skipSegmentLineageCheck
     )
   ),
   toolbox.getSegmentHandoffNotifierFactory(),
   new ActionBasedUsedSegmentChecker(toolbox.getTaskActionClient()),
   toolbox.getDataSegmentKiller(),
   toolbox.getObjectMapper(),
   metrics
 );
}

private static StreamAppenderatorDriver newDriver(
  final DataSchema dataSchema,
  final Appenderator appenderator,
  final TaskToolbox toolbox,
  final FireDepartmentMetrics metrics
)
{
 return new StreamAppenderatorDriver(
   appenderator,
   new ActionBasedSegmentAllocator(
     toolbox.getTaskActionClient(),
     dataSchema,
     (schema, row, sequenceName, previousSegmentId, skipSegmentLineageCheck) -> new SegmentAllocateAction(
       schema.getDataSource(),
       row.getTimestamp(),
       schema.getGranularitySpec().getQueryGranularity(),
       schema.getGranularitySpec().getSegmentGranularity(),
       sequenceName,
       previousSegmentId,
       skipSegmentLineageCheck
     )
   ),
   toolbox.getSegmentHandoffNotifierFactory(),
   new ActionBasedUsedSegmentChecker(toolbox.getTaskActionClient()),
   toolbox.getDataSegmentKiller(),
   toolbox.getObjectMapper(),
   metrics
 );
}