org.apache.nifi.processors.standard

Class SampleRecord.ReservoirSamplingStrategy

java.lang.Object

org.apache.nifi.processors.standard.SampleRecord.ReservoirSamplingStrategy

All Implemented Interfaces:

SampleRecord.SamplingStrategy

Enclosing class:

SampleRecord

static class SampleRecord.ReservoirSamplingStrategy
extends Object
implements SampleRecord.SamplingStrategy

Field Summary

Fields

Modifier and Type	Field and Description
(package private) int	currentCount
(package private) Random	randomNumberGenerator
(package private) ArrayList<Record>	reservoir
(package private) int	reservoirSize
(package private) RecordSetWriter	writer

Constructor Summary

Constructors

Constructor and Description
ReservoirSamplingStrategy(RecordSetWriter writer, int reservoirSize, Long randomSeed)

Method Summary

Modifier and Type	Method and Description
WriteResult	finish()
void	init()
void	sample(Record record)

Methods inherited from class java.lang.Object

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

Field Detail

writer

final RecordSetWriter writer

reservoirSize

final int reservoirSize

reservoir

final ArrayList<Record> reservoir

currentCount

int currentCount

randomNumberGenerator

final Random randomNumberGenerator

Constructor Detail

ReservoirSamplingStrategy

ReservoirSamplingStrategy(RecordSetWriter writer,
                          int reservoirSize,
                          Long randomSeed)

Method Detail

init

public void init()
          throws IOException

Specified by:

init in interface SampleRecord.SamplingStrategy

Throws:

IOException

sample

public void sample(Record record)

Specified by:

sample in interface SampleRecord.SamplingStrategy

finish

public WriteResult finish()
                   throws IOException

Specified by:

finish in interface SampleRecord.SamplingStrategy

Throws:

IOException