reach_commons 0.18.23__py3-none-any.whl → 0.18.25__py3-none-any.whl

reach_commons/reach_aws/sqs.py

@@ -146,6 +146,99 @@ class SQSClient(BaseSQSClient):
 
         return True
 
+    def get_visible_messages_count(self) -> int:
+        try:
+            resp = self.client.get_queue_attributes(
+                QueueUrl=self.topic_name,
+                AttributeNames=[
+                    "ApproximateNumberOfMessages",
+                    "ApproximateNumberOfMessagesNotVisible",
+                    "ApproximateNumberOfMessagesDelayed",
+                ],
+            )
+            visible = int(resp["Attributes"].get("ApproximateNumberOfMessages", 0))
+            in_flight = int(
+                resp["Attributes"].get("ApproximateNumberOfMessagesNotVisible", 0)
+            )
+            delayed = int(
+                resp["Attributes"].get("ApproximateNumberOfMessagesDelayed", 0)
+            )
+
+            return visible, in_flight, delayed
+        except ClientError as exc:
+            self.logger.error(
+                "error_fetching_queue_attributes, topic_name=%s, error=%s",
+                self.topic_name,
+                str(exc),
+            )
+            # If there's an error, return 0 so as not to block the caller.
+            return 0, 0, 0
+
+    def publish_with_capacity_guard(
+        self,
+        message_data: dict,
+        max_capacity: int,
+        delay_seconds: int = 60,
+        message_attributes: dict = None,
+    ) -> dict:
+        """
+        Attempts to publish a message while enforcing an approximate queue capacity limit.
+
+        Behavior:
+            - If the queue is below `max_capacity`, the message is published normally.
+            - If the queue is at or above `max_capacity`, the message is NOT published.
+
+        Returns a dict with:
+            - posted: bool        -> whether the message was actually published
+            - visible_count: int  -> ApproximateNumberOfMessages at publication time
+            - max_capacity: int   -> the capacity threshold used
+
+        Example:
+            sqs = SQSClient(topic_name="<enviroment>-messages-queue")
+
+            Result = sqs.publish_with_capacity_guard(
+                message_data={"foo": "bar"},
+                max_capacity=100_000,
+                delay_seconds=30,
+                message_attributes={"service_name": "TestClient"},
+            )
+
+            if not Result["posted"]:
+                print("Queue is over capacity:", Result)
+        """
+        (
+            visible_count,
+            in_flight_count,
+            delayed_count,
+        ) = self.get_visible_messages_count()
+
+        if visible_count >= max_capacity:
+            self.logger.warning(
+                "queue_over_capacity, topic_name=%s, visible=%s, max_capacity=%s",
+                self.topic_name,
+                visible_count,
+                max_capacity,
+            )
+            return {
+                "posted": False,
+                "visible_count": visible_count,
+                "max_capacity": max_capacity,
+            }
+
+        self.publish(
+            message_data=message_data,
+            delay_seconds=delay_seconds,
+            message_attributes=message_attributes,
+        )
+
+        return {
+            "posted": True,
+            "visible_count": visible_count,
+            "in_flight_count": in_flight_count,
+            "delayed_count": delayed_count,
+            "max_capacity": max_capacity,
+        }
+
 
 class MessageFilter:
     """

{reach_commons-0.18.23.dist-info → reach_commons-0.18.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reach_commons
-Version: 0.18.23
+Version: 0.18.25
 Summary: Reach Commons is a versatile utility library designed to streamline and enhance development workflows within the Reach ecosystem.
 License: MIT
 Author: Engineering

{reach_commons-0.18.23.dist-info → reach_commons-0.18.25.dist-info}/RECORD

@@ -21,12 +21,12 @@ reach_commons/reach_aws/exceptions.py,sha256=x0RL5ktNtzxg0KykhEVWReBq_dEtciK6B2v
 reach_commons/reach_aws/firehose.py,sha256=1xFKLWMv3bNo3PPW5gtaL6NqzUDyVil6B768slj2wbY,5674
 reach_commons/reach_aws/kms.py,sha256=ZOfyJMQUgxJEojRoB7-aCxtATpNx1Ig522IUYH11NZ4,4678
 reach_commons/reach_aws/s3.py,sha256=2MLlDNFx0SROJBpE_KjJefyrB7lMqTlrYuRhSZx4iKs,3945
-reach_commons/reach_aws/sqs.py,sha256=LKQ42vRJko2HTE8gO_lRvftnstIMI3DeFZakvLUM0BQ,15864
+reach_commons/reach_aws/sqs.py,sha256=lxdBzvCZ_cUy29se5LhUrIenPWdJcPHtRtFXazzhg-A,19214
 reach_commons/reach_base_model.py,sha256=vgdGDcZr3iXMmyRhmBhOf_LKWB_6QzT3r_Yiyo6OmEk,3009
 reach_commons/redis_manager.py,sha256=SgUtdtt0eV4bUwsWDankIa9Bjfgcm2DKcmVMQT6ptF0,2946
 reach_commons/sms_smart_encoding.py,sha256=92y0RmZ0l4ONHpC9qeO5KfViSNq64yE2rc7lhNDSZqE,1241
 reach_commons/utils.py,sha256=dMgKIGqTgoSItuBI8oz81gKtW3qi21Jkljv9leS_V88,8475
 reach_commons/validations.py,sha256=CJdWg8qc9ILe3rYHYTLdr2upRspsM_8ghhm4IXX4Pg8,1028
-reach_commons-0.18.23.dist-info/METADATA,sha256=rNrFAbeMVQa6i0ALab7iTY7VDXuPI5tF794nCu_PPJM,1965
-reach_commons-0.18.23.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-reach_commons-0.18.23.dist-info/RECORD,,
+reach_commons-0.18.25.dist-info/METADATA,sha256=CBq0zcdO1OOmh8Eq76u6Fa6ECC55JEtU5L3-F5C0Wa8,1965
+reach_commons-0.18.25.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+reach_commons-0.18.25.dist-info/RECORD,,