com.newrelic.opentracing.state.DistributedTracingState Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of newrelic-java-lambda Show documentation

New Relic OpenTracing Tracer implementation for instrumenting AWS Lambda functions.

The newest version!

/*
 * Copyright 2020 New Relic Corporation. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 */

package com.newrelic.opentracing.state;

import com.newrelic.opentracing.LambdaPayloadContext;
import com.newrelic.opentracing.LambdaSpan;
import com.newrelic.opentracing.dt.DistributedTracePayload;
import com.newrelic.opentracing.dt.DistributedTracePayloadImpl;
import com.newrelic.opentracing.dt.DistributedTracing;
import com.newrelic.opentracing.util.DistributedTraceUtil;

import java.util.Collections;
import java.util.Map;

public class DistributedTracingState {

    private final DistributedTracePayloadImpl inboundPayload;
    private final long transportTimeMillis;
    private final String traceId;
    private final Map baggage;

    public DistributedTracingState(LambdaPayloadContext context) {
        this.inboundPayload = context.getPayload();
        this.transportTimeMillis = context.getTransportDurationInMillis();
        this.baggage = context.getBaggage();
        this.traceId = inboundPayload.getTraceId();
    }

    public DistributedTracingState() {
        inboundPayload = null;
        transportTimeMillis = Long.MIN_VALUE;
        this.baggage = Collections.emptyMap();
        traceId = DistributedTraceUtil.generateGuid();
    }

    public DistributedTracePayloadImpl getInboundPayload() {
        return inboundPayload;
    }

    public String getTraceId() {
        return traceId;
    }

    public DistributedTracePayload createDistributedTracingPayload(LambdaSpan span) {
        return DistributedTracing.getInstance().createDistributedTracePayload(span);
    }

    public Map getBaggage() {
        return baggage;
    }

    /**
     * @return transport duration or Long.MIN_VALUE if not set
     */
    public long getTransportTimeMillis() {
        return transportTimeMillis;
    }

}