75 lines
2.0 KiB
Python
75 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
# requeue-dlq.py <domain>
|
|
"""
|
|
Verschiebt Messages aus DLQ zurück in Main Queue
|
|
"""
|
|
|
|
import sys
|
|
import boto3
|
|
|
|
sqs = boto3.client('sqs', region_name='eu-central-1')
|
|
|
|
def requeue_dlq(domain, max_messages=10):
|
|
"""Verschiebt Messages aus DLQ zurück in Main Queue"""
|
|
|
|
queue_name = domain.replace('.', '-') + '-queue'
|
|
dlq_name = queue_name + '-dlq'
|
|
|
|
try:
|
|
queue_url = sqs.get_queue_url(QueueName=queue_name)['QueueUrl']
|
|
dlq_url = sqs.get_queue_url(QueueName=dlq_name)['QueueUrl']
|
|
except Exception as e:
|
|
print(f"❌ Error: {e}")
|
|
return
|
|
|
|
print(f"Re-queuing up to {max_messages} messages from DLQ to main queue...")
|
|
print(f"Domain: {domain}")
|
|
print(f"From: {dlq_name}")
|
|
print(f"To: {queue_name}\n")
|
|
|
|
confirm = input("Continue? (yes/no): ")
|
|
if confirm.lower() != 'yes':
|
|
print("Cancelled.")
|
|
return
|
|
|
|
# Messages aus DLQ holen
|
|
response = sqs.receive_message(
|
|
QueueUrl=dlq_url,
|
|
MaxNumberOfMessages=max_messages,
|
|
WaitTimeSeconds=0
|
|
)
|
|
|
|
messages = response.get('Messages', [])
|
|
|
|
if not messages:
|
|
print("No messages in DLQ.")
|
|
return
|
|
|
|
print(f"\nRe-queuing {len(messages)} message(s)...\n")
|
|
|
|
for msg in messages:
|
|
# In Haupt-Queue schreiben
|
|
sqs.send_message(
|
|
QueueUrl=queue_url,
|
|
MessageBody=msg['Body']
|
|
)
|
|
|
|
# Aus DLQ löschen
|
|
sqs.delete_message(
|
|
QueueUrl=dlq_url,
|
|
ReceiptHandle=msg['ReceiptHandle']
|
|
)
|
|
|
|
print(f" ✓ Re-queued message {msg['MessageId']}")
|
|
|
|
print(f"\n✅ Done! {len(messages)} message(s) re-queued to {queue_name}")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
if len(sys.argv) < 2:
|
|
print("Usage: python requeue-dlq.py <domain>")
|
|
print("Example: python requeue-dlq.py andreasknuth.de")
|
|
sys.exit(1)
|
|
|
|
domain = sys.argv[1]
|
|
requeue_dlq(domain) |