chore: add server timeout

.gitignore

@@ -27,6 +27,11 @@ var/
 .installed.cfg
 *.egg
 
+# pyenv
+.python-version
+.vscode/
+.env
+
 # PyInstaller
 #  Usually these files are written by a python script from a template
 #  before PyInstaller builds the exe, so as to inject date/other infos into it.

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Pythie serving
 
+## 3.0.0.dev1
+## Change
+* Add GRPC server timeout 
+
 ## 3.0.0
 ## Change
 * Upgrade sklearn version to 1.2.0

VERSION

@@ -1 +1 @@
-3.0.0
+3.0.0.dev1

src/pythie_serving/run.py

@@ -1,3 +1,4 @@
+import logging
 import os
 import sys
 from argparse import ArgumentParser
@@ -85,8 +86,16 @@ def run():
         maximum_concurrent_rpcs=maximum_concurrent_rpcs,
         port=ns.port,
     )
+
+    logger = logging.getLogger("pythie_serving")
+
     server.start()
-    server.wait_for_termination()
+
+    timeout = os.environ.get("GRPC_SERVER_WAIT_TIMEOUT", None)
+    # wait _for_termination() returns True if a timeout was supplied and it was reached. False otherwise.
+    timeout_reached = server.wait_for_termination(timeout=float(timeout) if timeout is not None else None)
+    if timeout_reached:
+        logger.warning(f"Server shut down after timeout of {timeout} was reached")
 
 
 if __name__ == "__main__":