Examples with Emitter com.amazonaws.xray.emitters.Emitter used on opensource projects

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();
}

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();
}

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;
}

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");
}