NSTextStorage长文本分页性能瓶颈问题技术咨询

嘿，我之前处理过类似的超长篇电子书排版场景，你的代码里的核心性能瓶颈其实是每次循环调用textContainer(forGlyphAt:)都会触发布局管理器重新计算整个文本的排版——对于9000+页的内容来说，反复触发全量布局简直是性能灾难。下面给你几个亲测有效的优化思路：

1. 先预估容量，批量添加容器 #

与其一次次循环检查最后一个字符的容器，不如先算出单页能容纳的最大Glyph数，直接估算总容器数批量添加，最后再做一次校验补漏：

guard let layoutManager = layoutManager, let textStorage = defaultTextStorage else { return }

// 先测单页能装多少Glyph（用空白测试文本跑一次布局）
let testContainer = NSTextContainer(size: textContainerSize)
layoutManager.addTextContainer(testContainer)
let testString = String(repeating: "x", count: 1000) // 足够长的测试文本
let testStorage = NSTextStorage(string: testString)
testStorage.addLayoutManager(layoutManager)
let glyphsPerPage = layoutManager.numberOfGlyphs(for: testContainer)
// 清理测试资源
layoutManager.removeTextContainer(testContainer)
testStorage.removeLayoutManager(layoutManager)

// 计算总容器数（向上取整避免漏页）
let totalGlyphs = layoutManager.numberOfGlyphs
let estimatedContainers = (totalGlyphs + glyphsPerPage - 1) / glyphsPerPage

// 批量加容器
for _ in 0..<estimatedContainers {
    let container = NSTextContainer(size: textContainerSize)
    layoutManager.addTextContainer(container)
}
pagesCount = estimatedContainers

// 最后校验一下，确保所有文本都被覆盖（防止预估误差）
if layoutManager.textContainer(forGlyphAt: totalGlyphs - 1, effectiveRange: nil) == nil {
    while layoutManager.textContainer(forGlyphAt: totalGlyphs - 1, effectiveRange: nil) == nil {
        let container = NSTextContainer(size: textContainerSize)
        layoutManager.addTextContainer(container)
        pagesCount += 1
    }
}

这种方式把全量布局的触发次数从几千次降到1-2次，性能提升非常明显。

2. 把布局计算放到后台，别卡主线程 #

哪怕优化了计算次数，超长篇文本的布局还是需要时间，直接在主线程跑肯定会让APP卡死。把整个容器分配逻辑放到后台队列，完成后再切回主线程更新UI：

DispatchQueue.global(qos: .userInitiated).async { [weak self] in
    guard let self = self, let layoutManager = self.layoutManager, let textStorage = self.defaultTextStorage else { return }
    
    // 这里执行上面的预估+批量添加逻辑
    // ...（省略重复代码）
    
    DispatchQueue.main.async {
        // 更新分页计数、刷新UI
        self.pagesCount = estimatedContainers
        self.pageControl.numberOfPages = estimatedContainers
        // 比如刷新展示页面的CollectionView/TableView
        self.pageCollectionView.reloadData()
    }
}

⚠️ 注意：操作布局管理器和文本存储时要确保线程安全，别在后台和主线程同时读写这些对象。

3. 按需懒加载容器（推荐给阅读器场景） #

如果用户不会一次性跳到第9000页，完全没必要一次性创建所有容器。可以用懒加载的方式，只创建当前可见页前后的几个容器（比如前后各5页），用户滚动到边缘时再动态添加：

• 监听滚动视图的scrollViewDidScroll事件，计算当前可见的页码范围
• 检查已有的TextContainer是否覆盖了可见范围前后的N页
• 缺多少就动态加多少，甚至可以把超出缓存范围的旧容器删掉（节省内存）
  这种方式能把性能消耗分散到用户交互过程中，初始化时几乎瞬间完成。

4. 优化NSTextStorage本身的性能 #

超长篇文本的TextStorage也可能拖慢速度，你可以试试：

• 分段加载：别一次性把500页文本全塞进TextStorage，而是根据当前页码加载对应段落，滚动时再加载下一段
• 减少样式修改次数：尽量一次性设置好所有文本属性，避免反复修改触发重排

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