Stream 是一个抽象接口,Node 中有很多对象实现了这个接口。例如,对http 服务器发起请求的request 对象就是一个 Stream,还有stdout(标准输出)。
Node.js,Stream 有四种流类型:

  • Readable: 可读操作。
  • Writable: 可写操作。
  • Duplex: 可读可写操作.
  • Transform: 操作被写入数据,然后读出结果。

所有的 Stream 对象都是 EventEmitter 的实例。常用的事件有:

  • data: 当有数据可读时触发。
  • end: 没有更多的数据可读时触发。
  • error: 在接收和写入过程中发生错误时触发。
  • finish: 所有数据已被写入到底层系统时触发。

从流中读取数据

创建 input.txt 文件,内容如下:

  1. Hello,一缕清风

创建 main.js 文件, 代码如下:

  1. var fs = require("fs")
  2. var readerStream = fs.createReadStream('input.txt') // 创建可读流
  3. readerStream.setEncoding('UTF8') // 设置编码为 utf8
  4. var data = ''
  5. // 处理流事件 --> data, end, and error
  6. readerStream.on('data', function (chunk) {
  7. data += chunk
  8. })
  9. readerStream.on('end', function () {
  10. console.log(data)
  11. })
  12. readerStream.on('error', function (err) {
  13. console.log(err.stack)
  14. })
  15. /*
  16. Hello, 一缕清风
  17. */

写入流

创建 main.js 文件, 代码如下:

  1. var fs = require("fs");
  2. var data = 'Hello,一缕清风';
  3. var writerStream = fs.createWriteStream('output.txt')// 创建一个可以写入的流,写入到文件 output.txt 中
  4. writerStream.write(data, 'UTF8')// 使用 utf8 编码写入数据
  5. // 处理流事件 --> finish、error
  6. writerStream.on('finish', function () {
  7. console.log("写入完成")
  8. })
  9. writerStream.on('error', function (err) {
  10. console.log(err.stack);
  11. })
  12. writerStream.end() // 标记文件末尾

管道流

管道提供了一个输出流到输入流的机制。通常我们用于从一个流中获取数据并将数据传递到另外一个流中。image.png
如上面的图片所示,我们把文件比作装水的桶,而水就是文件里的内容,我们用一根管子(pipe)连接两个桶使得水从一个桶流入另一个桶,这样就慢慢的实现了大文件的复制过程。
以下实例我们通过读取一个文件内容并将内容写入到另外一个文件中。
设置 input.txt 文件内容如下:

  1. Hello,一缕清风

创建 main.js 文件, 代码如下:

  1. var fs = require("fs")
  2. var readerStream = fs.createReadStream('input.txt') // 创建一个可读流
  3. var writerStream = fs.createWriteStream('output.txt') // 创建一个可写流
  4. // 管道读写操作: 读取 input.txt 文件内容,并将内容写入到 output.txt 文件中
  5. readerStream.pipe(writerStream)
  6. console.log("程序执行完毕")

链式流

链式是通过连接输出流到另外一个流并创建多个流操作链的机制。链式流一般用于管道操作。
接下来我们就是用管道和链式来压缩和解压文件。
创建 compress.js 文件, 代码如下:

  1. var fs = require("fs")
  2. var zlib = require('zlib')
  3. // 压缩 input.txt 文件为 input.txt.gz
  4. fs.createReadStream('input.txt')
  5. .pipe(zlib.createGzip())
  6. .pipe(fs.createWriteStream('input.txt.gz'))

执行完以上操作后,我们可以看到当前目录下生成了 input.txt 的压缩文件 input.txt.gz。
接下来,让我们来解压该文件,创建 decompress.js 文件,代码如下:

  1. var fs = require("fs")
  2. var zlib = require('zlib')
  3. // 解压 input.txt.gz 文件为 input.txt
  4. fs.createReadStream('input.txt.gz')
  5. .pipe(zlib.createGunzip())
  6. .pipe(fs.createWriteStream('input.txt'))

创建流的常用方式

node.js中文件读取流方式

  1. var http = require('http');
  2. var fs = require("fs");
  3. var s1 = fs.createReadStream("input.txt");
  4. var data;
  5. s1.on("data", function (txt) {
  6. console.log(txt);
  7. data += txt;
  8. })
  9. s1.on("end", function (txt) {
  10. console.log("文件流式读取成功!")
  11. console.log(data)
  12. })

管道方式

  1. var fs = require("fs");
  2. var s1 = fs.createReadStream("input.txt");
  3. var s2 = fs.createWriteStream("output.txt");
  4. s1.pipe(s2)

获取请求的网站信息

  1. var https = require("https")
  2. var fs = require("fs")
  3. https.get("https://www.baidu.com/", function (res) {
  4. var data
  5. res.on("data", function (txt) {
  6. data += txt
  7. })
  8. res.on("end", function () {
  9. //console.log(data)
  10. fs.writeFile("input.txt", data, function () {
  11. console.log("文件写入完成!")
  12. })
  13. })
  14. })

批量下载图片

  1. var http = require("http");
  2. var fs = require("fs");
  3. var path = require("path"); //利用path模块,对url进行解析,以便得到想要的相关rootbase等信息;
  4. http.get("http://www.itsource.cn/", function (res) {
  5. var data;
  6. res.on("data", function (txt) {
  7. data += txt;
  8. })
  9. res.on("end", function () {
  10. var reg = /src="((.*?)\.jpg)"/img;
  11. var filename;
  12. while (filename = reg.exec(data)) {
  13. getImg(filename[1]);
  14. };
  15. // fs.writeFile("input.txt",filename[1],function(){
  16. // console.log("文件写入成功!")
  17. })
  18. })
  19. function getImg(url) {
  20. var obj = path.parse(url);
  21. var fn = obj.base;
  22. var stream1 = fs.createWriteStream("./files/" + fn);
  23. console.log(fn, stream1);
  24. if (obj.root.length === 0) {
  25. url = "/" + url;
  26. }
  27. url = "http://www.itsource.cn" + url;
  28. http.get(url, function (res) {
  29. res.pipe(stream1);
  30. console.log(fn + "读取完毕!")
  31. })
  32. }