"""Setup Datadog tags."""
from switchboard_consumer.connectors.sentry import sentry_client


def setup_datadog_tags(tracer, message, logger):
    """Setup Datadog tags."""
    try:
        current_span = tracer.current_span()
        if current_span:
            current_span.set_tag('correlationId', message.correlation_id)
            current_span.set_tag('entityType', message.entity_type)
            current_span.set_tag('messageType', message.message_type)
        logger.info('Setup tags for Datadog.')
    except AttributeError:
        if sentry_client:
            sentry_client.captureException()
        logger.warning('Failed to setup tags for Datadog.')