From 6a9575518114d1318ea1189dd9e139f83c23e499 Mon Sep 17 00:00:00 2001
From: Travis Rivers <trivers@contractor.usgs.gov>
Date: Mon, 31 Aug 2020 16:18:38 -0600
Subject: [PATCH] add middleware logic

---
 geomagio/api/ws/app.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/geomagio/api/ws/app.py b/geomagio/api/ws/app.py
index 9431fc447..9038d972d 100644
--- a/geomagio/api/ws/app.py
+++ b/geomagio/api/ws/app.py
@@ -31,6 +31,13 @@ app.include_router(elements.router)
 app.include_router(observatories.router)
 
 
+@app.middleware("http")
+async def add_process_time_header(request: Request, call_next):
+    response = await call_next(request)
+    response.headers["X-Process-Time"] = "45"
+    return response
+
+
 @app.get("/", include_in_schema=False)
 async def redirect_to_docs():
     return RedirectResponse("/ws/docs")
-- 
GitLab