mirror of
https://github.com/alerta/alerta.git
synced 2025-02-05 05:59:43 +00:00
30 lines
751 B
Python
30 lines
751 B
Python
import uuid
|
|
|
|
from flask import Flask, request, g
|
|
|
|
|
|
class Tracing:
|
|
|
|
def __init__(self, app=None) -> None:
|
|
self.app = None
|
|
if app:
|
|
self.setup_tracing(app)
|
|
|
|
def setup_tracing(self, app: Flask) -> None:
|
|
app.before_request(self.get_request_id)
|
|
app.after_request(self.set_request_id)
|
|
|
|
@staticmethod
|
|
def get_request_id():
|
|
request_id = (
|
|
request.headers.get('X-Request-ID') or
|
|
request.headers.get('X-Amzn-Trace-Id') or
|
|
str(uuid.uuid4())
|
|
)
|
|
g.request_id = request_id
|
|
|
|
@staticmethod
|
|
def set_request_id(response):
|
|
if hasattr(g, 'request_id'):
|
|
response.headers['X-Request-ID'] = g.request_id
|
|
return response
|