crossplane
/
function-sdk-python
mirror of https://github.com/crossplane/function-sdk-python.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Shutdown gracefully on SIGTERM 

Signed-off-by: Nic Cope <nicc@rk0n.org>
This commit is contained in:
Nic Cope 2025-04-01 07:00:44 -07:00
parent fcec201f01
commit cdaf6697c5
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
crossplane/function/runtime.py
View File

@ -16,6 +16,7 @@
import asyncio import asyncio
import os import os
import signal
import grpc import grpc
from grpc_reflection.v1alpha import reflection from grpc_reflection.v1alpha import reflection
@ -31,6 +32,8 @@ SERVICE_NAMES = (
fnv1beta1.DESCRIPTOR.services_by_name["FunctionRunnerService"].full_name, fnv1beta1.DESCRIPTOR.services_by_name["FunctionRunnerService"].full_name,
) )
SHUTDOWN_GRACE_PERIOD_SECONDS = 5
def load_credentials(tls_certs_dir: str) -> grpc.ServerCredentials: def load_credentials(tls_certs_dir: str) -> grpc.ServerCredentials:
"""Load TLS credentials for a composition function gRPC server. """Load TLS credentials for a composition function gRPC server.
@ -90,6 +93,11 @@ def serve(
server = grpc.aio.server() server = grpc.aio.server()
loop.add_signal_handler(
signal.SIGTERM,
lambda: asyncio.ensure_future(server.stop(grace=SHUTDOWN_GRACE_PERIOD_SECONDS)),
)
grpcv1.add_FunctionRunnerServiceServicer_to_server(function, server) grpcv1.add_FunctionRunnerServiceServicer_to_server(function, server)
grpcv1beta1.add_FunctionRunnerServiceServicer_to_server( grpcv1beta1.add_FunctionRunnerServiceServicer_to_server(
BetaFunctionRunner(wrapped=function), server BetaFunctionRunner(wrapped=function), server
@ -116,7 +124,7 @@ def serve(
try: try:
loop.run_until_complete(start()) loop.run_until_complete(start())
finally: finally:
loop.run_until_complete(server.stop(grace=5)) loop.run_until_complete(server.stop(grace=SHUTDOWN_GRACE_PERIOD_SECONDS))
loop.close() loop.close()