Allow user to specify route key

curl -H 'x-route-key: POST /path/{proxy+}' ...
amancevice · Apr 25, 2021 · 9a7d090 · 9a7d090
1 parent bf4d9d9
commit 9a7d090
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lambda_gateway/request_handler.py b/lambda_gateway/request_handler.py
@@ -63,10 +63,11 @@ def get_event_v2(self, httpMethod):
         """
         url = parse.urlparse(self.path)
         path, *_ = url.path.split('?')
+        route_key = self.headers.get('x-route-key') or f'{httpMethod} {path}'
         return {
             'version': '2.0',
             'body': self.get_body(),
-            'routeKey': f'{httpMethod} {path}',
+            'routeKey': route_key,
             'rawPath': path,
             'rawQueryString': url.query,
             'headers': dict(self.headers),