Replaced internal encoders with message encoder calling public API

packages/service/ni_measurement_plugin_sdk_service/_internal/grpc_servicer.py

@@ -13,8 +13,10 @@
 import grpc
 from google.protobuf import any_pb2
 
-from ni_measurement_plugin_sdk_service._internal.parameter import serializer
-from ni_measurement_plugin_sdk_service._internal.parameter.metadata import ParameterMetadata
+from ni_measurement_plugin_sdk_service._internal.parameter import decoder, encoder
+from ni_measurement_plugin_sdk_service._internal.parameter.metadata import (
+    ParameterMetadata,
+)
 from ni_measurement_plugin_sdk_service._internal.stubs.ni.measurementlink.measurement.v1 import (
     measurement_service_pb2 as v1_measurement_service_pb2,
     measurement_service_pb2_grpc as v1_measurement_service_pb2_grpc,
@@ -23,7 +25,10 @@
     measurement_service_pb2 as v2_measurement_service_pb2,
     measurement_service_pb2_grpc as v2_measurement_service_pb2_grpc,
 )
-from ni_measurement_plugin_sdk_service.measurement.info import MeasurementInfo
+from ni_measurement_plugin_sdk_service.measurement.info import (
+    MeasurementInfo,
+    ServiceInfo,
+)
 from ni_measurement_plugin_sdk_service.session_management import PinMapContext
 
 
@@ -131,9 +136,13 @@ def _get_mapping_by_parameter_name(
     return mapping_by_variable_name
 
 
-def _serialize_outputs(output_metadata: Dict[int, ParameterMetadata], outputs: Any) -> any_pb2.Any:
+def _serialize_outputs(
+    output_metadata: Dict[int, ParameterMetadata], outputs: Any, service_name: str
+) -> any_pb2.Any:
     if isinstance(outputs, collections.abc.Sequence):
-        return any_pb2.Any(value=serializer.serialize_parameters(output_metadata, outputs))
+        return any_pb2.Any(
+            value=encoder.serialize_parameters(output_metadata, outputs, service_name)
+        )
     elif outputs is None:
         raise ValueError(f"Measurement function returned None")
     else:
@@ -161,6 +170,7 @@ def __init__(
         output_parameter_list: List[ParameterMetadata],
         measure_function: Callable,
         owner: object,
+        service_info: ServiceInfo,
     ) -> None:
         """Initialize the measurement v1 servicer."""
         super().__init__()
@@ -169,6 +179,7 @@ def __init__(
         self._measurement_info = measurement_info
         self._measure_function = measure_function
         self._owner = weakref.ref(owner) if owner is not None else None  # avoid reference cycle
+        self._service_info = service_info
 
     def GetMetadata(  # noqa: N802 - function name should be lowercase
         self, request: v1_measurement_service_pb2.GetMetadataRequest, context: grpc.ServicerContext
@@ -193,8 +204,8 @@ def GetMetadata(  # noqa: N802 - function name should be lowercase
             )
             measurement_signature.configuration_parameters.append(configuration_parameter)
 
-        measurement_signature.configuration_defaults.value = serializer.serialize_default_values(
-            self._configuration_metadata
+        measurement_signature.configuration_defaults.value = encoder.serialize_default_values(
+            self._configuration_metadata, self._get_service_name() + ".Inputs"
         )
 
         for field_number, output_metadata in self._output_metadata.items():
@@ -224,8 +235,10 @@ def Measure(  # noqa: N802 - function name should be lowercase
         self, request: v1_measurement_service_pb2.MeasureRequest, context: grpc.ServicerContext
     ) -> v1_measurement_service_pb2.MeasureResponse:
         """RPC API that executes the registered measurement method."""
-        mapping_by_id = serializer.deserialize_parameters(
-            self._configuration_metadata, request.configuration_parameters.value
+        mapping_by_id = decoder.deserialize_parameters(
+            self._configuration_metadata,
+            request.configuration_parameters.value,
+            self._get_service_name() + ".Inputs",
         )
         mapping_by_variable_name = _get_mapping_by_parameter_name(
             mapping_by_id, self._measure_function
@@ -254,9 +267,15 @@ def Measure(  # noqa: N802 - function name should be lowercase
 
     def _serialize_response(self, outputs: Any) -> v1_measurement_service_pb2.MeasureResponse:
         return v1_measurement_service_pb2.MeasureResponse(
-            outputs=_serialize_outputs(self._output_metadata, outputs)
+            outputs=_serialize_outputs(
+                self._output_metadata, outputs, self._get_service_name() + ".Outputs"
+            )
         )
 
+    def _get_service_name(self) -> str:
+        service_name = "".join(char for char in self._service_info.service_class if char.isalpha())
+        return service_name
+
 
 class MeasurementServiceServicerV2(v2_measurement_service_pb2_grpc.MeasurementServiceServicer):
     """Measurement v2 servicer."""
@@ -268,6 +287,7 @@ def __init__(
         output_parameter_list: List[ParameterMetadata],
         measure_function: Callable,
         owner: object,
+        service_info: ServiceInfo,
     ) -> None:
         """Initialize the measurement v2 servicer."""
         super().__init__()
@@ -276,6 +296,7 @@ def __init__(
         self._measurement_info = measurement_info
         self._measure_function = measure_function
         self._owner = weakref.ref(owner) if owner is not None else None  # avoid reference cycle
+        self._service_info = service_info
 
     def GetMetadata(  # noqa: N802 - function name should be lowercase
         self, request: v2_measurement_service_pb2.GetMetadataRequest, context: grpc.ServicerContext
@@ -301,8 +322,8 @@ def GetMetadata(  # noqa: N802 - function name should be lowercase
             )
             measurement_signature.configuration_parameters.append(configuration_parameter)
 
-        measurement_signature.configuration_defaults.value = serializer.serialize_default_values(
-            self._configuration_metadata
+        measurement_signature.configuration_defaults.value = encoder.serialize_default_values(
+            self._configuration_metadata, self._get_service_name() + ".Inputs"
         )
 
         for field_number, output_metadata in self._output_metadata.items():
@@ -334,8 +355,10 @@ def Measure(  # noqa: N802 - function name should be lowercase
         self, request: v2_measurement_service_pb2.MeasureRequest, context: grpc.ServicerContext
     ) -> Generator[v2_measurement_service_pb2.MeasureResponse, None, None]:
         """RPC API that executes the registered measurement method."""
-        mapping_by_id = serializer.deserialize_parameters(
-            self._configuration_metadata, request.configuration_parameters.value
+        mapping_by_id = decoder.deserialize_parameters(
+            self._configuration_metadata,
+            request.configuration_parameters.value,
+            self._get_service_name() + ".Inputs",
         )
         mapping_by_variable_name = _get_mapping_by_parameter_name(
             mapping_by_id, self._measure_function
@@ -363,5 +386,11 @@ def Measure(  # noqa: N802 - function name should be lowercase
 
     def _serialize_response(self, outputs: Any) -> v2_measurement_service_pb2.MeasureResponse:
         return v2_measurement_service_pb2.MeasureResponse(
-            outputs=_serialize_outputs(self._output_metadata, outputs)
+            outputs=_serialize_outputs(
+                self._output_metadata, outputs, self._get_service_name() + ".Outputs"
+            )
         )
+
+    def _get_service_name(self) -> str:
+        service_name = "".join(char for char in self._service_info.service_class if char.isalpha())
+        return service_name