Search in sources :

Example 6 with SubsegmentImpl

use of com.amazonaws.xray.entities.SubsegmentImpl in project aws-xray-sdk-java by aws.

the class EntityTest method testEndingSubsegmentImplTwiceThrowsAlreadyEmittedException.

@Test(expected = AlreadyEmittedException.class)
public void testEndingSubsegmentImplTwiceThrowsAlreadyEmittedException() {
    SegmentImpl segment = new SegmentImpl(AWSXRay.getGlobalRecorder(), "test");
    segment.getName();
    SubsegmentImpl subsegment = new SubsegmentImpl(AWSXRay.getGlobalRecorder(), "test", segment);
    segment.end();
    subsegment.end();
    subsegment.end();
}
Also used : SubsegmentImpl(com.amazonaws.xray.entities.SubsegmentImpl) SegmentImpl(com.amazonaws.xray.entities.SegmentImpl) Test(org.junit.Test)

Example 7 with SubsegmentImpl

use of com.amazonaws.xray.entities.SubsegmentImpl in project aws-xray-sdk-java by aws.

the class LambdaSegmentContext method beginSubsegment.

@Override
public Subsegment beginSubsegment(AWSXRayRecorder recorder, String name) {
    if (logger.isDebugEnabled()) {
        logger.debug("Beginning subsegment named: " + name);
    }
    if (null == getTraceEntity()) {
        // First subsgment of a subsegment branch.
        Segment parentSegment = null;
        if (LambdaSegmentContext.isInitializing(LambdaSegmentContext.getTraceHeaderFromEnvironment())) {
            logger.warn(LAMBDA_TRACE_HEADER_KEY + " is missing a trace ID, parent ID, or sampling decision. Subsegment " + name + " discarded.");
            parentSegment = new FacadeSegment(recorder, new TraceID(), "", SampleDecision.NOT_SAMPLED);
        } else {
            parentSegment = LambdaSegmentContext.newFacadeSegment(recorder);
        }
        Subsegment subsegment = new SubsegmentImpl(recorder, name, parentSegment);
        subsegment.setParent(parentSegment);
        setTraceEntity(subsegment);
        return subsegment;
    } else {
        // Continuation of a subsegment branch.
        Subsegment parentSubsegment = (Subsegment) getTraceEntity();
        // Ensure customers have not leaked subsegments across invocations
        TraceID environmentRootTraceId = LambdaSegmentContext.getTraceHeaderFromEnvironment().getRootTraceId();
        if (null != environmentRootTraceId && !environmentRootTraceId.equals(parentSubsegment.getParentSegment().getTraceId())) {
            clearTraceEntity();
            return beginSubsegment(recorder, name);
        }
        Subsegment subsegment = new SubsegmentImpl(recorder, name, parentSubsegment.getParentSegment());
        subsegment.setParent(parentSubsegment);
        parentSubsegment.addSubsegment(subsegment);
        setTraceEntity(subsegment);
        return subsegment;
    }
}
Also used : TraceID(com.amazonaws.xray.entities.TraceID) SubsegmentImpl(com.amazonaws.xray.entities.SubsegmentImpl) Subsegment(com.amazonaws.xray.entities.Subsegment) FacadeSegment(com.amazonaws.xray.entities.FacadeSegment) Segment(com.amazonaws.xray.entities.Segment) FacadeSegment(com.amazonaws.xray.entities.FacadeSegment)

Aggregations

SubsegmentImpl (com.amazonaws.xray.entities.SubsegmentImpl)7 SegmentImpl (com.amazonaws.xray.entities.SegmentImpl)5 Test (org.junit.Test)5 Segment (com.amazonaws.xray.entities.Segment)4 Subsegment (com.amazonaws.xray.entities.Subsegment)3 TraceID (com.amazonaws.xray.entities.TraceID)2 Entity (com.amazonaws.xray.entities.Entity)1 FacadeSegment (com.amazonaws.xray.entities.FacadeSegment)1