Streamlit仪表盘突发tornado.iostream.StreamClosedError问题求助

这个WebSocketClosedError本质是Streamlit会话已终止（比如用户关闭浏览器标签、刷新页面），但后台无限循环仍在尝试向已断开的WebSocket连接发送UI更新。

你的代码存在两个核心问题：

• 用@st.cache_resource创建的队列是全局共享的，跨所有用户会话。当某个用户关闭页面后，队列仍可能向该会话的已断开连接推送数据，触发报错。
• while True的阻塞循环完全脱离了Streamlit的会话生命周期管理，会话终止后循环不会自动停止，持续尝试发送更新。

1. 为每个会话独立维护队列 #

将全局队列改为会话级队列，用st.session_state存储，避免跨会话干扰：

import streamlit as st
from queue import Queue

# 初始化会话级队列
if "kafka_queue" not in st.session_state:
    st.session_state.kafka_queue = Queue()

queue = st.session_state.kafka_queue

2. 适配Streamlit的更新机制，添加会话存活检查 #

替换无限阻塞循环为带会话检查的循环，同时用st.empty()动态更新图表，避免重复渲染：

chart_placeholder = st.empty()

while True:
    # 检查当前会话是否仍活跃，若已终止则退出循环
    if not st.session_state.get("_session_active", True):
        break
    
    data = queue.get()
    if data:
        # 用占位符更新图表，而非重复创建
        chart_placeholder.line_chart(data)
    # 手动触发页面重跑，保证UI更新
    st.experimental_rerun()

3. 捕获WebSocket异常，避免报错扩散 #

在更新UI的代码块中捕获WebSocketClosedError，阻止异常堆栈输出：

from tornado.websocket import WebSocketClosedError

chart_placeholder = st.empty()

while True:
    if not st.session_state.get("_session_active", True):
        break
    
    data = queue.get()
    if data:
        try:
            chart_placeholder.line_chart(data)
        except WebSocketClosedError:
            # 会话已关闭，退出循环
            break
    st.experimental_rerun()

4. 升级Streamlit版本 #

你的1.22版本存在多个WebSocket相关的已知bug，升级到较新的稳定版本（比如1.30+）可以直接修复部分底层连接问题：

pip install --upgrade streamlit

• 外部Kafka客户端推送数据时，建议针对每个会话的队列单独推送，而非全局队列，避免向已关闭的会话发送无效数据。
• 避免在Streamlit中使用无限制的阻塞循环，尽量利用其内置的重跑机制实现动态更新。

内容的提问来源于stack exchange，提问作者saqx1