from flask import Flask, Response, Blueprint
import time

app = Flask(__name__)

bp1 = Blueprint('sse1', __name__)
bp2 = Blueprint('sse2', __name__)

@bp1.route('/sse1')
def sse1():
    def generate():
        while True:
            yield 'data: {}\n\n'.format(time.time())
            time.sleep(1)
    return Response(generate(), mimetype='text/event-stream')

@bp2.route('/sse2')
def sse2():
    def generate():
        while True:
            yield 'data: {}\n\n'.format(time.strftime('%Y-%m-%d %H:%M:%S'))
            time.sleep(1)
    return Response(generate(), mimetype='text/event-stream')

app.register_blueprint(bp1)
app.register_blueprint(bp2)

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

from flask import Flask, Response, request

app = Flask(__name__)

# 存儲(chǔ)所有的訂閱者
subscribers = {}

# 訂閱接口，每個(gè)瀏覽器對(duì)應(yīng)一個(gè)topic
@app.route('/subscribe/<topic>')
def subscribe(topic):
    def stream():
        # 將當(dāng)前請(qǐng)求的客戶端添加到訂閱者列表中
        subscribers.setdefault(topic, []).append(stream)
        # 無限循環(huán)，等待新的消息
        while True:
            # 等待新的消息
            message = request.args.get('message')
            if message:
                # 向所有訂閱者推送新的消息
                for subscriber in subscribers.get(topic, []):
                    subscriber.put(message)
            # 休眠一段時(shí)間，減少服務(wù)器壓力
            time.sleep(1)

    # 設(shè)置響應(yīng)頭，告訴客戶端這是一個(gè)SSE流
    response = Response(stream(), mimetype='text/event-stream')
    response.headers['Cache-Control'] = 'no-cache'
    response.headers['Connection'] = 'keep-alive'
    return response

# 推送接口，向指定topic的所有訂閱者推送消息
@app.route('/publish/<topic>')
def publish(topic):
    # 獲取要推送的消息
    message = request.args.get('message')
    if message:
        # 向所有訂閱者推送新的消息
        for subscriber in subscribers.get(topic, []):
            subscriber.put(message)
    return 'OK'

import uuid
from flask import Flask, jsonify, request, Response, g
from flask_cors import CORS
from flask_sse import sse
import time
import json

app = Flask(__name__)
cros = CORS(app)
app.config['REDIS_URL'] = 'redis://localhost'
app.register_blueprint(sse, url_prefix='/stream')
# SSE 推送函數(shù)


# SSE 推送路由


@app.route('/register', methods=["GET"])
def register():
    # 獲取客戶端標(biāo)識(shí)符
    client_id = str(uuid.uuid4())

    # 返回 SSE 響應(yīng)
    return jsonify({"client_id": client_id})

# SSE 推送路由


@app.route('/sse', methods=['POST'])
def stream():
    # 獲取客戶端標(biāo)識(shí)符
    data = request.get_json()
    client_id = data['clientId']
    print("client_id", client_id)

    def aa():
        # 循環(huán)發(fā)送 SSE 數(shù)據(jù)
        for i in range(10):
            data = 'Hello, %s!' % client_id+str(i)
            sse.publish(data, channel=client_id, type='message')
            time.sleep(1)
        sse.publish("end", channel=client_id, type='message')
     # 返回 SSE 響應(yīng)
    response = Response(aa(), mimetype='text/event-stream')
    response.headers.add('Cache-Control', 'no-cache')
    response.headers.add('Connection', 'keep-alive')
    response.headers.add('X-Accel-Buffering', 'no')
    return response


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