wip perfetto

Author: brian14708

pyproject.toml

@@ -9,6 +9,11 @@ dependencies = [
     "typing-extensions>=4.15.0",
 ]
 
+[project.optional-dependencies]
+perfetto = [
+    "perfetto>=0.14.0",
+]
+
 [build-system]
 requires = ["uv_build>=0.8.19,<0.9.0"]
 build-backend = "uv_build"
@@ -25,12 +30,14 @@ test = [
     "pytest-asyncio",
     "pytest-codspeed",
     "pytest-xdist",
+    "duron[perfetto]",
 ]
 typing = [
     "basedmypy",
     "basedpyright",
     "pydantic",
     "pytest",
+    "duron[perfetto]",
 ]
 lint = [
     "ruff",
@@ -87,6 +94,10 @@ disable_error_code = [
     "any",
 ]
 
+[[tool.mypy.overrides]]
+module = ["perfetto.*"]
+follow_untyped_imports = true
+
 [tool.pyright]
 include = ["src", "tests"]
 venv = "."

src/duron/contrib/perfetto.py

@@ -0,0 +1,19 @@
+from typing import IO
+
+from perfetto.protos.perfetto.trace.perfetto_trace_pb2 import Trace, TracePacket
+
+
+class StreamingTraceProtoBuilder:
+    def __init__(self, file: IO[bytes]) -> None:
+        self._file = file
+        self._trace = Trace()
+
+    @staticmethod
+    def create_packet() -> TracePacket:
+        return TracePacket()
+
+    def write_packet(self, packet: TracePacket) -> None:
+        # Clear the packet list.
+        del self._trace.packet[:]
+        self._trace.packet.append(packet)
+        self._file.write(self._trace.SerializeToString())