分享一下Node中的stream之用法

Stream也就是流，我们可以把它理解为一连串的字节像水流那样的存在。流是用于在 Node.js 中处理流数据，也就是连续字节的抽象接口。 流有 4 种基本类型，本篇文章主要介绍其中两种 —— 可读流和可写流，下面一起来看一下。

分享一下Node中的stream之用法

可读的（Readable）

我们可以通过 fs.createReadStream() 创建一个可读流 readableStream，第 1 个参数可以是要读取的文件路径，比如现有文件 test.txt 内容为：

hello juejin

第 2 个参数（可选）可以传入一个选项对象，用来控制读取数据的起止位置等： 

const fs = require('fs')
 
const readableStream = fs.createReadStream('./test.txt', {
  start: 6,
  end: 11
})

注意，读取的数据是包括了 start 和 end 的。 在之前介绍 events 模块时说过，所有的流都是 EventEmitter 的实例。所以获取数据是通过监听 'data' 事件：

readableStream.on('data', data => {
  console.log(data)
  console.log(data.toString())
})

注意，读取数据时一次最多读取 64 * 1024 字节，如果想改变该数值，可以通过 highWaterMark 选项更改。另外还可以使用 pause() 进行暂停操作，使用 resume() 继续读取：

const readableStream = fs.createReadStream('./test.txt', {
  start: 6,
  end: 11,
  highWaterMark: 2 // 默认为 64 * 1024
})
 
readableStream.on('data', data => {
  console.log(data)
  console.log(data.toString())
  readableStream.pause() // 暂停读取
  setTimeout(() => {
    readableStream.resume() // 恢复读取
  }, 2000)
})

除了 'data' 事件，可读流还有其它一些事件，比如监听文件被打开的 'open'（回调会被传入 fd 参数），监听文件读取到 end（默认为读取到最后）的 'end'，监听文件关闭的 'close'（文件读取完会自动关闭）：

readableStream.on('data', data => console.log(data))
readableStream.on('open', fd => console.log(`${fd}文件打开了`))
readableStream.on('end', () => console.log('文件读取到 end 位置了'))
readableStream.on('close', () => console.log('文件已关闭'))

可写的（Writable）

使用 fs.createWriteStream() 可以创建可写流，第 1 个参数传入要写入的文件，第 2 个同样是用于配置的可选参数，这里我们将 flags 由表示覆盖写入的 'w' 改为 'a'，即追加写入（如果还指定了写入的起始位置 start，则为了兼容 windows 系统， flags 应该写成 'r+'）：

const writableStream = fs.createWriteStream('test.txt', {
  flags: 'a'
})
 
writableStream.write('养成', err => console.log(`错误信息：${err}`)) // 错误信息：undefined
writableStream.write('写作的')
writableStream.end('好习惯')

写入数据是通过 write 方法，上方代码执行的结果就是在原本的 "hello juejin" 后添加上了 "养成写作的好习惯"。最后一次写入可以使用 end 方法，这样就能监听到可写流的 'close' 事件了：

writableStream.on('close', () => {
  console.log('关闭')
})

否则，可写流需要手动 writableStream.close() 关闭才能监听到 'close' 事件。如果写入成功，则作为 write()/end() 的第 2 个参数传入的回调的 err 就为 undefined。

可读流还有另外一些事件，比如文件打开时的 'open' 事件，当 close() 或 end() 被调用后文件写入完成的 'finish' 事件。

可读流与可写流的连接

可读流与可写流可以通过 pipe 方法进行连接，实现对文件的复制：

const readableStream = fs.createReadStream('./test.txt')
const writableStream = fs.createWriteStream('./copy.txt')
 
readableStream.pipe(writableStream)

如此，便能将 ./test.txt 内的文本复制到 ./copy.txt 中。

关于Node中的stream之用法到这里就结束了，翼速应用平台内有更多相关资讯，欢迎查阅！