com.netflix.dyno.connectionpool.impl.OperationResultImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dyno-core Show documentation
dyno-core
There is a newer version: 1.9.1
/*******************************************************************************
 * Copyright 2011 Netflix
 *
 * 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 com.netflix.dyno.connectionpool.impl;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;

import com.netflix.dyno.connectionpool.Host;
import com.netflix.dyno.connectionpool.OperationMonitor;
import com.netflix.dyno.connectionpool.OperationResult;
import com.netflix.dyno.connectionpool.exception.DynoException;

/**
 * Impl for {@link OperationResult}
 * It tracks operation result, op attempts, latency, execution host etc
 *
 * @param 
 * @author poberai
 */
public class OperationResultImpl implements OperationResult {

    private final String opName;
    private final R result;
    private final Future futureResult;
    private Host host = null;
    private long duration = 0;
    private int attempts = 0;
    private final OperationMonitor opMonitor;
    private final ConcurrentHashMap metadata = new ConcurrentHashMap();

    public OperationResultImpl(String name, R r, OperationMonitor monitor) {
        opName = name;
        result = r;
        futureResult = null;
        opMonitor = monitor;
    }

    public OperationResultImpl(String name, Future future, OperationMonitor monitor) {
        opName = name;
        result = null;
        futureResult = future;
        opMonitor = monitor;
    }

    @Override
    public Host getNode() {
        return host;
    }

    @Override
    public R getResult() {
        try {
            return futureResult != null ? futureResult.get() : result;
        } catch (Exception e) {
            throw new DynoException(e);
        }
    }

    @Override
    public long getLatency() {
        return duration;
    }

    @Override
    public long getLatency(TimeUnit units) {
        return units.convert(duration, TimeUnit.MILLISECONDS);
    }

    @Override
    public int getAttemptsCount() {
        return attempts;
    }

    @Override
    public OperationResultImpl setAttemptsCount(int count) {
        attempts = count;
        return this;
    }

    public OperationResultImpl setNode(Host h) {
        host = h;
        return this;
    }

    public OperationResultImpl attempts(int count) {
        attempts = count;
        return this;
    }

    public OperationResultImpl latency(long time) {
        this.duration = time;
        if (opMonitor != null) {
            opMonitor.recordLatency(opName, time, TimeUnit.MILLISECONDS);
        }
        return this;
    }

    @Override
    public OperationResultImpl setLatency(long time, TimeUnit unit) {
        this.duration = TimeUnit.MILLISECONDS.convert(time, unit);
        if (opMonitor != null) {
            opMonitor.recordLatency(opName, time, unit);
        }
        return this;
    }

    @Override
    public Map getMetadata() {
        return metadata;
    }

    @Override
    public OperationResultImpl addMetadata(String key, String value) {
        metadata.put(key, value);
        return this;
    }

    @Override
    public OperationResultImpl addMetadata(Map map) {
        for (String key : map.keySet()) {
            metadata.put(key, map.get(key).toString());
        }
        return this;
    }
}