Search in sources :

Example 21 with Emitter

use of com.amazonaws.xray.emitters.Emitter in project aws-xray-sdk-java by aws.

the class CustomSegmentContextTest method setupAWSXRay.

@BeforeEach
void setupAWSXRay() {
    Emitter blankEmitter = Mockito.mock(Emitter.class);
    Mockito.doReturn(true).when(blankEmitter).sendSegment(Mockito.anyObject());
    Mockito.doReturn(true).when(blankEmitter).sendSubsegment(Mockito.anyObject());
    LocalizedSamplingStrategy defaultSamplingStrategy = new LocalizedSamplingStrategy();
    SegmentContextResolverChain chain = new SegmentContextResolverChain();
    chain.addResolver(new GlobalMapSegmentContextResolver());
    AWSXRay.setGlobalRecorder(AWSXRayRecorderBuilder.standard().withEmitter(blankEmitter).withSegmentContextResolverChain(chain).withSamplingStrategy(defaultSamplingStrategy).build());
    AWSXRay.clearTraceEntity();
}
Also used : Emitter(com.amazonaws.xray.emitters.Emitter) LocalizedSamplingStrategy(com.amazonaws.xray.strategy.sampling.LocalizedSamplingStrategy) BeforeEach(org.junit.jupiter.api.BeforeEach)

Example 22 with Emitter

use of com.amazonaws.xray.emitters.Emitter in project aws-xray-sdk-java by aws.

the class LambdaSegmentContextTest method setupAWSXRay.

@BeforeEach
public void setupAWSXRay() {
    Emitter blankEmitter = Mockito.mock(Emitter.class);
    Mockito.doReturn(true).when(blankEmitter).sendSegment(Mockito.anyObject());
    Mockito.doReturn(true).when(blankEmitter).sendSubsegment(Mockito.anyObject());
    AWSXRay.setGlobalRecorder(AWSXRayRecorderBuilder.standard().withEmitter(blankEmitter).withSamplingStrategy(new LocalizedSamplingStrategy()).build());
    AWSXRay.clearTraceEntity();
}
Also used : Emitter(com.amazonaws.xray.emitters.Emitter) LocalizedSamplingStrategy(com.amazonaws.xray.strategy.sampling.LocalizedSamplingStrategy) BeforeEach(org.junit.jupiter.api.BeforeEach)

Example 23 with Emitter

use of com.amazonaws.xray.emitters.Emitter in project aws-xray-sdk-java by aws.

the class AWSXRayRecorderBuilder method build.

/**
 * Constructs and returns an AWSXRayRecorder with the provided configuration.
 *
 * @return a configured instance of AWSXRayRecorder
 */
public AWSXRayRecorder build() {
    AWSXRayRecorder client = new AWSXRayRecorder();
    if (samplingStrategy != null) {
        client.setSamplingStrategy(samplingStrategy);
    }
    if (streamingStrategy != null) {
        client.setStreamingStrategy(streamingStrategy);
    }
    if (prioritizationStrategy != null) {
        client.setPrioritizationStrategy(prioritizationStrategy);
    }
    if (throwableSerializationStrategy != null) {
        client.setThrowableSerializationStrategy(throwableSerializationStrategy);
    }
    ContextMissingStrategy contextMissingStrategy = this.contextMissingStrategy;
    if (contextMissingStrategy != null && !AWSXRayRecorderBuilder.contextMissingStrategyFromEnvironmentVariable().isPresent() && !AWSXRayRecorderBuilder.contextMissingStrategyFromSystemProperty().isPresent()) {
        client.setContextMissingStrategy(contextMissingStrategy);
    }
    if (segmentContextResolverChain != null) {
        client.setSegmentContextResolverChain(segmentContextResolverChain);
    }
    if (emitter != null) {
        client.setEmitter(emitter);
    }
    if (!segmentListeners.isEmpty()) {
        client.addAllSegmentListeners(segmentListeners);
    }
    if (useFastIdGenerator) {
        client.useFastIdGenerator();
    } else {
        client.useSecureIdGenerator();
    }
    if (forcedTraceIdGeneration) {
        client.setForcedTraceIdGeneration(true);
    }
    plugins.stream().filter(Objects::nonNull).filter(p -> p.isEnabled()).forEach(plugin -> {
        logger.info("Collecting trace metadata from " + plugin.getClass().getName() + ".");
        try {
            Map<String, @Nullable Object> runtimeContext = plugin.getRuntimeContext();
            if (!runtimeContext.isEmpty()) {
                client.putRuntimeContext(plugin.getServiceName(), runtimeContext);
                /**
                 * Given several enabled plugins, the recorder should resolve a single one that's most representative of this
                 * environment
                 * Resolution order: EB > EKS > ECS > EC2
                 * EKS > ECS because the ECS plugin checks for an environment variable whereas the EKS plugin checks for a
                 * kubernetes authentication file, which is a stronger enable condition
                 */
                String clientOrigin = client.getOrigin();
                if (clientOrigin == null || ORIGIN_PRIORITY.getOrDefault(plugin.getOrigin(), 0) < ORIGIN_PRIORITY.getOrDefault(clientOrigin, 0)) {
                    client.setOrigin(plugin.getOrigin());
                }
            } else {
                logger.warn(plugin.getClass().getName() + " plugin returned empty runtime context data. The recorder will " + "not be setting segment origin or runtime context values from this plugin.");
            }
        } catch (Exception e) {
            logger.warn("Failed to get runtime context from " + plugin.getClass().getName() + ".", e);
        }
        try {
            Set<AWSLogReference> logReferences = plugin.getLogReferences();
            if (Objects.nonNull(logReferences)) {
                if (!logReferences.isEmpty()) {
                    client.addAllLogReferences(logReferences);
                } else {
                    logger.debug(plugin.getClass().getName() + " plugin returned empty Log References. The recorder will not " + "reflect the logs from this plugin.");
                }
            }
        } catch (Exception e) {
            logger.warn("Failed to get log references from " + plugin.getClass().getName() + ".", e);
        }
    });
    String logGroupFromEnv = System.getenv(LOG_GROUP_KEY);
    if (StringValidator.isNotNullOrBlank(logGroupFromEnv)) {
        logger.info("Recording log group " + logGroupFromEnv + " from environment variable.");
        AWSLogReference logReference = new AWSLogReference();
        logReference.setLogGroup(logGroupFromEnv);
        client.addAllLogReferences(Collections.singleton(logReference));
    }
    return client;
}
Also used : HashMap(java.util.HashMap) ECSPlugin(com.amazonaws.xray.plugins.ECSPlugin) StreamingStrategy(com.amazonaws.xray.strategy.StreamingStrategy) Plugin(com.amazonaws.xray.plugins.Plugin) ArrayList(java.util.ArrayList) RuntimeErrorContextMissingStrategy(com.amazonaws.xray.strategy.RuntimeErrorContextMissingStrategy) HashSet(java.util.HashSet) IgnoreErrorContextMissingStrategy(com.amazonaws.xray.strategy.IgnoreErrorContextMissingStrategy) Map(java.util.Map) StringValidator(com.amazonaws.xray.entities.StringValidator) SegmentListener(com.amazonaws.xray.listeners.SegmentListener) Emitter(com.amazonaws.xray.emitters.Emitter) Nullable(org.checkerframework.checker.nullness.qual.Nullable) SamplingStrategy(com.amazonaws.xray.strategy.sampling.SamplingStrategy) Set(java.util.Set) SegmentContextResolverChain(com.amazonaws.xray.contexts.SegmentContextResolverChain) PrioritizationStrategy(com.amazonaws.xray.strategy.PrioritizationStrategy) ContextMissingStrategy(com.amazonaws.xray.strategy.ContextMissingStrategy) EKSPlugin(com.amazonaws.xray.plugins.EKSPlugin) Objects(java.util.Objects) List(java.util.List) ElasticBeanstalkPlugin(com.amazonaws.xray.plugins.ElasticBeanstalkPlugin) ThrowableSerializationStrategy(com.amazonaws.xray.strategy.ThrowableSerializationStrategy) Optional(java.util.Optional) Log(org.apache.commons.logging.Log) LogFactory(org.apache.commons.logging.LogFactory) LogErrorContextMissingStrategy(com.amazonaws.xray.strategy.LogErrorContextMissingStrategy) Collections(java.util.Collections) AWSLogReference(com.amazonaws.xray.entities.AWSLogReference) EC2Plugin(com.amazonaws.xray.plugins.EC2Plugin) AWSLogReference(com.amazonaws.xray.entities.AWSLogReference) Objects(java.util.Objects) RuntimeErrorContextMissingStrategy(com.amazonaws.xray.strategy.RuntimeErrorContextMissingStrategy) IgnoreErrorContextMissingStrategy(com.amazonaws.xray.strategy.IgnoreErrorContextMissingStrategy) ContextMissingStrategy(com.amazonaws.xray.strategy.ContextMissingStrategy) LogErrorContextMissingStrategy(com.amazonaws.xray.strategy.LogErrorContextMissingStrategy)

Example 24 with Emitter

use of com.amazonaws.xray.emitters.Emitter in project aws-xray-sdk-java by aws.

the class TracingInterceptorTest method setup.

@Before
public void setup() {
    Emitter blankEmitter = Mockito.mock(Emitter.class);
    Mockito.doReturn(true).when(blankEmitter).sendSegment(Mockito.anyObject());
    AWSXRay.setGlobalRecorder(AWSXRayRecorderBuilder.standard().withEmitter(blankEmitter).build());
    AWSXRay.clearTraceEntity();
    AWSXRay.beginSegment("test");
}
Also used : Emitter(com.amazonaws.xray.emitters.Emitter) Before(org.junit.Before)

Aggregations

Emitter (com.amazonaws.xray.emitters.Emitter)24 Test (org.junit.Test)8 BeforeEach (org.junit.jupiter.api.BeforeEach)8 PrepareForTest (org.powermock.core.classloader.annotations.PrepareForTest)8 LocalizedSamplingStrategy (com.amazonaws.xray.strategy.sampling.LocalizedSamplingStrategy)7 Before (org.junit.Before)6 Subsegment (com.amazonaws.xray.entities.Subsegment)4 TraceHeader (com.amazonaws.xray.entities.TraceHeader)3 SegmentContextResolverChain (com.amazonaws.xray.contexts.SegmentContextResolverChain)1 AWSLogReference (com.amazonaws.xray.entities.AWSLogReference)1 Segment (com.amazonaws.xray.entities.Segment)1 StringValidator (com.amazonaws.xray.entities.StringValidator)1 SegmentListener (com.amazonaws.xray.listeners.SegmentListener)1 EC2Plugin (com.amazonaws.xray.plugins.EC2Plugin)1 ECSPlugin (com.amazonaws.xray.plugins.ECSPlugin)1 EKSPlugin (com.amazonaws.xray.plugins.EKSPlugin)1 ElasticBeanstalkPlugin (com.amazonaws.xray.plugins.ElasticBeanstalkPlugin)1 Plugin (com.amazonaws.xray.plugins.Plugin)1 ContextMissingStrategy (com.amazonaws.xray.strategy.ContextMissingStrategy)1 IgnoreErrorContextMissingStrategy (com.amazonaws.xray.strategy.IgnoreErrorContextMissingStrategy)1