消息推送系统

小明：嘿，李工，我最近在做一个消息管理系统，想看看能不能实现批量发消息的功能，你有什么建议吗？

李工：嗯，批量发消息是常见的需求。首先你要考虑系统架构和前端交互设计。你用的是什么技术栈？

小明：后端是Python Flask，前端是React。现在前端已经能展示消息列表了，但批量发送还没实现。

李工：那我们可以从两个方面入手：前端选择多条消息，后端处理这些消息并批量发送。先说前端，你需要一个复选框来让用户选择消息。

小明：明白了，那怎么把这些选中的消息传给后端呢？

李工：你可以用AJAX请求，把选中的消息ID或数据作为参数传递过去。比如，使用POST方法发送一个包含message_ids的JSON对象。

小明：好的，那后端怎么处理这个请求呢？

李工：在Flask中，你可以定义一个路由，接收POST请求。然后从请求体中解析出message_ids，再根据这些ID查询对应的消息内容，最后调用发送接口。

小明：听起来挺直接的。那发送接口是怎样的？是不是需要异步处理？

李工：是的，如果消息量很大，最好用异步任务来处理，比如用Celery或者RQ。这样不会阻塞主线程，提高系统性能。

小明：那我应该怎么写这个异步任务呢？

李工：以Celery为例，你可以在Flask应用中配置Celery，然后定义一个任务函数，接收消息列表，逐条发送。记得还要处理发送失败的情况。

小明：明白了。那前端怎么知道发送是否成功呢？

李工：后端可以返回一个状态码和消息结果，前端根据返回结果更新UI，比如显示“发送成功”或“部分失败”。也可以使用WebSocket实时通知用户。

小明：那有没有现成的库或者框架推荐？

李工：前端的话，你可以用axios或fetch API发送请求；后端可以用Flask-RESTful或者直接使用Flask的request对象。异步任务的话，Celery是一个不错的选择。

小明：那我可以先写一个简单的例子试试看？

李工：当然可以。下面我给你一个前端和后端的代码示例，你可以参考一下。

前端代码（React）：

      
import React, { useState } from 'react';
import axios from 'axios';

function MessageList() {
  const [messages, setMessages] = useState([]);
  const [selectedIds, setSelectedIds] = useState([]);

  const handleSelect = (id) => {
    if (selectedIds.includes(id)) {
      setSelectedIds(selectedIds.filter(i => i !== id));
    } else {
      setSelectedIds([...selectedIds, id]);
    }
  };

  const handleBatchSend = async () => {
    try {
      const response = await axios.post('/api/batch-send', { ids: selectedIds });
      alert(response.data.message);
    } catch (error) {
      alert('发送失败');
    }
  };

  return (
    

      

        {messages.map(msg => (
          

             handleSelect(msg.id)}
            />
            {msg.content}
          
        ))}
      
      批量发送
    
  );
}

export default MessageList;
      
    

后端代码（Flask + Celery）：

      
from flask import Flask, request, jsonify
from celery import Celery
import time

app = Flask(__name__)
app.config['CELERY_BROKER_URL'] = 'redis://localhost:6379/0'
celery = Celery(app.name, broker=app.config['CELERY_BROKER_URL'])

# 模拟消息数据库
messages_db = [
    {'id': 1, 'content': '你好！'},
    {'id': 2, 'content': '今天天气不错。'},
    {'id': 3, 'content': '明天见。'}
]

@app.route('/api/messages', methods=['GET'])
def get_messages():
    return jsonify(messages_db)

@app.route('/api/batch-send', methods=['POST'])
def batch_send():
    data = request.get_json()
    ids = data.get('ids', [])
    # 调用异步任务
    send_messages.delay(ids)
    return jsonify({'message': '正在批量发送消息...'})

@celery.task
def send_messages(ids):
    for id in ids:
        msg = next((m for m in messages_db if m['id'] == id), None)
        if msg:
            print(f"发送消息: {msg['content']}")
            time.sleep(1)  # 模拟发送延迟
    print("所有消息已发送完成")

if __name__ == '__main__':
    app.run(debug=True)
      
    

小明：谢谢李工，这代码示例对我帮助很大。

李工：不客气，记得测试时注意异常处理和日志记录，确保系统稳定。

小明：明白了，我会继续完善这个功能。

李工：加油，有需要随时问我。