Skip to content

Commit a18ac2a

Browse files
committed
refactor(idempotent): remove include_function_name
1 parent 03a40cb commit a18ac2a

File tree

3 files changed

+1
-22
lines changed

3 files changed

+1
-22
lines changed

aws_lambda_powertools/utilities/idempotency/config.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,6 @@ def __init__(
3333
Max number of items to store in local cache, by default 1024
3434
hash_function: str, optional
3535
Function to use for calculating hashes, by default md5.
36-
include_function_name: bool, optional
37-
Whether to include the function name in the idempotent key
3836
"""
3937
self.event_key_jmespath = event_key_jmespath
4038
self.payload_validation_jmespath = payload_validation_jmespath
@@ -44,4 +42,3 @@ def __init__(
4442
self.use_local_cache = use_local_cache
4543
self.local_cache_max_items = local_cache_max_items
4644
self.hash_function = hash_function
47-
self.include_function_name = include_function_name

aws_lambda_powertools/utilities/idempotency/persistence/base.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,6 @@ def __init__(self):
122122
self.use_local_cache = False
123123
self._cache: Optional[LRUDict] = None
124124
self.hash_function = None
125-
self.include_function_name = True
126125

127126
def configure(self, config: IdempotencyConfig) -> None:
128127
"""
@@ -153,7 +152,6 @@ def configure(self, config: IdempotencyConfig) -> None:
153152
if self.use_local_cache:
154153
self._cache = LRUDict(max_items=config.local_cache_max_items)
155154
self.hash_function = getattr(hashlib, config.hash_function)
156-
self.include_function_name = config.include_function_name
157155

158156
def _get_hashed_idempotency_key(self, event: Dict[str, Any], context: LambdaContext) -> str:
159157
"""
@@ -183,10 +181,7 @@ def _get_hashed_idempotency_key(self, event: Dict[str, Any], context: LambdaCont
183181
warnings.warn(f"No value found for idempotency_key. jmespath: {self.event_key_jmespath}")
184182

185183
generated_hash = self._generate_hash(data)
186-
if self.include_function_name:
187-
return f"{context.function_name}#{generated_hash}"
188-
else:
189-
return generated_hash
184+
return f"{context.function_name}#{generated_hash}"
190185

191186
@staticmethod
192187
def is_missing_idempotency_key(data) -> bool:

tests/functional/idempotency/test_idempotency.py

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -828,16 +828,3 @@ def lambda_handler(event, context):
828828
stubber.assert_no_pending_responses()
829829
stubber.deactivate()
830830
assert "Failed to save in progress record to idempotency store" == e.value.args[0]
831-
832-
833-
def test_include_function_name_false(persistence_store: DynamoDBPersistenceLayer):
834-
# GIVEN include_function_name=False
835-
persistence_store.configure(IdempotencyConfig(event_key_jmespath="body", include_function_name=False))
836-
value = "true"
837-
api_gateway_proxy_event = {"body": value}
838-
839-
# WHEN
840-
result = persistence_store._get_hashed_idempotency_key(api_gateway_proxy_event, None)
841-
842-
# THEN
843-
assert result == persistence_store._generate_hash(value)

0 commit comments

Comments
 (0)