From bdd7e2054280cb8b7351a3327f2c57976f2bb39b Mon Sep 17 00:00:00 2001
From: Juan Amari <juan.amari@dialpad.com>
Date: Wed, 13 Mar 2024 14:48:46 -0300
Subject: [PATCH] chore(pubsub): add 20s, 30s buckets to consume_latency prom
 metrics

We're increasing the amount of buckets in
`gcloud_aio_pubsub_subscriber_consume_latency_seconds_bucket` to enable
better tracking of tail end latencies.
---
 pubsub/gcloud/aio/pubsub/metrics.py | 2 ++
 pubsub/pyproject.toml               | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/pubsub/gcloud/aio/pubsub/metrics.py b/pubsub/gcloud/aio/pubsub/metrics.py
index 64e18829f..2acb84a67 100644
--- a/pubsub/gcloud/aio/pubsub/metrics.py
+++ b/pubsub/gcloud/aio/pubsub/metrics.py
@@ -35,6 +35,8 @@
         namespace=_NAMESPACE,
         subsystem=_SUBSYSTEM,
         unit='seconds',
+        buckets=(.005, .01, .025, .05, .075, .1, .25, .5, .75, 1.0, 2.5,
+                 5.0, 7.5, 10.0, 20.0, 30.0, float('inf')),
     )
 
     BATCH_STATUS = prometheus_client.Counter(
diff --git a/pubsub/pyproject.toml b/pubsub/pyproject.toml
index cca1e33d5..f75f6ef86 100644
--- a/pubsub/pyproject.toml
+++ b/pubsub/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "gcloud-aio-pubsub"
-version = "6.0.0"
+version = "6.0.1"
 description = "Python Client for Google Cloud Pub/Sub"
 readme = "README.rst"