add consumer_member_id tag to partitions assign / revoked metrics

getsentry · Apr 22, 2024 · b5e426f · b5e426f
1 parent 8a90458
commit b5e426f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/arroyo/processing/processor.py b/arroyo/processing/processor.py
@@ -235,7 +235,7 @@ def on_partitions_assigned(partitions: Mapping[Partition, int]) -> None:
             logger.info("New partitions assigned: %r", partitions)
             logger.info("Member id: %r", self.__consumer.member_id)
             self.__metrics_buffer.metrics.increment(
-                "arroyo.consumer.partitions_assigned.count", len(partitions)
+                "arroyo.consumer.partitions_assigned.count", len(partitions), tags={"consumer_member_id": self.__consumer.member_id}
             )
 
             self.__buffered_messages.reset()
@@ -254,7 +254,7 @@ def on_partitions_revoked(partitions: Sequence[Partition]) -> None:
             logger.info("Partitions to revoke: %r", partitions)
 
             self.__metrics_buffer.metrics.increment(
-                "arroyo.consumer.partitions_revoked.count", len(partitions)
+                "arroyo.consumer.partitions_revoked.count", len(partitions), tags={"consumer_member_id": self.__consumer.member_id}
             )
 
             if partitions: