All Downloads are FREE. Search and download functionalities are using the official Maven repository.

io.sbk.api.impl.SbkCallbackReader Maven / Gradle / Ivy

Go to download 

/**
 * 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.impl;

import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
import io.perl.api.PerlChannel;
import io.sbk.action.Action;
import io.sbk.api.Benchmark;
import io.sbk.api.Callback;
import io.sbk.params.ParameterOptions;
import io.sbk.api.Worker;
import io.sbk.data.DataType;
import io.time.Time;

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.atomic.AtomicLong;

/**
 * SBK Callback reader implementation.
 *
 * @deprecated This interface is replaced by Abstract class AbstractCallbackReader and SbkReader.
 */
@Deprecated
final public class SbkCallbackReader extends Worker implements Callback, Benchmark {
    final private DataType dataType;
    final private Time time;
    final private CompletableFuture ret;
    final private Callback callback;
    final private AtomicLong readCnt;
    final private double msToRun;
    final private long totalRecords;
    private long beginTime;

    @SuppressFBWarnings("EI_EXPOSE_REP2")
    public SbkCallbackReader(int readerId, ParameterOptions params, PerlChannel perlChannel, int idMax,
                             DataType dataType, Time time) {
        super(readerId, params, perlChannel);
        this.dataType = dataType;
        this.time = time;
        this.ret = new CompletableFuture<>();
        this.readCnt = new AtomicLong(0);
        this.beginTime = 0;
        this.msToRun = params.getTotalSecondsToRun() * Time.MS_PER_SEC;
        this.totalRecords = params.getTotalRecords();

        if (params.getAction() == Action.Write_Reading) {
            callback = this::consumeRW;
        } else {
            callback = this::consumeRead;
        }
    }

    @Override
    public CompletableFuture start() {
        this.beginTime = time.getCurrentTime();
        return ret.toCompletableFuture();
    }

    @Override
    public void stop() {
        ret.complete(null);
    }


    @Override
    public void record(long startTime, long endTime, int dataSize, int events) {
        final long cnt = readCnt.incrementAndGet();
        perlChannel.send(startTime, endTime, events, dataSize);
        if (this.msToRun > 0 && (time.elapsedMilliSeconds(endTime, beginTime) >= this.msToRun)) {
            ret.complete(null);
        } else if (this.totalRecords > cnt) {
            ret.complete(null);
        }
    }

    @Override
    public void consume(Object data) {
        callback.consume(data);
    }

    private void consumeRead(Object data) {
        final long endTime = time.getCurrentTime();
        record(endTime, endTime, dataType.length(data), 1);
    }

    private void consumeRW(Object data) {
        record(dataType.getTime(data), time.getCurrentTime(), dataType.length(data), 1);
    }
}
    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy