tech.aroma.data.performance.Operations Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2017 RedRoma, Inc.
 *
 * 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
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

 
package tech.aroma.data.performance;


import com.google.common.base.Strings;
import org.apache.thrift.TException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import tech.sirwellington.alchemy.annotations.access.Internal;
import tech.sirwellington.alchemy.annotations.access.NonInstantiable;
import tech.sirwellington.alchemy.annotations.arguments.Required;

import static tech.sirwellington.alchemy.arguments.Arguments.*;
import static tech.sirwellington.alchemy.arguments.assertions.Assertions.notNull;

/**
 *
 * @author SirWellington
 */
@Internal
@NonInstantiable
final class Operations 
{
    private final static Logger LOG = LoggerFactory.getLogger(Operations.class);

    
    /**
     * Measures the time it takes to run an operation, in ms.
     * 
     * @param operation The operation to call
     * 
     * @return The Time it took for the operation to complete, in ms.
     * @throws Exception 
     */
    public static long measureOperation(@Required Operation operation) throws TException
    {
        checkThat(operation).is(notNull());
        
        long start = System.currentTimeMillis();
        
        operation.call();
        
        long end = System.currentTimeMillis();
        
        return end - start;
    }
    
    public static  T logLatency(@Required Operation operation, String operationName) throws TException
    {
        checkThat(operation).is(notNull());
        
        operationName = Strings.nullToEmpty(operationName);
        
        long start = System.currentTimeMillis();
        
        try
        {
            return operation.call();
        }
        finally
        {
            long end = System.currentTimeMillis();
            LOG.debug("{} Operation took {} ms", operationName, end - start);
        }
    }
}