Flask-Bokeh服务:关闭浏览器窗口时释放内存的方法求助
咱们先搞清楚核心原因:Bokeh Server默认会保留用户会话一段时间,哪怕浏览器窗口关闭了,它也不会立刻销毁会话释放内存——这些会话对象还被Bokeh的内部管理机制持有引用,所以手动调用gc.collect()根本没用,垃圾回收器碰不到这些被引用的对象。
下面是几个实用的解决方案,按优先级排序:
1. 配置Bokeh会话自动超时(最推荐)
这是最简单直接的方法,让Bokeh自动清理闲置会话。你只需要在创建BokehTornado实例时,添加session_expiration参数,设置一个合理的超时时长(单位是毫秒):
bokeh_tornado = BokehTornado( apps, extra_websocket_origins=["myhost.com"], session_expiration=600000 # 设置为10分钟,可根据需求调整 )
当用户关闭窗口后,会话闲置超过设定时间,Bokeh就会自动销毁会话,释放对应的内存资源。
2. 前端监听窗口关闭事件,主动销毁会话
如果需要更即时的内存释放,可以在你的模板页面bokserv.html中添加JavaScript代码,监听窗口关闭/标签页切换事件,主动调用Bokeh的API销毁当前会话:
<script> window.addEventListener('beforeunload', function() { // 获取当前Bokeh文档的会话ID const currentDoc = Bokeh.documents[0]; if (currentDoc?.id) { // 发送DELETE请求销毁会话,注意替换成你的Bokeh服务地址 fetch(`http://11.111.11.111:{{ port }}/${currentDoc.id.split('/')[0]}/session/${currentDoc.id.split('/')[1]}`, { method: 'DELETE' }); } }); </script>
记得在Flask渲染模板时把port变量传进去,这样用户一关闭窗口,就会立刻触发会话销毁,内存能快速释放。
3. 自定义Bokeh会话销毁钩子,清理自定义资源
如果你的Bokeh应用里创建了自定义资源(比如数据库连接、大型数据集),可以在Bokeh脚本(比如bokeh1.py)中添加会话生命周期钩子,在会话销毁时手动清理这些资源:
from bokeh.io import curdoc def cleanup_session(session_context): # 在这里清理当前会话的自定义资源,比如关闭数据库连接、删除临时文件等 print(f"Session {session_context.session.id} 已销毁,正在清理资源") curdoc().on_session_destroyed(cleanup_session)
这个方法能确保你自己创建的资源被正确释放,辅助整体的内存回收。
补充:为什么gc.collect()没效果?
因为Bokeh Server的会话对象被Tornado的IOLoop和Bokeh的会话管理器持有引用,只要这些引用存在,垃圾回收器就无法回收这些对象。只有当Bokeh主动移除这些引用(比如超时销毁会话),gc.collect()才能真正发挥作用。
内容的提问来源于stack exchange,提问作者machump

