atlan-application-sdk 0.1.1rc45__py3-none-any.whl → 0.1.1rc46__py3-none-any.whl

application_sdk/activities/lock_management.py

@@ -5,15 +5,15 @@ allowing the workflow to orchestrate locking without hitting Temporal's
 deadlock timeout.
 """
 
-import asyncio
 import random
 from typing import Any, Dict
 
 from temporalio import activity
+from temporalio.exceptions import ApplicationError
 
 from application_sdk.clients.redis import RedisClientAsync
 from application_sdk.common.error_codes import ActivityError
-from application_sdk.constants import APPLICATION_NAME, LOCK_RETRY_INTERVAL
+from application_sdk.constants import APPLICATION_NAME
 from application_sdk.observability.logger_adaptor import get_logger
 
 logger = get_logger(__name__)
@@ -45,40 +45,41 @@ async def acquire_distributed_lock(
     """
     # Input validation
     if max_locks <= 0:
-        raise ActivityError(
-            f"{ActivityError.LOCK_ACQUISITION_ERROR}: max_locks must be greater than 0, got {max_locks}"
+        raise ApplicationError(
+            f"{ActivityError.LOCK_ACQUISITION_ERROR}: max_locks must be greater than 0, got {max_locks}",
+            non_retryable=True,
+        )
+    slot = random.randint(0, max_locks - 1)
+    resource_id = f"{APPLICATION_NAME}:{lock_name}:{slot}"
+
+    try:
+        async with RedisClientAsync() as redis_client:
+            # Acquire lock - connection will stay open until context exits
+            acquired = await redis_client._acquire_lock(
+                resource_id, owner_id, ttl_seconds
+            )
+            if acquired:
+                logger.info(f"Lock acquired for slot {slot}, resource: {resource_id}")
+                return {
+                    "status": True,
+                    "slot_id": slot,
+                    "resource_id": resource_id,
+                    "owner_id": owner_id,
+                }
+
+            raise ActivityError(
+                f"{ActivityError.LOCK_ACQUISITION_ERROR}: Lock not acquired for {resource_id}, will retry after some time"
+            )
+    except Exception as e:
+        # Redis connection or operation failed - propagate as activity error
+        if isinstance(e, (ActivityError)):
+            raise e
+        logger.error(f"Redis error during lock acquisition: {e}")
+        raise ApplicationError(
+            f"Redis error during lock acquisition for {resource_id}, error: {e}",
+            non_retryable=True,
+            type=type(e).__name__,
         )
-
-    async with RedisClientAsync() as redis_client:
-        while True:
-            slot = random.randint(0, max_locks - 1)
-            resource_id = f"{APPLICATION_NAME}:{lock_name}:{slot}"
-
-            try:
-                # Acquire lock - connection will stay open until context exits
-                acquired = await redis_client._acquire_lock(
-                    resource_id, owner_id, ttl_seconds
-                )
-                if acquired:
-                    logger.info(
-                        f"Lock acquired for slot {slot}, resource: {resource_id}"
-                    )
-                    return {
-                        "slot_id": slot,
-                        "resource_id": resource_id,
-                        "owner_id": owner_id,
-                    }
-                # If not acquired, continue retrying (lock held by another owner)
-
-            except Exception as e:
-                # Redis connection or operation failed - propagate as activity error
-                logger.error(f"Redis error during lock acquisition: {e}")
-                raise ActivityError(
-                    f"{ActivityError.LOCK_ACQUISITION_ERROR}: Redis error during lock acquisition for {resource_id}"
-                )
-
-            # Wait before retrying
-            await asyncio.sleep(LOCK_RETRY_INTERVAL)
 
 
 @activity.defn
@@ -94,8 +95,8 @@ async def release_distributed_lock(
     Returns:
         True if lock was released successfully, False otherwise
     """
-    async with RedisClientAsync() as redis_client:
-        try:
+    try:
+        async with RedisClientAsync() as redis_client:
             released, result = await redis_client._release_lock(resource_id, owner_id)
             if released:
                 logger.info(
@@ -103,8 +104,8 @@ async def release_distributed_lock(
                 )
             return released
 
-        except Exception as e:
-            logger.error(f"Redis error during lock release for {resource_id}: {e}")
-            # Don't raise exception for lock release failures - log and return False
-            # Lock release is best-effort and shouldn't fail the workflow
-            return False
+    except Exception as e:
+        logger.error(f"Redis error during lock release for {resource_id}: {e}")
+        # Don't raise exception for lock release failures - log and return False
+        # Lock release is best-effort and shouldn't fail the workflow
+        return False

application_sdk/interceptors/lock.py

@@ -22,6 +22,7 @@ from application_sdk.constants import (
     APPLICATION_NAME,
     IS_LOCKING_DISABLED,
     LOCK_METADATA_KEY,
+    LOCK_RETRY_INTERVAL,
 )
 from application_sdk.observability.logger_adaptor import get_logger
 
@@ -106,13 +107,17 @@ class RedisLockOutboundInterceptor(WorkflowOutboundInterceptor):
         lock_result = None
 
         try:
-            # Step 1: Acquire lock via dedicated activity (can take >2s safely)
-            start_to_close_timeout = workflow.info().execution_timeout
-            lock_result = await workflow.execute_activity(
+            # Step 1: Acquire lock via dedicated activity with Temporal retry policy
+            schedule_to_close_timeout = workflow.info().execution_timeout
+            lock_result = await workflow.execute_local_activity(
                 "acquire_distributed_lock",
                 args=[lock_name, max_locks, ttl_seconds, owner_id],
-                start_to_close_timeout=start_to_close_timeout,
-                retry_policy=RetryPolicy(maximum_attempts=1),
+                start_to_close_timeout=timedelta(seconds=30),
+                retry_policy=RetryPolicy(
+                    initial_interval=timedelta(seconds=int(LOCK_RETRY_INTERVAL)),
+                    backoff_coefficient=1.0,
+                ),
+                schedule_to_close_timeout=schedule_to_close_timeout,
             )
 
             logger.debug(f"Lock acquired: {lock_result}, executing {input.activity}")

application_sdk/version.py

@@ -2,4 +2,4 @@
 Version information for the application_sdk package.
 """
 
-__version__ = "0.1.1rc45"
+__version__ = "0.1.1rc46"

{atlan_application_sdk-0.1.1rc45.dist-info → atlan_application_sdk-0.1.1rc46.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atlan-application-sdk
-Version: 0.1.1rc45
+Version: 0.1.1rc46
 Summary: Atlan Application SDK is a Python library for developing applications on the Atlan Platform
 Project-URL: Repository, https://github.com/atlanhq/application-sdk
 Project-URL: Documentation, https://github.com/atlanhq/application-sdk/README.md

{atlan_application_sdk-0.1.1rc45.dist-info → atlan_application_sdk-0.1.1rc46.dist-info}/RECORD

@@ -1,9 +1,9 @@
 application_sdk/__init__.py,sha256=2e2mvmLJ5dxmJGPELtb33xwP-j6JMdoIuqKycEn7hjg,151
 application_sdk/constants.py,sha256=eLHmH9GukXCKK-u5a4bAqz8BeCOCusCM0eW0Q0Bwjns,10947
-application_sdk/version.py,sha256=p3JLlDmx39Ch7lMX6DS5ZKpVpATBikUiznrMlVxdpTA,88
+application_sdk/version.py,sha256=TTHbof7z1rqMmnKkkw-wrzn3nRqtDcXfc_kbnHfnflc,88
 application_sdk/worker.py,sha256=i5f0AeKI39IfsLO05QkwC6uMz0zDPSJqP7B2byri1VI,7489
 application_sdk/activities/__init__.py,sha256=QaXLOBYbb0zPOY5kfDQh56qbXQFaYNXOjJ5PCvatiZ4,9530
-application_sdk/activities/lock_management.py,sha256=L__GZ9BsArwU1ntYwAgCKsSjCqN6QBeOfT-OT4WyD4Y,3983
+application_sdk/activities/lock_management.py,sha256=oX2qPpfEu_xP0MiaCakVGk9ivZDvG4EddVZag1DuHSE,3976
 application_sdk/activities/.cursor/BUGBOT.md,sha256=FNykX5aMkdOhzgpiGqstOnSp9JN63iR2XP3onU4AGh8,15843
 application_sdk/activities/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 application_sdk/activities/common/models.py,sha256=LIZfWvTtgtbAUvvn-rwrPQgD7fP2J0Gxdxr_ITgw-jM,1243
@@ -67,7 +67,7 @@ application_sdk/inputs/.cursor/BUGBOT.md,sha256=hwKGDbopv3NU0bpC_ElpAPDFcS59GWS3
 application_sdk/interceptors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 application_sdk/interceptors/cleanup.py,sha256=JlFcM_2Y5AIEfGTSNe0aoon7eoE68MIXI0rA3LHsSeY,5966
 application_sdk/interceptors/events.py,sha256=TeStWmBbc4v1-dm2DWeKYsUfUhJLR8CtTQhu3TWOZWM,6524
-application_sdk/interceptors/lock.py,sha256=Xe9TSjYKtDZUB94hbV7rHG_9rgKUJPTACeB8z8xsJ0w,5577
+application_sdk/interceptors/lock.py,sha256=K1e1p11OYDDTy5TFMHcKXAvY4H86yXgpAZiuncEyH2M,5810
 application_sdk/interceptors/.cursor/BUGBOT.md,sha256=pxmUF2c7dtaXAX8yAa1-LBa6FCrj_uw7aQcHrppjf1A,14570
 application_sdk/observability/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 application_sdk/observability/logger_adaptor.py,sha256=WTqnNg78W2SRGOQVhELVLn6KMRsurkG1kc7essL08Lk,29529
@@ -156,8 +156,8 @@ application_sdk/workflows/metadata_extraction/__init__.py,sha256=jHUe_ZBQ66jx8bg
 application_sdk/workflows/metadata_extraction/sql.py,sha256=6ZaVt84n-8U2ZvR9GR7uIJKv5v8CuyQjhlnoRJvDszc,12435
 application_sdk/workflows/query_extraction/__init__.py,sha256=n066_CX5RpJz6DIxGMkKS3eGSRg03ilaCtsqfJWQb7Q,117
 application_sdk/workflows/query_extraction/sql.py,sha256=kT_JQkLCRZ44ZpaC4QvPL6DxnRIIVh8gYHLqRbMI-hA,4826
-atlan_application_sdk-0.1.1rc45.dist-info/METADATA,sha256=nuX1LdMTNoLQW3eKYU9KdNp2vXPiLG3RnKxZeu5ZcpM,5567
-atlan_application_sdk-0.1.1rc45.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-atlan_application_sdk-0.1.1rc45.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-atlan_application_sdk-0.1.1rc45.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
-atlan_application_sdk-0.1.1rc45.dist-info/RECORD,,
+atlan_application_sdk-0.1.1rc46.dist-info/METADATA,sha256=lFNImf3iOTzlw4bxeegOBO1YNjNfQ3yhAFzluCASQNI,5567
+atlan_application_sdk-0.1.1rc46.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+atlan_application_sdk-0.1.1rc46.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+atlan_application_sdk-0.1.1rc46.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
+atlan_application_sdk-0.1.1rc46.dist-info/RECORD,,