Node.js Stream (下)：通过文件流操作文件

2024-02-10 03:00:36

上一篇文章中，我们介绍了Node.js中的stream及其在网络请求中的应用。在这篇文章中，我们将重点讨论文件流的操作。文件流允许我们以更有效的方式读取和写入文件，尤其是当文件非常大的时候。

文件流

文件流与其他类型的流非常相似，它也是一个可读或可写的对象，可以从它读取数据或向它写入数据。然而，文件流有一些独特的特点，使其非常适合处理文件。

文件流是基于事件的，这意味着当数据可读或可写时，它们会触发事件。
文件流是双向的，这意味着它们可以同时读取和写入数据。
文件流是可暂停的，这意味着我们可以暂停数据流，然后在需要时继续。

读取文件流

为了读取文件流，我们可以使用fs.createReadStream()函数。此函数接受一个文件路径作为参数，并返回一个文件流对象。我们可以使用以下代码来读取文件流：

const fs = require('fs');

const readStream = fs.createReadStream('file.txt');

readStream.on('data', (chunk) => {
  console.log(chunk.toString());
});

readStream.on('end', () => {
  console.log('All data has been read.');
});

在上面的示例中，我们首先创建一个文件流对象。然后，我们使用on()方法监听data事件和end事件。当文件流中有数据可读时，data事件会被触发，并将数据块作为参数传递给事件处理程序。当文件流中的所有数据都被读取完时，end事件会被触发。

写入文件流

为了写入文件流，我们可以使用fs.createWriteStream()函数。此函数接受一个文件路径作为参数，并返回一个文件流对象。我们可以使用以下代码来写入文件流：

const fs = require('fs');

const writeStream = fs.createWriteStream('file.txt');

writeStream.write('Hello world!');

writeStream.end();

在上面的示例中，我们首先创建一个文件流对象。然后，我们使用write()方法将数据写入文件流。最后，我们使用end()方法结束文件流。

管道传输

管道传输是一种将一个流的数据直接传输到另一个流的技术。在Node.js中，我们可以使用pipe()方法来实现管道传输。例如，我们可以将文件流管道传输到控制台流，以便在控制台打印文件内容：

const fs = require('fs');

const readStream = fs.createReadStream('file.txt');
const writeStream = process.stdout;

readStream.pipe(writeStream);

在上面的示例中，我们将文件流管道传输到控制台流。这意味着文件流中的所有数据都会被直接传输到控制台。

自定义流

自定义流允许我们创建自己的stream对象。自定义流可以用于各种目的，例如，我们可以创建一个自定义流来处理压缩数据或加密数据。

为了创建一个自定义流，我们需要继承stream.Transform类。stream.Transform类是一个双向流，这意味着它可以同时读取和写入数据。我们可以使用以下代码来创建一个自定义流：

const { Transform } = require('stream');

class MyTransformStream extends Transform {
  _transform(chunk, encoding, callback) {
    // Do something with the data chunk
    this.push(chunk);

    callback();
  }
}

const myTransformStream = new MyTransformStream();

myTransformStream.on('data', (chunk) => {
  console.log(chunk.toString());
});

myTransformStream.write('Hello world!');

myTransformStream.end();

在上面的示例中，我们首先创建一个自定义流类，继承stream.Transform类。然后，我们实现_transform()方法，该方法将对数据块进行处理。最后，我们创建一个自定义流对象，并使用on()方法监听data事件。当自定义流中有数据可读时，data事件会被触发，并将数据块作为参数传递给事件处理程序。