Skip to content

Commit a43164a

Browse files
committed
Avoid invoking UDF while shutting down
Signed-off-by: Sreekanth <prsreekanth920@gmail.com>
1 parent 977f787 commit a43164a

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

packages/pynumaflow/pynumaflow/mapper/_servicer/_sync_servicer.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,6 @@ def MapFn(
6262
# if error handler accordingly
6363
if isinstance(res, BaseException):
6464
err_msg = f"{ERR_UDF_EXCEPTION_STRING}: {repr(res)}"
65-
_LOGGER.critical(err_msg, exc_info=True)
6665
update_context_err(context, res, err_msg)
6766
# Unblock the reader thread if it is waiting on queue.put()
6867
result_queue.close()
@@ -132,6 +131,11 @@ def _invoke_map(
132131
result_queue: SyncIterator,
133132
):
134133
try:
134+
# Skip processing if a shutdown is already in progress
135+
# (e.g. a prior invocation raised an exception)
136+
if self.shutdown_event.is_set():
137+
return
138+
135139
(user_metadata, system_metadata) = _user_and_system_metadata_from_proto(
136140
request.request.metadata
137141
)

0 commit comments

Comments
 (0)