如何基于苹果CryptoKit实现低内存占用的大文件SHA-256哈希计算

完全存在适配需求的成熟实现，不需要引入任何第三方工具链，全程基于苹果官方CryptoKit就能做到：计算大文件SHA-256时全程内存占比恒定、不把整个文件读入内存、不会出现大文件触发的内存告警甚至闪退。

实现核心逻辑 #

SHA-256本身是支持流式计算的块哈希算法，CryptoKit提供的哈希计算接口从设计上就支持分块更新计算状态，根本不需要一次性把整个文件加载到内存。实际运行时只需要每次读取固定大小的文件分片，逐片喂给计算上下文更新状态，内存占用全程保持在固定的极低水平，和要计算的文件总大小没有任何关系，哪怕是数GB的大文件也不会造成内存飙升。

具体操作步骤 #

• 初始化计算上下文：直接实例化SHA256结构，这个可变实例会全程维护哈希计算的中间状态，完全不需要用户手动处理分块数据的对齐、补位逻辑
• 打开文件句柄：用FileHandle(forReadingFrom:)拿到目标文件的只读操作句柄，全程严禁使用Data(contentsOf:)这类会自动把整个文件映射读入内存的API，从调用源头杜绝全量加载的可能
• 流式逐块计算：按固定的分片大小循环从文件句柄读取数据，每读出来一块小Data，直接传给哈希实例的update(data:)方法更新计算状态，这块分片数据用完就直接释放，不做任何内存留存
• 完成摘要计算：当读到文件末尾、没有更多数据可以读的时候，调用哈希实例的finalize()方法拿到最终的SHA256摘要值，再按需转成十六进制的标准字符串格式即可

落地关键注意事项 #

• 分片大小不用盲目选大或选小：16KB到256KB区间的分片就能兼顾IO调用效率和内存占用，实测64KB分片的综合表现最好，单块数据占64KB内存，对系统来说几乎可以忽略不计
• 一定要处理好异常场景的资源回收：哪怕读文件、计算过程中抛出错误，也要确保调用close()关闭打开的文件句柄，避免造成系统文件锁、句柄泄漏
• 不要自己手动实现哈希的分块拼接逻辑：CryptoKit的update方法已经处理好了所有流式计算的内部细节，自己额外拼接数据反而会增加不必要的内存占用

方案优势 #

• 完全基于苹果原生CryptoKit，哈希计算走系统内置的硬件加速通道，比第三方纯软件实现的哈希算法速度快2-3倍
• 内存占用和待计算文件大小完全脱钩，哪怕是几个GB的安装包、视频文件，计算全程的内存峰值也不会超过100KB，从根本上解决了大文件全量加载带来的内存过高问题

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