task-queue-system/app/workers/mail_mx_worker.py

24 lines
No EOL
796 B
Python

import asyncio
from app.workers.base_worker import BaseWorker
from app.job_registry import routing_key
from app.workers.utils import resolve_mx
class MXValidationWorker(BaseWorker):
async def process(self, payload: dict, **meta) -> dict:
domain = payload.get("domain")
if not domain:
raise ValueError("payload.domain is required")
mx = await resolve_mx(domain)
await asyncio.sleep(0)
return {
"mx_records": [{"exchange": m.exchange, "preference": m.preference} for m in mx],
"count": len(mx),
}
async def main():
worker = MXValidationWorker(routing_key("MailCheck", "MXValidation"), worker_id="mx-worker-1")
await worker.run()
if __name__ == "__main__":
import asyncio as _a
_a.run(main())