diff --git a/lib/madara_orchestrator/localstack.star b/lib/madara_orchestrator/localstack.star index 9f7ab76..a1d9f86 100644 --- a/lib/madara_orchestrator/localstack.star +++ b/lib/madara_orchestrator/localstack.star @@ -18,9 +18,14 @@ def run(plan, args, suffix): name = args["name"] + suffix + reset_sqs_file_artifact = plan.upload_file( + src="../scripts/reset_sqs.sh", + ) + config = ServiceConfig( image=args["image"], ports=ports, + files={"/localstack-setup.sh": reset_sqs_file_artifact}, env_vars={ "DEFAULT_REGION": args["default_region"], "AWS_ACCESS_KEY_ID": args["aws_access_key_id"], diff --git a/scripts/reset_sqs.sh b/scripts/reset_sqs.sh new file mode 100644 index 0000000..3f29f98 --- /dev/null +++ b/scripts/reset_sqs.sh @@ -0,0 +1,36 @@ +#!/bin/bash -e + +# Set the endpoint URL +ENDPOINT_URL="http://localhost:4566" + +# Define an array of queue names +QUEUES=( + "madara_orchestrator_snos_job_processing_queue" + "madara_orchestrator_snos_job_verification_queue" + "madara_orchestrator_proving_job_processing_queue" + "madara_orchestrator_proving_job_verification_queue" + "madara_orchestrator_data_submission_job_processing_queue" + "madara_orchestrator_data_submission_job_verification_queue" + "madara_orchestrator_update_state_job_processing_queue" + "madara_orchestrator_update_state_job_verification_queue" + "madara_orchestrator_job_handle_failure_queue" + "madara_orchestrator_worker_trigger_queue" +) + +# Loop through the queues and purge each one +echo "----------------------------" +for QUEUE in "${QUEUES[@]}"; do + QUEUE_URL="${ENDPOINT_URL}/000000000000/${QUEUE}" + echo "Purging queue: ${QUEUE}" + aws --endpoint-url ${ENDPOINT_URL} sqs purge-queue --queue-url "${QUEUE_URL}" + echo "Queue purged: ${QUEUE}" +done + +# Loop through the queues and recreate each one +echo "----------------------------" +for QUEUE in "${QUEUES[@]}"; do + echo "Creating queue: ${QUEUE}" + aws --endpoint-url ${ENDPOINT_URL} sqs create-queue --queue-name "${QUEUE}" +done + +echo "All queues have been created." \ No newline at end of file