fix

Author: JqRrt

cozeloop/internal/consts/__init__.py

@@ -39,6 +39,7 @@
 THREAD_ID = "thread_id"
 START_TIME_FIRST_RESP = "start_time_first_resp"
 LATENCY_FIRST_RESP = "latency_first_resp"
+DEPLOYMENT_ENV = "deployment_env"
 CUT_OFF = "cut_off"
 
 # ReserveFieldTypes Define the allowed types for each reserved field.

cozeloop/internal/trace/noop_span.py

@@ -119,6 +119,9 @@ def set_log_id(self, log_id: str) -> None:
     def set_system_tags(self, system_tags: Dict[str, Any]) -> None:
         pass
 
+    def set_deployment_env(self, deployment_env: str) -> None:
+        pass
+
     def __enter__(self):
         return self
 

cozeloop/internal/trace/span.py

@@ -383,6 +383,9 @@ def set_system_tags(self, system_tags: Dict[str, Any]) -> None:
             for key, value in system_tags.items():
                 self.system_tag_map[key] = value
 
+    def set_deployment_env(self, deployment_env: str) -> None:
+        self.set_tags({DEPLOYMENT_ENV: deployment_env})
+
     def get_rectified_map(self, input_map: Dict[str, Any]) -> (Dict[str, Any], List[str], int):
         validate_map = {}
         cut_off_keys = []

cozeloop/span.py

@@ -186,6 +186,11 @@ def set_system_tags(self, system_tags: Dict[str, Any]) -> None:
         Set system tags. DO NOT use this method unless you know what you are doing.
         """
 
+    def set_deployment_env(self, deployment_env: str) -> None:
+        """
+        Set the deployment environment of the span, identify custom environments.
+        """
+
 
 class Span(CommonSpanSetter, SpanContext):
     """