com.rbkmoney.woody.api.interceptor.ContextInterceptor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of woody-api Show documentation

There is a newer version: 1.1.23

package com.rbkmoney.woody.api.interceptor;

import com.rbkmoney.woody.api.trace.ContextUtils;
import com.rbkmoney.woody.api.trace.TraceData;
import com.rbkmoney.woody.api.trace.context.TraceContext;

public class ContextInterceptor implements CommonInterceptor {
    private final TraceContext traceContext;
    private final CommonInterceptor interceptor;

    public ContextInterceptor(TraceContext traceContext, CommonInterceptor interceptor) {
        this.traceContext = traceContext;
        this.interceptor = interceptor != null ? interceptor : new EmptyCommonInterceptor();
    }

    @Override
    public boolean interceptRequest(TraceData traceData, Object providerContext, Object... contextParams) {
        traceContext.init();
        return interceptor.interceptRequest(traceData, providerContext, contextParams);
    }

    @Override
    public boolean interceptResponse(TraceData traceData, Object providerContext, Object... contextParams) {
        try {
            return interceptor.interceptResponse(traceData, providerContext, contextParams);
        } finally {
            traceContext.destroy(ContextUtils.hasCallErrors(traceData.getActiveSpan()));
        }
    }
}