Thanks to visit codestin.com
Credit goes to github.com

Skip to content

[SQS] Improve shutdown capabilities of SqsProvider #12252

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
merged 3 commits into from
Feb 12, 2025
Merged

[SQS] Improve shutdown capabilities of SqsProvider #12252

merged 3 commits into from
Feb 12, 2025

Conversation

gregfurman
Copy link
Contributor

Motivation

There are still some hanging threads on SQS shutdown due to:

  • Cleanup fixtures deleting SQS queues lifecycle hooks are able to shut them down
  • CloudWatch metric dispatcher does not guard against new submit jobs when executor is shutdown

Changes

  • Shutdown internal SQS queue resource upon deletion.
  • Prevent dispatching new CW jobs when ThreadPoolExecutor object is closed (causing a RuntimeException)

@gregfurman gregfurman added aws:sqs Amazon Simple Queue Service semver: patch Non-breaking changes which can be included in patch releases labels Feb 11, 2025
@gregfurman gregfurman requested a review from dfangl February 11, 2025 15:46
@gregfurman gregfurman self-assigned this Feb 11, 2025
@github-actions GitHub Actions
Copy link

github-actions bot commented Feb 11, 2025 •
edited
Loading

LocalStack Community integration with Pro

    2 files  ±    0      2 suites  ±0   1h 15m 7s ⏱️ - 38m 21s
3 010 tests  - 1 087  2 879 ✅  - 902  131 💤  - 185  0 ❌ ±0 
3 012 runs   - 1 087  2 879 ✅  - 902  133 💤  - 185  0 ❌ ±0 

Results for commit 15b4253. ± Comparison against base commit 57aeecb.

This pull request removes 1087 tests. 
tests.aws.scenario.bookstore.test_bookstore.TestBookstoreApplication ‑ test_lambda_dynamodb
tests.aws.scenario.bookstore.test_bookstore.TestBookstoreApplication ‑ test_opensearch_crud
tests.aws.scenario.bookstore.test_bookstore.TestBookstoreApplication ‑ test_search_books
tests.aws.scenario.bookstore.test_bookstore.TestBookstoreApplication ‑ test_setup
tests.aws.scenario.kinesis_firehose.test_kinesis_firehose.TestKinesisFirehoseScenario ‑ test_kinesis_firehose_s3
tests.aws.scenario.lambda_destination.test_lambda_destination_scenario.TestLambdaDestinationScenario ‑ test_destination_sns
tests.aws.scenario.lambda_destination.test_lambda_destination_scenario.TestLambdaDestinationScenario ‑ test_infra
tests.aws.scenario.loan_broker.test_loan_broker.TestLoanBrokerScenario ‑ test_prefill_dynamodb_table
tests.aws.scenario.loan_broker.test_loan_broker.TestLoanBrokerScenario ‑ test_stepfunctions_input_recipient_list[step_function_input0-SUCCEEDED]
tests.aws.scenario.loan_broker.test_loan_broker.TestLoanBrokerScenario ‑ test_stepfunctions_input_recipient_list[step_function_input1-SUCCEEDED]
…

♻️ This comment has been updated with latest results.

@gregfurman gregfurman marked this pull request as ready for review February 11, 2025 17:26
bentsku
bentsku approved these changes Feb 12, 2025
View reviewed changes
Copy link
Contributor

@bentsku bentsku left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM! Good catch!

I think we could replace the threading.Event with a boolean in this case, as we don't need to pass the reference around and mutate it. But both works!

bentsku
bentsku reviewed Feb 12, 2025
View reviewed changes
Copy link
Contributor

@bentsku bentsku left a comment •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I've just realized we have an issue with the CloudwatchDispatcher: if you call LocalStack reset for SQS, we will shutdown the dispatcher in on_before_stop, but when calling _start_cloudwatch_metrics_reporting in on_before_start we do not reset the flags back, so it will stay in shutdown, as we create the dispatcher in the __init__ method of the provider 🤔 and initialize the event/boolean flag in the __init__ method of the dispatcher too.

Bigger than that, we also don't restart the executor itself. A bit outside of the scope of the PR, but we should probably fix that.

@gregfurman
Copy link
Contributor Author

@bentsku There are a couple issues with the CloudwatchDispatcher imo. We also don't batch requests which means that for each send, receieve, and delete message call we'll send 2, 1, and 1 requests to the CW provider (and hence the gateway) respectively.

@gregfurman gregfurman merged commit 25d4d82 into master Feb 12, 2025
31 checks passed
@gregfurman gregfurman deleted the fix/sqs/cw-shutdown branch February 12, 2025 19:43
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@bentsku bentsku bentsku approved these changes

@dfangl dfangl Awaiting requested review from dfangl

@thrau thrau Awaiting requested review from thrau

@baermat baermat Awaiting requested review from baermat

Assignees

@gregfurman gregfurman

Labels
aws:sqs Amazon Simple Queue Service semver: patch Non-breaking changes which can be included in patch releases
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@gregfurman @bentsku