From 8308be3e49db1925b0219080b8f9dd9b0fe84ae1 Mon Sep 17 00:00:00 2001
From: Jonas Linter <{email_address}>
Date: Wed, 19 Nov 2025 20:53:27 +0100
Subject: [PATCH] Increased timeout limit

---
 Dockerfile                        |  3 ++-
 src/alpine_bits_python/run_api.py | 10 ++++++++++
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index eccbf66..4e6792f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -66,4 +66,5 @@ CMD python -m alpine_bits_python.run_api \
     --access-log \
     --forwarded-allow-ips "${FORWARDED_ALLOW_IPS:-127.0.0.1}" \
     --proxy-headers \
-    --no-server-header
\ No newline at end of file
+    --no-server-header \
+    --timeout-graceful-shutdown 300
\ No newline at end of file
diff --git a/src/alpine_bits_python/run_api.py b/src/alpine_bits_python/run_api.py
index 6975a34..c07185e 100644
--- a/src/alpine_bits_python/run_api.py
+++ b/src/alpine_bits_python/run_api.py
@@ -81,6 +81,15 @@ def parse_args() -> argparse.Namespace:
         default=False,
         help="Disable Server header in responses (default: False)",
     )
+    parser.add_argument(
+        "--timeout-graceful-shutdown",
+        type=int,
+        default=300,
+        help=(
+            "Graceful shutdown timeout in seconds. Workers have this long to finish "
+            "background tasks before being killed (default: 300)"
+        ),
+    )
 
     return parser.parse_args()
 
@@ -112,4 +121,5 @@ if __name__ == "__main__":
         forwarded_allow_ips=args.forwarded_allow_ips,
         proxy_headers=args.proxy_headers,
         server_header=not args.no_server_header,
+        timeout_graceful_shutdown=args.timeout_graceful_shutdown,
     )