统一消息系统

import flask

from flask import Flask, request, jsonify

import threading

import time

app = Flask(__name__)

# 模拟消息队列

message_queue = []

@app.route('/send_message', methods=['POST'])

def send_message():

data = request.get_json()

message = data.get('message')

user_id = data.get('user_id')

# 将消息添加到队列中

message_queue.append({'user_id': user_id, 'message': message})

return jsonify({'status': 'success', 'message': 'Message sent'})

@app.route('/get_messages', methods=['GET'])

def get_messages():

user_id = request.args.get('user_id')

# 过滤出该用户的消息

messages = [msg for msg in message_queue if msg['user_id'] == user_id]

return jsonify({'messages': messages})

# 模拟定时推送消息

def push_messages():

while True:

time.sleep(5)

print("Pushing messages to clients...")

if __name__ == '__main__':

threading.Thread(target=push_messages).start()

app.run(debug=True)

from flask import Flask, render_template

from flask_socketio import SocketIO, emit

app = Flask(__name__)

app.config['SECRET_KEY'] = 'secret!'

socketio = SocketIO(app)

# 模拟消息队列

message_queue = []

@socketio.on('connect')

def handle_connect():

print('Client connected')

@socketio.on('disconnect')

def handle_disconnect():

print('Client disconnected')

@socketio.on('send_message')

def handle_send_message(data):

message = data['message']

user_id = data['user_id']

# 存入消息队列

message_queue.append({'user_id': user_id, 'message': message})

# 推送消息给所有客户端

emit('receive_message', {'user_id': user_id, 'message': message}, broadcast=True)

@app.route('/')

def index():

return render_template('index.html')

if __name__ == '__main__':

socketio.run(app, debug=True)

import jwt

from datetime import datetime, timedelta

SECRET_KEY = 'your-secret-key'

def generate_token(user_id):

payload = {

'user_id': user_id,

'exp': datetime.utcnow() + timedelta(hours=1)

}

token = jwt.encode(payload, SECRET_KEY, algorithm='HS256')

return token

def verify_token(token):

try:

payload = jwt.decode(token, SECRET_KEY, algorithms=['HS256'])

return payload['user_id']

except:

return None