From 632356acebe04a660b8dbc14d563ed9bb645fc86 Mon Sep 17 00:00:00 2001
From: Nandita Koppisetty <nandita.iitkgp@gmail.com>
Date: Wed, 16 Nov 2022 13:36:56 -0800
Subject: [PATCH] Update server.py (#26)

---
 pynumaflow/function/server.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/pynumaflow/function/server.py b/pynumaflow/function/server.py
index ab5dbd92..c6817d00 100644
--- a/pynumaflow/function/server.py
+++ b/pynumaflow/function/server.py
@@ -65,10 +65,14 @@ def MapFn(
         Applies a function to each datum element.
         The pascal case function name comes from the generated udfunction_pb2_grpc.py file.
         """
+        _LOGGER.debug("Starting MapFn for request: %s", request.value)
+
         key = ""
         for metadata_key, metadata_value in context.invocation_metadata():
             if metadata_key == DATUM_KEY:
                 key = metadata_value
+        _LOGGER.debug("MapFn Key: %s", key)
+
 
         try:
             msgs = self.__map_handler(
@@ -88,7 +92,8 @@ def MapFn(
         datums = []
         for msg in msgs.items():
             datums.append(udfunction_pb2.Datum(key=msg.key, value=msg.value))
-
+        
+        _LOGGER.debug("MapFn Msgs: %s", msgs)
         return udfunction_pb2.DatumList(elements=datums)
 
     def ReduceFn(
@@ -110,6 +115,7 @@ def IsReady(
         IsReady is the heartbeat endpoint for gRPC.
         The pascal case function name comes from the generated udfunction_pb2_grpc.py file.
         """
+        _LOGGER.debug("I am ready")
         return udfunction_pb2.ReadyResponse(ready=True)
 
     async def __serve(self) -> None: