notifications-api/app/statsd_decorators.py

import functools

from app import statsd_client
from flask import current_app
from monotonic import monotonic


def statsd(namespace):
    def time_function(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            start_time = monotonic()
            try:
                res = func(*args, **kwargs)
                elapsed_time = monotonic() - start_time
                statsd_client.incr('{namespace}.{func}'.format(
                    namespace=namespace, func=func.__name__)
                )
                statsd_client.timing('{namespace}.{func}'.format(
                    namespace=namespace, func=func.__name__), elapsed_time
                )

            except Exception as e:
                current_app.logger.error(
                    "{namespace} call {func} failed".format(
                        namespace=namespace, func=func.__name__
                    )
                )
                raise e
            else:
                current_app.logger.info(
                    "{namespace} call {func} took {time}".format(
                        namespace=namespace, func=func.__name__, time="{0:.4f}".format(elapsed_time)
                    )
                )
                return res
        wrapper.__wrapped__.__name__ = func.__name__
        return wrapper

    return time_function