ResultsLogValidationSummaryCalculator.java

package com.ldbc.driver.validation;

import org.HdrHistogram.Histogram;

import java.util.HashMap;
import java.util.Map;

class ResultsLogValidationSummaryCalculator
{
    private final Histogram delays;
    private final Map<String,Histogram> delaysPerType;
    private final Map<String,Long> delaysAboveThresholdPerType;
    private final long maxDelayAsMilli;
    private final long excessiveDelayThresholdAsMilli;
    private long delaysAboveThreshold;

    ResultsLogValidationSummaryCalculator( long maxDelayAsMilli, long excessiveDelayThresholdAsMilli )
    {
        this.maxDelayAsMilli = maxDelayAsMilli;
        this.excessiveDelayThresholdAsMilli = excessiveDelayThresholdAsMilli;
        this.delays = new Histogram( 1, Math.max( 2, maxDelayAsMilli ), 5 );
        this.delaysPerType = new HashMap<>();
        this.delaysAboveThresholdPerType = new HashMap<>();
        this.delaysAboveThreshold = 0;
    }

    void recordDelay( String operationType, long delayAsMilli )
    {
        delays.recordValue( delayAsMilli );

        Histogram delayForType = delaysPerType.get( operationType );
        if ( null == delayForType )
        {
            delayForType = new Histogram( 1, Math.max( 2, maxDelayAsMilli ), 5 );
            delaysPerType.put( operationType, delayForType );
        }
        delayForType.recordValue( delayAsMilli );

        Long delaysAboveThresholdForType = delaysAboveThresholdPerType.get( operationType );
        if ( null == delaysAboveThresholdForType )
        {
            delaysAboveThresholdForType = 0L;
            delaysAboveThresholdPerType.put( operationType, delaysAboveThresholdForType );
        }
        if ( delayAsMilli > excessiveDelayThresholdAsMilli )
        {
            delaysAboveThreshold++;
            delaysAboveThresholdPerType.put( operationType, delaysAboveThresholdForType + 1 );
        }
    }

    ResultsLogValidationSummary snapshot()
    {
        Map<String,Long> minDelayAsMilliPerType = new HashMap<>();
        Map<String,Long> maxDelayAsMilliPerType = new HashMap<>();
        Map<String,Long> meanDelayAsMilliPerType = new HashMap<>();
        for ( String operationType : delaysPerType.keySet() )
        {
            minDelayAsMilliPerType.put(
                    operationType,
                    delaysPerType.get( operationType ).getMinValue()
            );
            maxDelayAsMilliPerType.put(
                    operationType,
                    delaysPerType.get( operationType ).getMaxValue()
            );
            meanDelayAsMilliPerType.put(
                    operationType,
                    Math.round( Math.ceil( delaysPerType.get( operationType ).getMean() ) )
            );
        }
        return new ResultsLogValidationSummary(
                excessiveDelayThresholdAsMilli,
                delaysAboveThreshold,
                delaysAboveThresholdPerType,
                delays.getMinValue(),
                delays.getMaxValue(),
                Math.round( delays.getMean() ),
                minDelayAsMilliPerType,
                maxDelayAsMilliPerType,
                meanDelayAsMilliPerType
        );
    }
}