Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/**
* Copyright (c) KMG. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*/
package io.sbk.api;
import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
import io.sbk.data.DataType;
import io.sbk.logger.ReadRequestsLogger;
import io.time.Time;
import java.io.EOFException;
import java.io.IOException;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.atomic.AtomicLong;
/**
* Abstract class for Callback Reader.
*/
public abstract non-sealed class AbstractCallbackReader implements DataReader {
private DataType dataType;
private Time time;
private CompletableFuture ret;
private AtomicLong readCnt;
private long beginTime;
private Worker reader;
private double msToRun;
private long recordsCount;
/**
* set the Callback and start the CallBack Reader.
*
* @param callback Reader callback.
* @throws IOException If an exception occurred.
*/
public abstract void start(Callback callback) throws IOException;
/**
* Stop the CallBack Reader.
*
* @throws IOException If an exception occurred.
*/
public abstract void stop() throws IOException;
/**
* Close the CallBack Reader.
* stops the callback reader.
*
* @throws IOException If an exception occurred.
*/
@Override
public final void close() throws IOException {
stop();
complete();
}
/**
* Keeps record of Benchmark if record gets completed it call complete function.
*
* @param startTime long
* @param endTime long
* @param dataSize int
* @param events int
*/
public void recordBenchmark(long startTime, long endTime, int dataSize, int events) {
final long cnt = readCnt.incrementAndGet();
reader.perlChannel.send(startTime, endTime, events, dataSize);
if (this.msToRun > 0 && ((endTime - beginTime) >= this.msToRun)) {
complete();
} else if (this.recordsCount > cnt) {
complete();
}
}
/**
* Default Implementation to initialize the callback reader.
*
* @param reader Reader Descriptor
* @param secondsToRun Number of seconds to run
* @param recordsCount Records count
* @param dType dataType
* @param time time interface
* @param callback Callback interface
* @throws IOException If an exception occurred.
*/
@SuppressFBWarnings("EI_EXPOSE_REP2")
public void initialize(Worker reader, long secondsToRun, long recordsCount, DataType dType, Time time,
Callback callback) throws IOException {
this.reader = reader;
this.dataType = dType;
this.time = time;
this.readCnt = new AtomicLong(0);
this.beginTime = time.getCurrentTime();
this.msToRun = secondsToRun * Time.MS_PER_SEC;
this.recordsCount = recordsCount;
this.ret = new CompletableFuture<>();
start(callback);
}
/**
* Default Implementation complete the read.
*/
public void complete() {
if (ret != null) {
ret.complete(null);
}
}
/**
* Default Implementation to wait for the readers to complete.
*
* @throws IOException If an exception occurred.
*/
public void waitToComplete() throws IOException {
try {
if (ret != null) {
ret.get();
}
} catch (ExecutionException | InterruptedException ex) {
throw new IOException(ex);
}
}
/**
* Default Implementation run the Benchmark.
*
* @param reader Reader Descriptor
* @param secondsToRun Number of seconds to run
* @param recordsCount Records count
* @param dType dataType
* @param time time interface
* @param callback Callback interface
* @throws IOException If an exception occurred.
*/
public void run(Worker reader, long secondsToRun, long recordsCount, DataType dType, Time time,
Callback callback) throws IOException {
initialize(reader, secondsToRun, recordsCount, dType, time, callback);
waitToComplete();
}
/**
* Implementation for benchmarking reader by reading given number of records.
*
* @param reader Reader Descriptor
* @param recordsCount Records count
* @param dType dataType
* @param time time interface
* @throws EOFException If the End of the file occurred.
* @throws IOException If an exception occurred.
*/
public void RecordsReader(Worker reader, long recordsCount, DataType dType, Time time) throws EOFException,
IOException {
run(reader, 0, recordsCount, dType, time, new ConsumeRead());
}
/**
* Implementation for benchmarking reader by reading given number of records.
*
* @param reader Reader Descriptor
* @param recordsCount Records count
* @param dType dataType
* @param time time interface
* @param logger log read requests
* @throws EOFException If the End of the file occurred.
* @throws IOException If an exception occurred.
*/
public void RecordsReader(Worker reader, long recordsCount, DataType dType, Time time,
ReadRequestsLogger logger) throws EOFException, IOException {
RecordsReader(reader, recordsCount, dType, time);
}
/**
* Default implementation for benchmarking reader by reading given number of records.
*
* @param reader Reader Descriptor
* @param recordsCount Records count
* @param dType dataType
* @param time time interface
* @throws EOFException If the End of the file occurred.
* @throws IOException If an exception occurred.
*/
public void RecordsReaderRW(Worker reader, long recordsCount, DataType dType, Time time) throws EOFException,
IOException {
run(reader, 0, recordsCount, dType, time, new ConsumeRW());
}
/**
* Default implementation for benchmarking reader by reading given number of records.
*
* @param reader Reader Descriptor
* @param recordsCount Records count
* @param dType dataType
* @param time time interface
* @param logger log read requests
* @throws EOFException If the End of the file occurred.
* @throws IOException If an exception occurred.
*/
public void RecordsReaderRW(Worker reader, long recordsCount, DataType dType, Time time,
ReadRequestsLogger logger) throws EOFException,
IOException {
RecordsReaderRW(reader, recordsCount, dType, time);
}
/**
* Default implementation for benchmarking reader by reading events/records for specific time duration.
*
* @param reader Reader Descriptor
* @param secondsToRun Number of seconds to run
* @param dType dataType
* @param time time interface
* @throws EOFException If the End of the file occurred.
* @throws IOException If an exception occurred.
*/
public void RecordsTimeReader(Worker reader, long secondsToRun, DataType dType, Time time)
throws EOFException, IOException {
run(reader, secondsToRun, 0, dType, time, new ConsumeRead());
}
/**
* Default implementation for benchmarking reader by reading events/records for specific time duration.
*
* @param reader Reader Descriptor
* @param secondsToRun Number of seconds to run
* @param dType dataType
* @param time time interface
* @param logger log read requests
* @throws EOFException If the End of the file occurred.
* @throws IOException If an exception occurred.
*/
public void RecordsTimeReader(Worker reader, long secondsToRun, DataType dType, Time time,
ReadRequestsLogger logger) throws EOFException, IOException {
RecordsTimeReader(reader, secondsToRun, dType, time);
}
/**
* Default implementation for benchmarking reader by reading events/records for specific time duration.
*
* @param reader Reader Descriptor
* @param secondsToRun Number of seconds to run
* @param dType dataType
* @param time time interface
* @throws EOFException If the End of the file occurred.
* @throws IOException If an exception occurred.
*/
public void RecordsTimeReaderRW(Worker reader, long secondsToRun, DataType dType, Time time) throws EOFException, IOException {
run(reader, secondsToRun, 0, dType, time, new ConsumeRW());
}
/**
* Default implementation for benchmarking reader by reading events/records for specific time duration.
*
* @param reader Reader Descriptor
* @param secondsToRun Number of seconds to run
* @param dType dataType
* @param time time interface
* @param logger log read requests
* @throws EOFException If the End of the file occurred.
* @throws IOException If an exception occurred.
*/
public void RecordsTimeReaderRW(Worker reader, long secondsToRun, DataType dType, Time time,
ReadRequestsLogger logger) throws EOFException, IOException {
RecordsTimeReaderRW(reader, secondsToRun, dType, time);
}
/**
* Benchmarking reader by reading given number of records with Rate controlled.
*
* @param reader Reader Descriptor
* @param recordsCount Records count
* @param dType dataType
* @param time time interface
* @param rController Rate Controller
* @throws EOFException If the End of the file occurred.
* @throws IOException If an exception occurred.
*/
public void RecordsReaderRateControl(Worker reader, long recordsCount, DataType dType, Time time,
RateController rController) throws EOFException, IOException {
run(reader, 0, recordsCount, dType, time, new ConsumeRead());
}
/**
* Benchmarking reader by reading given number of records with Rate controlled.
*
* @param reader Reader Descriptor
* @param recordsCount Records count
* @param dType dataType
* @param time time interface
* @param rController Rate Controller
* @param logger log read requests
* @throws EOFException If the End of the file occurred.
* @throws IOException If an exception occurred.
*/
public void RecordsReaderRateControl(Worker reader, long recordsCount, DataType dType, Time time,
RateController rController, ReadRequestsLogger logger)
throws EOFException, IOException {
RecordsReaderRateControl(reader, recordsCount, dType, time, rController);
}
/**
* Benchmarking reader by reading given number of records with Rate controlled.
* used while another writer is writing the data.
*
* @param reader Reader Descriptor
* @param recordsCount Records count
* @param dType dataType
* @param time time interface
* @param rController Rate Controller
* @throws EOFException If the End of the file occurred.
* @throws IOException If an exception occurred.
*/
public void RecordsReaderRWRateControl(Worker reader, long recordsCount, DataType dType, Time time,
RateController rController) throws EOFException, IOException {
run(reader, 0, recordsCount, dType, time, new ConsumeRW());
}
/**
* Benchmarking reader by reading given number of records with Rate controlled.
* used while another writer is writing the data.
*
* @param reader Reader Descriptor
* @param recordsCount Records count
* @param dType dataType
* @param time time interface
* @param rController Rate Controller
* @param logger log read requests
* @throws EOFException If the End of the file occurred.
* @throws IOException If an exception occurred.
*/
public void RecordsReaderRWRateControl(Worker reader, long recordsCount, DataType dType, Time time,
RateController rController, ReadRequestsLogger logger)
throws EOFException, IOException {
RecordsReaderRWRateControl(reader, recordsCount, dType, time, rController);
}
/**
* Benchmarking reader by reading events/records for specific time duration with Rate controlled.
*
* @param reader Reader Descriptor
* @param secondsToRun Number of seconds to run
* @param dType dataType
* @param time time interface
* @param rController Rate Controller
* @throws EOFException If the End of the file occurred.
* @throws IOException If an exception occurred.
*/
public void RecordsTimeReaderRateControl(Worker reader, long secondsToRun, DataType dType, Time time,
RateController rController) throws EOFException, IOException {
run(reader, secondsToRun, 0, dType, time, new ConsumeRead());
}
/**
* Benchmarking reader by reading events/records for specific time duration with Rate controlled.
*
* @param reader Reader Descriptor
* @param secondsToRun Number of seconds to run
* @param dType dataType
* @param time time interface
* @param rController Rate Controller
* @param logger log read requests
* @throws EOFException If the End of the file occurred.
* @throws IOException If an exception occurred.
*/
public void RecordsTimeReaderRateControl(Worker reader, long secondsToRun, DataType dType, Time time,
RateController rController, ReadRequestsLogger logger)
throws EOFException, IOException {
RecordsTimeReaderRateControl(reader, secondsToRun, dType, time, rController);
}
/**
* Benchmarking reader by reading events/records for specific time duration with Rate controlled.
* used while another writer is writing the data.
*
* @param reader Reader Descriptor
* @param secondsToRun Number of seconds to run
* @param dType dataType
* @param time time interface
* @param rController Rate Controller
* @throws EOFException If the End of the file occurred.
* @throws IOException If an exception occurred.
*/
public void RecordsTimeReaderRWRateControl(Worker reader, long secondsToRun, DataType dType, Time time,
RateController rController) throws EOFException, IOException {
run(reader, secondsToRun, 0, dType, time, new ConsumeRW());
}
/**
* Benchmarking reader by reading events/records for specific time duration with Rate controlled.
* used while another writer is writing the data.
*
* @param reader Reader Descriptor
* @param secondsToRun Number of seconds to run
* @param dType dataType
* @param time time interface
* @param rController Rate Controller
* @param logger log read requests
* @throws EOFException If the End of the file occurred.
* @throws IOException If an exception occurred.
*/
public void RecordsTimeReaderRWRateControl(Worker reader, long secondsToRun, DataType dType, Time time,
RateController rController, ReadRequestsLogger logger)
throws EOFException, IOException {
RecordsTimeReaderRWRateControl(reader, secondsToRun, dType, time, rController);
}
private class ConsumeRead implements Callback {
public void consume(final T data) {
final long endTime = time.getCurrentTime();
recordBenchmark(endTime, endTime, dataType.length(data), 1);
}
public void record(long startTime, long endTime, int dataSize, int records) {
recordBenchmark(startTime, endTime, dataSize, records);
}
}
private class ConsumeRW implements Callback {
public void consume(final T data) {
recordBenchmark(dataType.getTime(data), time.getCurrentTime(), dataType.length(data), 1);
}
public void record(long startTime, long endTime, int dataSize, int records) {
recordBenchmark(startTime, endTime, dataSize, records);
}
}
}
