如何在Node.js中流式处理大JSON文件

开发 前端
本文介绍一个概念 SAX 的设计模式,这个概念虽然不是来源于 Node.js,但它解决问题的一些思想当我们在使用 Node.js 或一些其它的编程语言中遇到类似问题时也会受到一些启发。

解决一个问题不只要搜寻最终的答案,寻找答案的过程同样也是重要的,善于思考与总结总归是好的。

本文介绍一个概念 SAX 的设计模式,这个概念虽然不是来源于 Node.js,但它解决问题的一些思想当我们在使用 Node.js 或一些其它的编程语言中遇到类似问题时也会受到一些启发,本文后面会介绍如何流式处理一个大 JSON 文件,下面先给出了两个问题,可以先思考下如果是你会怎么做?

场景描述

问题一:假设现在有一个场景,有一个大的 JSON 文件,需要读取每一条数据经过处理之后输出到一个文件或生成报表数据,怎么能够流式的每次读取一条记录?

  1.   {"id": 1}, 
  2.   {"id": 2}, 
  3.   ... 

问题二:同样一个大的 JSON 文件,我只读取其中的某一块数据,想只取 list 这个对象数组怎么办?

  1.  "list": [], 
  2.   "otherList": [] 

在 Node.js 中我们可以基于以下几种方式读取数据,也是通常首先能够想到的:

  • fs.readFile():这个是一次性读取数据到内存,数据量大了都占用到内存也不是好办法,很容易造成内存溢出。
  • fs.createReadStream():创建一个可读流,能解决避免大量数据占用内存的问题,这是一个系统提供的基础 API 读取到的是一个个的数据块,因为我们的 JSON 对象是结构化的,也不能直接解决上面提的两个问题。
  • 还有一个 require() 也可以加载 JSON 文件,但是稍微熟悉点 Node.js CommonJS 规范的应该知道 require 加载之后是会缓存的,会一直占用在服务的内存里。

了解下什么是 SAX

SAX 是 Simple API for XML 的简称,目前没有一个标准的 SAX 参考标准,最早是在 Java 编程语言里被实现和流行开的,以 Java 对 SAX 的实现后来也被认为是一种规范。其它语言的实现也是遵循着该规则,尽管每门语言实现都有区别,但是这里有一个重要的概念 “事件驱动” 是相同的。

实现了 SAX 的解析器拥有事件驱动那样的 API,像 Stream 的方式来工作,边读取边解析,用户可以定义回调函数获取数据,无论 XML 内容多大,内存占用始终都会很小。

这对我们本节有什么帮助?我们读取解析一个大 JSON 文件的时候,也不能把所有数据都加载到内存里,我们也需要一个类似 SAX 这样的工具帮助我们实现。

基于 SAX 的流式 JSON 解析器

这是一个流式 JSON 解析器 https://github1s.com/creationix/jsonparse 周下载量在 600 多万,但是这个源码看起来很难梳理。如果是学习,推荐一个基于 SAX 的更简单版本

https://gist.github.com/creationix/1821394 感兴趣的可以看看。

JSON 是有自己的标准的,有规定的数据类型、格式。这个 JSON 解析器也是在解析到特定的格式或类型后触发相应的事件,我们在使用时也要注册相应的回调函数。

下面示例,创建一个可读流对象,在流的 data 事件里注册 SaxParser 实例对象的 parse 方法,也就是将读取到的原始数据(默认是 Buffer 类型)传递到 parse() 函数做解析,当解析到数据之后触发相应事件。

对应的 Node.js 代码如下:

  1. const SaxParser = require('./jsonparse').SaxParser; 
  2. const p = new SaxParser({ 
  3.   onNull: function () { console.log("onNull") }, 
  4.   onBoolean: function (value) { console.log("onBoolean", value) }, 
  5.   onNumber: function (value) { console.log("onNumber", value) }, 
  6.   onString: function (value) { console.log("onString", value) }, 
  7.   onStartObject: function () { console.log("onStartObject") }, 
  8.   onColon: function () { console.log("onColon") }, 
  9.   onComma: function () { console.log("onComma") }, 
  10.   onEndObject: function () { console.log("onEndObject") }, 
  11.   onStartArray: function () { console.log("onEndObject") }, 
  12.   onEndArray: function () { console.log("onEndArray") } 
  13. }); 
  14.  
  15. const stream = require('fs').createReadStream("./example.json"); 
  16. const pparse = p.parse.bind(p); 
  17. stream.on('data', parse); 

怎么去解析一个 JSON 文件的数据已经解决了,但是如果直接这样使用还是需要在做一些处理工作的。

JSONStream 处理大文件

这里推荐一个 NPM 模块 JSONStream,在它的实现中就是依赖的 jsonparse 这个模块来解析原始的数据,在这基础之上做了一些处理,根据一些匹配模式返回用户想要的数据,简单易用。

下面我们用 JSONStream 解决上面提到的两个问题。

问题一:

假设现在有一个场景,有一个大的 JSON 文件,需要读取每一条数据经过处理之后输出到一个文件或生成报表数据,怎么能够流式的每次读取一条记录?

因为测试,所以我将 highWaterMark 这个值调整了下,现在我们的数据是下面这样的。

  1.   { "id": 1 }, 
  2.   { "id": 2 } 

重点是 JSONStream 的 parse 方法,我们传入了一个 '.',这个 data 事件也是该模块自己处理过的,每次会为我们返回一个对象:

  • 第一次返回 { id: 1 }
  • 第二次返回 { id: 2 }
  1. const fs = require('fs'); 
  2. const JSONStream = require('JSONStream'); 
  3.  
  4. (async () => { 
  5.   const readable = fs.createReadStream('./list.json', { 
  6.     encoding: 'utf8', 
  7.     highWaterMark: 10 
  8.   }) 
  9.   const parser = JSONStream.parse('.'); 
  10.   readable.pipe(parser); 
  11.   parser.on('data', console.log); 
  12. })() 

问题二:

同样一个大的 JSON 文件,我只读取其中的某一块数据,想只取 list 这个数组对象怎么办?

解决第二个问题,现在我们的 JSON 文件是下面这样的。

  1.   "list": [ 
  2.     { "name": "1" }, 
  3.     { "name": "2" } 
  4.   ], 
  5.   "other": [ 
  6.     { "key": "val" } 
  7.   ] 

与第一个解决方案不同的是改变了 parse('list.*') 方法,现在只会返回 list 数组,other 是不会返回的,其实在 list 读取完成之后这个工作就结束了。

  • 第一次返回 { name: '1' }
  • 第二次返回 { name: '2' }
  1. (async () => { 
  2.   const readable = fs.createReadStream('./list.json', { 
  3.     encoding: 'utf8', 
  4.     highWaterMark: 10 
  5.   }) 
  6.   const parser = JSONStream.parse('list.*'); 
  7.   readable.pipe(parser); 
  8.   parser.on('data', console.log); 
  9. })(); 

总结

当我们遇到类似的大文件需要处理时,尽可能避免将所有的数据存放于内存操作,应用服务的内存都是有限制的,这也不是最好的处理方式。

文中主要介绍如何流式处理类似的大文件,更重要的是掌握编程中的一些思想,例如 SAX 一个核心点就是实现了 “事件驱动” 的设计模式,同时结合 Stream 做到边读取边解析。

处理问题的方式是多样的,还可以在生成 JSON 文件时做拆分,将一个大文件拆分为不同的小文件。

学会寻找答案,NPM 生态发展的还是不错的,基本上你能遇到的问题大多已有一些解决方案了,例如本次问题,不知道如何使用 Stream 来读取一个 JSON 文件时,可以在 NPM 上搜索关键词尝试着找下。

 

 

责任编辑:赵宁宁 来源: Nodejs技术栈
相关推荐

2021-07-15 10:15:52

Node.jsJSON前端

2021-07-03 17:43:03

Node.jsNode变量

2021-07-30 11:20:53

JavaScriptNode.jsWeb Develop

2020-08-05 08:31:51

SSL TLSNode.js

2020-10-26 08:34:13

Node.jsCORS前端

2023-10-04 07:35:03

2011-09-09 14:23:13

Node.js

2020-08-24 08:07:32

Node.js文件函数

2022-11-17 09:52:12

RHEL 9Node.js

2021-12-25 22:29:57

Node.js 微任务处理事件循环

2021-10-25 09:00:37

Node.jsJS前端

2020-05-29 15:33:28

Node.js框架JavaScript

2021-09-07 07:53:43

工具

2022-08-12 07:01:00

Node.jsXSS脚本

2013-11-01 09:34:56

Node.js技术

2015-03-10 10:59:18

Node.js开发指南基础介绍

2021-08-20 16:05:28

JavaScript node.js 应用安全

2020-04-20 16:00:05

Node.js框架JavaScript

2022-09-04 15:54:10

Node.jsAPI技巧

2021-05-27 09:00:00

Node.js开发线程
点赞
收藏

51CTO技术栈公众号