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

reach_commons/reach_aws/sqs.py

@@ -146,13 +146,25 @@ class SQSClient(BaseSQSClient):
 
         return True
 
-    def _get_visible_messages_count(self) -> int:
+    def get_visible_messages_count(self) -> int:
         try:
             resp = self.client.get_queue_attributes(
                 QueueUrl=self.topic_name,
-                AttributeNames=["ApproximateNumberOfMessages"],
+                AttributeNames=[
+                    "ApproximateNumberOfMessages",
+                    "ApproximateNumberOfMessagesNotVisible",
+                    "ApproximateNumberOfMessagesDelayed",
+                ],
             )
-            return int(resp["Attributes"].get("ApproximateNumberOfMessages", 0))
+            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",
@@ -160,7 +172,7 @@ class SQSClient(BaseSQSClient):
                 str(exc),
             )
             # If there's an error, return 0 so as not to block the caller.
-            return 0
+            return 0, 0, 0
 
     def publish_with_capacity_guard(
         self,
@@ -194,7 +206,11 @@ class SQSClient(BaseSQSClient):
             if not Result["posted"]:
                 print("Queue is over capacity:", Result)
         """
-        visible_count = self._get_visible_messages_count()
+        (
+            visible_count,
+            in_flight_count,
+            delayed_count,
+        ) = self.get_visible_messages_count()
 
         if visible_count >= max_capacity:
             self.logger.warning(
@@ -218,6 +234,8 @@ class SQSClient(BaseSQSClient):
         return {
             "posted": True,
             "visible_count": visible_count,
+            "in_flight_count": in_flight_count,
+            "delayed_count": delayed_count,
             "max_capacity": max_capacity,
         }
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reach_commons
-Version: 0.18.24
+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.24.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=9_sVM7bqhswBSksJ2bhPvtDmW1N9jHh1OUC25vXYLdU,18563
+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.24.dist-info/METADATA,sha256=udH-_sPBf_Cvuyx34M5z16tOxlnu_eLdiMpdONMTmag,1965
-reach_commons-0.18.24.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-reach_commons-0.18.24.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,,