com.github.kristofa.brave.ClientResponseInterceptor Maven / Gradle / Ivy

Go to download
package com.github.kristofa.brave;

import java.util.Objects;

/**
 * Contains logic for dealing with response from client request.
 * This means it will:
 *
 * - Submit potential annotations
 * - Submit client received annotation
 *
 * You will have to implement ClientResponseAdapter.
 *
 * @see ClientResponseAdapter
 */
public class ClientResponseInterceptor {

    private final ClientTracer clientTracer;

    public ClientResponseInterceptor(ClientTracer clientTracer) {
        this.clientTracer = Objects.requireNonNull(clientTracer);
    }

    /**
     * Handle a client response.
     *
     * @param adapter Adapter that hides implementation details.
     */
    public void handle(ClientResponseAdapter adapter) {
        try {
            for (KeyValueAnnotation annotation : adapter.responseAnnotations()) {
                clientTracer.submitBinaryAnnotation(annotation.getKey(), annotation.getValue());
            }
        }
        finally
        {
            clientTracer.setClientReceived();
        }
    }
}