【后端开发实习】用MongoDB和Redis实现消息队列搭建分布式邮件消息系统

from pymongo import MongoClient
from redis import Redis
import time
import uuid
 
# 连接MongoDB和Redis
mongo_client = MongoClient('mongodb://localhost:27017/')
db = mongo_client['email_queue']
redis_client = Redis(host='localhost', port=6379)
 
# 邮件内容
email_content = {
    'to': 'recipient@example.com',
    'from': 'sender@example.com',
    'subject': 'Distributed Email System Test',
    'text': 'This is a test email sent by our distributed email system.'
}
 
# 将邮件内容插入MongoDB
def insert_email_to_mongo(email_content):
    email_content['_id'] = str(uuid.uuid4())
    db.emails.insert_one(email_content)
 
# 从MongoDB获取邮件内容并发送
def send_email_from_mongo():
    while True:
        # 假设的邮件发送函数
        def send_email(email_content):
            print(f"Sending email to {email_content['to']}")
            # 实际的邮件发送逻辑应该在这里
 
        # 从MongoDB查询邮件
        email = db.emails.find_one({'status': 'pending'})
        if email:
            # 更新邮件状态为'sending'
            db.emails.update_one({'_id': email['_id']}, {'$set': {'status': 'sending'}})
            # 调用模拟的发送邮件函数
            send_email(email)
            # 更新邮件状态为'sent'
            db.emails.update_one({'_id': email['_id']}, {'$set': {'status': 'sent'}})
            print("Email sent.")
        else:
            print("No emails to send.")
        time.sleep(5)  # 每5秒检查一次
 
# 将邮件ID添加到Redis队列
def add_email_to_redis_queue(email_id):
    redis_client.rpush('email_queue', email_id)
 
# 从Redis队列获取邮件ID并处理邮件
def process_email_from_redis_queue():
    while True:
        # 从队列中取出一个邮件ID
        email_id = redis_client.blpop(['email_queue'], timeout=5)[1].decode('utf-8')
        # 更新邮件状态为'pending'
        db.emails.update_one({'_id': email_id, 'status': 'queued'}, {'$set': {'status': 'pending'}})
        send_email_from_mongo()  # 尝试发送邮件
 
# 示例使用
if __name__ == '__main__':
    # 插入邮件到MongoDB
    insert_email_to_mongo(email_content)
    # 将邮件ID添加到Redis队列
    add_email_to_redis_queue(email_content['_id'])
    # 处理邮件队列
    process_email_from_redis_queue()

这个代码示例展示了如何使用MongoDB和Redis来构建一个简单的分布式邮件系统。它首先连接到MongoDB和Redis，然后定义了插入邮件内容到MongoDB的函数，一个从MongoDB获取邮件并模拟发送邮件的函数，一个将邮件ID添加到Redis队列的函数，以及一个从Redis队列获取邮件ID并处理邮件的函数。最后，它提供了使用这些组件的示例。