forked from D-Net/dnet-hadoop
51 lines
1.1 KiB
Java
51 lines
1.1 KiB
Java
package eu.dnetlib.dhp.wf.importer;
|
|
|
|
import java.io.IOException;
|
|
|
|
import org.apache.avro.file.DataFileWriter;
|
|
|
|
/**
|
|
* {@link DataFileWriter} based record receiver with counter of
|
|
* received records.
|
|
*
|
|
* @author madryk
|
|
*/
|
|
public class DataFileRecordReceiverWithCounter<T> extends DataFileRecordReceiver<T> {
|
|
|
|
private long receivedCount = 0L;
|
|
|
|
|
|
//------------------------ CONSTRUCTORS --------------------------
|
|
|
|
/**
|
|
* Default constructor
|
|
*
|
|
* @param writer - writer of the received records
|
|
*/
|
|
public DataFileRecordReceiverWithCounter(DataFileWriter<T> writer) {
|
|
super(writer);
|
|
}
|
|
|
|
|
|
//------------------------ GETTERS --------------------------
|
|
|
|
/**
|
|
* Returns number of received records
|
|
*/
|
|
public long getReceivedCount() {
|
|
return receivedCount;
|
|
}
|
|
|
|
|
|
//------------------------ LOGIC --------------------------
|
|
|
|
/**
|
|
* Receives passed record and increments the counter.
|
|
*/
|
|
@Override
|
|
public void receive(T record) throws IOException {
|
|
super.receive(record);
|
|
++receivedCount;
|
|
}
|
|
}
|