React Context 牵连整页重渲染：拆状态订阅并验证更新范围

RAG 检索页把输入、结果、选中项和流式文本塞进一个 Context，更新范围很容易失控。先用 Profiler 看是谁重渲染，再决定拆 Store 还是拆订阅。

如何定位 Context 更新范围

打开 Chrome DevTools 的 Memory 面板，录制一段打字检索过程中的 Allocation instrumentation on timeline，问题可以看到更新范围。

对比输入前后两次内存快照，检查是否存在持续保留的FiberNode或事件对象；再从 React Profiler 读取一次 SSE 更新触发的组件数量。具体数量由当前页面采集，不预设为固定值。

一个常见问题是把检索结果、选中节点、流式文本和输入控制都放进同一个 Context：

// ❌ 错误示范：把高频更新和低频配置混放在一个 Context 中 const RAGKnowledgeContext = createContext<{ searchQuery: string; streamingTokens: string; searchResults: VectorDocumentNode[]; selectedDocId: string; }>(null!);

Context 的 value 引用变化时，读取该 Context 的消费者会收到更新。未订阅 Context 的组件是否重新渲染还取决于父组件和 props；因此应先在 Profiler 中确认实际提交范围。

高频流式更新可能扩大渲染范围，但不会自动造成 DOM 事件监听器泄漏。若内存持续增长，应检查未取消的请求、定时器、事件监听器以及缓存引用。

基于 useSyncExternalStore 拆分状态与虚拟列表防泄露

解决这个问题的关键，是将频繁变动的 RAG 状态从 React 声明式 Context 中剥离出来，改用外部订阅模式。利用 React 18 提供的useSyncExternalStoreHook，实现精准的颗粒度订阅。

下面是经过重构的RAG 上下文状态管理器示例与虚拟列表封装代码。

import { useSyncExternalStore, useCallback, useRef, useEffect } from 'react'; export interface VectorDocumentNode { id: string; title: string; score: number; snippet: string; } type Listener = () => void; class RAGSearchStore { private searchQuery = ''; private results: VectorDocumentNode[] = []; private streamingToken = ''; private listeners: Set<Listener> = new Set(); public subscribe = (listener: Listener) => { this.listeners.add(listener); return () => { this.listeners.delete(listener); }; }; public getSearchQuerySnapshot = () => this.searchQuery; public getResultsSnapshot = () => this.results; public getStreamingTokenSnapshot = () => this.streamingToken; public setSearchQuery(query: string) { this.searchQuery = query; this.emitChange(); } public setStreamingToken(token: string) { this.streamingToken = token; this.emitChange(); } public setResults(results: VectorDocumentNode[]) { this.results = results; this.emitChange(); } private emitChange() { this.listeners.forEach((listener) => listener()); } } export const ragStore = new RAGSearchStore(); /** * 订阅检索结果快照；订阅者只会在该快照引用变化时更新。 */ export function useRAGResults(): VectorDocumentNode[] { return useSyncExternalStore( ragStore.subscribe, ragStore.getResultsSnapshot, ragStore.getResultsSnapshot ); } /** * 精准订阅高频 Token，仅限制在 Log 预览组件内部 */ export function useRAGStreamingToken(): string { return useSyncExternalStore( ragStore.subscribe, ragStore.getStreamingTokenSnapshot, ragStore.getStreamingTokenSnapshot ); }

紧接着，重构虚拟列表的绑定逻辑，严防 DOM 泄漏与闭包循环引用：

import React, { useState, useMemo } from 'react'; interface VirtualListProps<T> { items: T[]; itemHeight: number; containerHeight: number; renderItem: (item: T, index: number) => React.ReactNode; } export function OptimizedRAGVirtualList<T extends { id: string }>({ items, itemHeight, containerHeight, renderItem, }: VirtualListProps<T>) { const [scrollTop, setScrollTop] = useState(0); const containerRef = useRef<HTMLDivElement>(null); const handleScroll = useCallback((e: React.UIEvent<HTMLDivElement>) => { // 使用 requestAnimationFrame 避开高频 setState 阻塞主线程 const target = e.currentTarget; requestAnimationFrame(() => { if (target) { setScrollTop(target.scrollTop); } }); }, []); const totalHeight = items.length * itemHeight; const { startIndex, endIndex, visibleItems } = useMemo(() => { const start = Math.max(0, Math.floor(scrollTop / itemHeight) - 2); const end = Math.min(items.length, Math.ceil((scrollTop + containerHeight) / itemHeight) + 2); return { startIndex: start, endIndex: end, visibleItems: items.slice(start, end), }; }, [items, scrollTop, itemHeight, containerHeight]); return ( <div ref={containerRef} onScroll={handleScroll} style={{ height: `${containerHeight}px`, overflowY: 'auto', position: 'relative' }} > <div style={{ height: `${totalHeight}px`, width: '100%', position: 'relative' }}> <div style={{ transform: `translateY(${startIndex * itemHeight}px)` }}> {visibleItems.map((item, idx) => ( <div key={item.id} style={{ height: `${itemHeight}px` }}> {renderItem(item, startIndex + idx)} </div> ))} </div> </div> </div> ); }

如何验证这类重构

重构后，录制相同输入序列并比较 React Profiler 的提交次数、提交耗时和渲染组件范围；再用 Allocation instrumentation 与 Heap Snapshot 检查页面空闲后是否仍保留无用对象。只有在相同设备、数据量和网络条件下，前后结果才可比较。

这类问题可以留下两个可执行的检查。

第一，流式响应和检索结果属于高频异步状态，应避免与低频 UI 配置共用一个粗粒度 Context。

第二，虚拟列表只解决可见项数量问题，不能替代对父组件更新和副作用生命周期的排查。