perf: Replace MessageToDict with optimized custom dict builder

Signed-off-by: abhijeet-dhumal <abhijeetdhumal652@gmail.com>
franciscojavierarceo · May 28, 2026 · Feb 25, 2026 · Mar 2, 2026 · Mar 10, 2026 · Mar 10, 2026
commit b28ddf9bd362a6636621ac0244b28cd7d71a2cd2
@@ -38,9 +38,8 @@
 )
 from fastapi.concurrency import run_in_threadpool
 from fastapi.logger import logger
-from fastapi.responses import JSONResponse
+from fastapi.responses import JSONResponse, ORJSONResponse
 from fastapi.staticfiles import StaticFiles
-from google.protobuf.json_format import MessageToDict
 from pydantic import BaseModel, field_validator
 
 import feast
@@ -52,6 +51,7 @@
     FeastError,
 )
 from feast.feast_object import FeastObject
+from feast.feature_server_utils import response_to_dict_fast
 from feast.feature_view_utils import get_feature_view_from_feature_store
 from feast.permissions.action import WRITE, AuthzedAction
 from feast.permissions.security_manager import assert_permissions
@@ -391,13 +391,8 @@ async def get_online_features(request: GetOnlineFeaturesRequest) -> Any:
                     lambda: store.get_online_features(**read_params)  # type: ignore
                 )
 
-            response_dict = await run_in_threadpool(
-                MessageToDict,
-                response.proto,
-                preserving_proto_field_name=True,
-                float_precision=18,
-            )
-            return response_dict
+            response_dict = await run_in_threadpool(response_to_dict_fast, response.proto)
+            return ORJSONResponse(content=response_dict)
 
     @app.post(
         "/retrieve-online-documents",
@@ -433,13 +428,8 @@ async def retrieve_online_documents(
                     lambda: store.retrieve_online_documents(**read_params)  # type: ignore
                 )
 
-            response_dict = await run_in_threadpool(
-                MessageToDict,
-                response.proto,
-                preserving_proto_field_name=True,
-                float_precision=18,
-            )
-            return response_dict
+            response_dict = await run_in_threadpool(response_to_dict_fast, response.proto)
+            return ORJSONResponse(content=response_dict)
 
     @app.post("/push", dependencies=[Depends(inject_user_details)])
     async def push(request: PushFeaturesRequest) -> Response:

@@ -0,0 +1,79 @@
+"""Fast serialization utilities for Feature Server responses.
+
+Matches the output format of MessageToDict with proto_json.patch() applied.
+Values are serialized as native Python types (not wrapped dicts).
+"""
+
+from datetime import datetime, timezone
+from typing import Any, Dict, Optional
+
+from feast.protos.feast.serving.ServingService_pb2 import GetOnlineFeaturesResponse
+from feast.protos.feast.types.Value_pb2 import Value
+
+# FieldStatus enum mapping (protos/feast/serving/ServingService.proto)
+_STATUS_NAMES: Dict[int, str] = {
+    0: "INVALID",
+    1: "PRESENT",
+    2: "NULL_VALUE",
+    3: "NOT_FOUND",
+    4: "OUTSIDE_MAX_AGE",
+}
+
+
+def response_to_dict_fast(response: GetOnlineFeaturesResponse) -> Dict[str, Any]:
+    """Convert GetOnlineFeaturesResponse to dict (matches proto_json.patch() format)."""
+    result: Dict[str, Any] = {
+        "results": [
+            {
+                "values": [_value_to_native(v) for v in feature_vector.values],
+                "statuses": [
+                    _STATUS_NAMES.get(s, "INVALID") for s in feature_vector.statuses
+                ],
+                **(
+                    {
+                        "event_timestamps": [
+                            _timestamp_to_str(ts)
+                            for ts in feature_vector.event_timestamps
+                        ]
+                    }
+                    if feature_vector.event_timestamps
+                    else {}
+                ),
+            }
+            for feature_vector in response.results
+        ]
+    }
+
+    if response.HasField("metadata"):
+        result["metadata"] = _metadata_to_dict(response.metadata)
+
+    return result
+
+
+def _value_to_native(v: Value) -> Optional[Any]:
+    """Convert a Value proto to native Python type (matches proto_json.patch() format)."""
+    which = v.WhichOneof("val")
+    if which is None or which == "null_val":
+        return None
+    elif "_list_" in which:
+        return list(getattr(v, which).val)
+    else:
+        return getattr(v, which)
+
+
+def _timestamp_to_str(ts) -> str:
+    """Convert protobuf Timestamp to RFC 3339 format with Z suffix."""
+    if ts.seconds == 0 and ts.nanos == 0:
+        return "1970-01-01T00:00:00Z"
+    dt = datetime.fromtimestamp(ts.seconds + ts.nanos / 1e9, tz=timezone.utc)
+    return dt.strftime("%Y-%m-%dT%H:%M:%S") + (
+        ".%06dZ" % (ts.nanos // 1000) if ts.nanos else "Z"
+    )
+
+
+def _metadata_to_dict(metadata) -> Dict[str, Any]:
+    """Convert FeatureResponseMeta to dict (matches proto_json.patch() format)."""
+    result: Dict[str, Any] = {}
+    if metadata.HasField("feature_names"):
+        result["feature_names"] = list(metadata.feature_names.val)
+    return result