✨ Allow specifying the thread counts.

Author: coderanger

src/kubernetes_wsgi/__init__.py

@@ -1 +1 @@
-__version__ = "1.1"
+__version__ = "1.2"

src/kubernetes_wsgi/__main__.py

@@ -46,12 +46,28 @@ def parse_args(argv: Sequence[Text]) -> Dict[str, Any]:
         default="/healthz",
         help="URL path to the health check endpoint",
     )
+    parser.add_argument(
+        "--min-threads",
+        metavar="N",
+        type=int,
+        default=5,
+        help="Minimum number of threads to run",
+    )
+    parser.add_argument(
+        "--max-threads",
+        metavar="N",
+        type=int,
+        default=20,
+        help="Maximum number of threads to run",
+    )
     args = parser.parse_args(argv)
     return {
         "application": args.application,
         "port": args.port,
         "metrics_port": args.metrics_port,
         "health_check_path": args.health_check_path,
+        "min_threads": args.min_threads,
+        "max_threads": args.max_threads,
     }
 
 

src/kubernetes_wsgi/server.py

@@ -51,6 +51,8 @@ def serve(
     metrics_port: int = 9000,
     access_log_formatter: LogFormatter = proxiedLogFormatter,
     health_check_path: str = "/healthz",
+    min_threads: int = 5,
+    max_threads: int = 20,
 ):
     # Quiet the Twisted factory logging.
     Factory.noisy = False
@@ -63,7 +65,9 @@ def serve(
     )
 
     # Create the server.
-    pool = threadpool.ThreadPool()
+    pool = threadpool.ThreadPool(
+        minthreads=min_threads, maxthreads=max_threads
+    )
     reactor.callWhenRunning(pool.start)
     _listen_wsgi(
         reactor,