rebase main

Author: JqRrt

CHANGLOG.md

@@ -1,9 +1,13 @@
-## [0.1.22] - 2026-01-05
+## [0.1.23] - 2026-01-07
 ### Added
 - lcc support child_of and state_span_ctx_key
 - lcc support multi clients
 - llc support get trace_id and root_span_id
 
+## [0.1.22] - 2026-01-06
+### Added
+- support span discard
+
 ## [0.1.21] - 2025-12-23
 ### Added
 - runtime scene support get from env

cozeloop/decorator/decorator.py

@@ -101,7 +101,7 @@ def sync_wrapper(*args: Any, **kwargs: Any):
                     span.set_tags(tags)
                     span.finish()
 
-                    if res:
+                    if res is not None:
                         return res
 
             @wraps(func)
@@ -140,7 +140,7 @@ async def async_wrapper(*args: Any, **kwargs: Any):
                     span.set_tags(tags)
                     span.finish()
 
-                    if res:
+                    if res is not None:
                         return res
 
             @wraps(func)
@@ -249,7 +249,7 @@ def sync_stream_wrapper(*args: Any, **kwargs: Any):
                     span.set_input(input)
                     span.set_tags(tags)
 
-                    if not hasattr(res, "__iter__") and res:
+                    if not hasattr(res, "__iter__") and res is not None:
                         return res
 
             @wraps(func)
@@ -288,7 +288,7 @@ async def async_stream_wrapper(*args: Any, **kwargs: Any):
                     span.set_input(input)
                     span.set_tags(tags)
 
-                    if not hasattr(res, "__aiter__") and res:
+                    if not hasattr(res, "__aiter__") and res is not None:
                         return res
 
             if is_async_gen_func(func):
@@ -423,7 +423,7 @@ async def __aiter__(self) -> AsyncIterator[S]:
             await self._aend()
             raise
         except Exception as e:
-            await self._aend()
+            await self._aend(e)
             raise e
         else:
             await self._aend()
@@ -449,7 +449,7 @@ async def __async_streamer__(
                     self.__span.set_start_time_first_resp(time.time_ns() // 1_000)
                     self.__is_set_start_time_first_token = True
                 yield s
-        except StopIteration:
+        except (StopIteration, StopAsyncIteration):
             pass
 
 

cozeloop/internal/trace/noop_span.py

@@ -34,6 +34,9 @@ def set_tags(self, tag_kvs: Dict[str, Any]) -> None:
     def set_baggage(self, baggage_items: Dict[str, str]) -> None:
         pass
 
+    def discard(self) -> None:
+        pass
+
     def finish(self) -> None:
         pass
 

cozeloop/internal/trace/span.py

@@ -490,6 +490,9 @@ def set_baggage_item(self, restricted_key: str, value: str):
         with self.lock:
             super().set_baggage_item(restricted_key, value)
 
+    def discard(self) -> None:
+        delete_span_in_context(self.span_id)
+
     def finish(self):
         try:
             if not self.is_do_finish():

cozeloop/internal/version.py

@@ -1,4 +1,4 @@
 # Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
 # SPDX-License-Identifier: MIT
 
-VERSION = 'v0.1.21'
+VERSION = 'v0.1.23'

cozeloop/span.py

@@ -218,6 +218,12 @@ def finish(self) -> None:
         Under the hood, it is actually placed in an asynchronous queue waiting to be reported.
         """
 
+    @abstractmethod
+    def discard(self) -> None:
+        """
+        The span will be discarded, not be reported.
+        """
+
     @property
     @abstractmethod
     def start_time(self) -> datetime: