co.cask.cdap.etl.mock.realtime

Class MockSink

java.lang.Object

co.cask.cdap.etl.api.realtime.RealtimeSink<co.cask.cdap.api.data.format.StructuredRecord>

co.cask.cdap.etl.mock.realtime.MockSink

All Implemented Interfaces:

Destroyable, PipelineConfigurable, StageLifecycle<RealtimeContext>

public class MockSink
extends RealtimeSink<co.cask.cdap.api.data.format.StructuredRecord>

Mock sink to keep track of what was written. For every call to write, it will write its records to a new file.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	MockSink.Config Config for the sink.

Field Summary

Fields

Modifier and Type	Field and Description
static PluginClass	PLUGIN_CLASS

Fields inherited from class co.cask.cdap.etl.api.realtime.RealtimeSink

PLUGIN_TYPE

Constructor Summary

Constructors

Constructor and Description
MockSink(MockSink.Config config)

Method Summary

Methods

Modifier and Type	Method and Description
static ETLPlugin	getPlugin(File dir)
static List<co.cask.cdap.api.data.format.StructuredRecord>	getRecords(File dir, int writeNum, long timeout, TimeUnit timeUnit) Get the records written out by the writeNum call to write.
void	initialize(RealtimeContext context)
int	write(Iterable<co.cask.cdap.api.data.format.StructuredRecord> records, DataWriter dataWriter)

Methods inherited from class co.cask.cdap.etl.api.realtime.RealtimeSink

configurePipeline, destroy

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

PLUGIN_CLASS

public static final PluginClass PLUGIN_CLASS

Constructor Detail

MockSink

public MockSink(MockSink.Config config)

Method Detail

initialize

public void initialize(RealtimeContext context)
                throws Exception

Specified by:

initialize in interface StageLifecycle<RealtimeContext>

Overrides:

initialize in class RealtimeSink<co.cask.cdap.api.data.format.StructuredRecord>

Throws:

Exception

write

public int write(Iterable<co.cask.cdap.api.data.format.StructuredRecord> records,
        DataWriter dataWriter)
          throws Exception

Specified by:

write in class RealtimeSink<co.cask.cdap.api.data.format.StructuredRecord>

Throws:

Exception

getPlugin

public static ETLPlugin getPlugin(File dir)

getRecords

public static List<co.cask.cdap.api.data.format.StructuredRecord> getRecords(File dir,
                                                             int writeNum,
                                                             long timeout,
                                                             TimeUnit timeUnit)
                                                                      throws IOException,
                                                                             InterruptedException,
                                                                             ExecutionException,
                                                                             TimeoutException

Get the records written out by the writeNum call to write. Waits until the results are available.

Parameters:

dir - the temporary directory that results are written to

writeNum - the number of times write was called

timeout - how long to wait

timeUnit - time unit for how long to wait

Returns:

results for the writeNum call to write.

Throws:

IOException

InterruptedException

ExecutionException

TimeoutException