1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380## Stream 流 从[早先的unix](http://www.youtube.com/watch?v=tc4ROCJYbm0)开始,stream便开始进入了人们的视野,在过去的几十年的时间里,它被证明是一种可依赖的编程方式,它可以将一个大型的系统拆成一些很小的部分,并且让这些部分之间完美地进行合作。在unix中,我们可以使用`|`符号来实现流。在node中,node内置的[stream模块](http://nodejs.org/docs/latest/api/stream.html)已经被多个核心模块使用,同时也可以被用户自定义的模块使用。和unix类似,node中的流模块的基本操作符叫做`.pipe()`,同时你也可以使用一个后压机制来应对那些对数据消耗较慢的对象。 ### 为什么应该使用流 在node中,I/O都是异步的,所以在和硬盘以及网络的交互过程中会涉及到传递回调函数的过程。你之前可能会写出这样的代码: ```js var http = require('http'); var fs = require('fs'); var server = http.createServer(function (req, res) { fs.readFile(__dirname + 'data.txt', function (err, data) { res.end(data); }); }); server.listen(8000); ``` 上面的这段代码并没有什么问题,但是在每次请求时,我们都会把整个`data.txt`文件读入到内存中,然后再把结果返回给客户端。想想看,如果`data.txt`文件非常大,在响应大量用户的并发请求时,程序可能会消耗大量的内存,这样很可能会造成用户连接缓慢的问题。 其次,上面的代码可能会造成很不好的用户体验,因为用户在接收到任何的内容之前首先需要等待程序将文件内容完全读入到内存中。 所幸的是,`(req,res)`参数都是流对象,这意味着我们可以使用一种更好的方法来实现上面的需求: ```js var http = require('http'); var fs = require('fs'); var server = http.createServer(function (req, res) { var stream = fs.createReadStream(__dirname + '/data.txt'); stream.pipe(res); }); server.listen(8000); ``` 在这里,`.pipe()`方法会自动帮助我们监听`data`和`end`事件。上面的这段代码不仅简洁,而且`data.txt`文件中每一小段数据都将源源不断的发送到客户端。 除此之外,使用`.pipe()`方法还有别的好处,比如说它可以自动控制后端压力,以便在客户端连接缓慢的时候node可以将尽可能少的缓存放到内存中。 想要将数据进行压缩?我们可以使用相应的流模块完成这项工作! ```js var http = require('http'); var fs = require('fs'); var oppressor = require('oppressor'); var server = http.createServer(function (req, res) { var stream = fs.createReadStream(__dirname + '/data.txt'); stream.pipe(oppressor(req)).pipe(res); }); server.listen(8000); ``` 通过上面的代码,我们成功的将发送到浏览器端的数据进行了gzip压缩。我们只是使用了一个oppressor模块来处理这件事情。 一旦你学会使用流api,你可以将这些流模块像搭乐高积木或者像连接水管一样拼凑起来,从此以后你可能再也不会去使用那些没有流API的模块获取和推送数据了。 ### 流模块基础 nodejs 底层一共提供了4个流, Readable 流、Writable 流、Duplex 流和 Transform 流。 使用情景 | 类 | 需要重写的方法 -------| ------| ------------- 只读 | Readable| _read 只写 | Writable | _write 双工 | Duplex | _read, _write 操作被写入数据,然后读出结果| Transform| _transform, _flush #### pipe 无论哪一种流,都会使用`.pipe()`方法来实现输入和输出。 `.pipe()`函数很简单,它仅仅是接受一个源头`src`并将数据输出到一个可写的流`dst`中: src.pipe(dst) `.pipe(dst)`将会返回`dst`因此你可以链式调用多个流: a.pipe(b).pipe(c).pipe(d) 上面的代码也可以等价为: a.pipe(b); b.pipe(c); c.pipe(d); 这和你在unix中编写流代码很类似: a | b | c | d 只不过此时你是在node中编写而不是在shell中! ### readable流 Readable流可以产出数据,你可以将这些数据传送到一个writable,transform或者duplex流中,只需要调用`pipe()`方法: readableStream.pipe(dst) #### 创建一个readable流 现在我们就来创建一个readable流! var Readable = require('stream').Readable; var rs = new Readable; rs.push('beep '); rs.push('boop\n'); rs.push(null); rs.pipe(process.stdout); 下面运行代码: $ node read0.js beep boop 在上面的代码中`rs.push(null)`的作用是告诉`rs`输出数据应该结束了。 需要注意的一点是我们在将数据输出到`process.stdout`之前已经将内容推送进readable流`rs`中,但是所有的数据依然是可写的。 这是因为在你使用`.push()`将数据推进一个readable流中时,一直要到另一个东西来消耗数据之前,数据都会存在一个缓存中。 然而,在更多的情况下,我们想要的是当需要数据时数据才会产生,以此来避免大量的缓存数据。 我们可以通过定义一个`._read`函数来实现按需推送数据: var Readable = require('stream').Readable; var rs = Readable(); var c = 97; rs._read = function () { rs.push(String.fromCharCode(c++)); if (c > 'z'.charCodeAt(0)) rs.push(null); }; rs.pipe(process.stdout); 代码的运行结果如下所示: $ node read1.js abcdefghijklmnopqrstuvwxyz 在这里我们将字母`a`到`z`推进了rs中,但是只有当数据消耗者出现时,数据才会真正实现推送。 `_read`函数也可以获取一个`size`参数来指明消耗者想要读取多少比特的数据,但是这个参数是可选的。 需要注意到的是你可以使用`util.inherit()`来继承一个Readable流。 为了说明只有在数据消耗者出现时,`_read`函数才会被调用,我们可以将上面的代码简单的修改一下: var Readable = require('stream').Readable; var rs = Readable(); var c = 97 - 1; rs._read = function () { if (c >= 'z'.charCodeAt(0)) return rs.push(null); setTimeout(function () { rs.push(String.fromCharCode(++c)); }, 100); }; rs.pipe(process.stdout); process.on('exit', function () { console.error('\n_read() called ' + (c - 97) + ' times'); }); process.stdout.on('error', process.exit); 运行上面的代码我们可以发现如果我们只请求5比特的数据,那么`_read`只会运行5次: $ node read2.js | head -c5 abcde _read() called 5 times 在上面的代码中,`setTimeout`很重要,因为操作系统需要花费一些时间来发送程序结束信号。 另外,`process.stdout.on('error',fn)`处理器也很重要,因为当`head`不再关心我们的程序输出时,操作系统将会向我们的进程发送一个`SIGPIPE`信号,此时`process.stdout`将会捕获到一个`EPIPE`错误。 上面这些复杂的部分在和操作系统相关的交互中是必要的,但是如果你直接和node中的流交互的话,则可有可无。 如果你创建了一个readable流,并且想要将任何的值推送到其中的话,确保你在创建流的时候指定了objectMode参数,`Readable({ objectMode: true })`。 #### 消耗一个readable流 大部分时候,将一个readable流直接pipe到另一种类型的流或者使用through或者concat-stream创建的流中,是一件很容易的事情。但是有时我们也会需要直接来消耗一个readable流。 process.stdin.on('readable', function () { var buf = process.stdin.read(); console.dir(buf); }); 代码运行结果如下所示: $ (echo abc; sleep 1; echo def; sleep 1; echo ghi) | node consume0.js <Buffer 61 62 63 0a> <Buffer 64 65 66 0a> <Buffer 67 68 69 0a> null 当数据可用时,`readable`事件将会被触发,此时你可以调用`.read()`方法来从缓存中获取这些数据。 当流结束时,`.read()`将返回`null`,因为此时已经没有更多的字节可以供我们获取了。 你也可以告诉`.read()`方法来返回`n`个字节的数据。虽然所有核心对象中的流都支持这种方式,但是对于对象流来说这种方法并不可用。 下面是一个例子,在这里我们制定每次读取3个字节的数据: process.stdin.on('readable', function () { var buf = process.stdin.read(3); console.dir(buf); }); 运行上面的例子,我们将获取到不完整的数据: $ (echo abc; sleep 1; echo def; sleep 1; echo ghi) | node consume1.js <Buffer 61 62 63> <Buffer 0a 64 65> <Buffer 66 0a 67> 这是因为多余的数据都留在了内部的缓存中,因此这个时候我们需要告诉node我们还对剩下的数据感兴趣,我们可以使用`.read(0)`来完成这件事: process.stdin.on('readable', function () { var buf = process.stdin.read(3); console.dir(buf); process.stdin.read(0); }); 到现在为止我们的代码和我们所期望的一样了! $ (echo abc; sleep 1; echo def; sleep 1; echo ghi) | node consume2.js <Buffer 61 62 63> <Buffer 0a 64 65> <Buffer 66 0a 67> <Buffer 68 69 0a> 我们也可以使用`.unshift()`方法来放置多余的数据。 使用`unshift()`方法能够放置我们进行不必要的缓存拷贝。在下面的代码中我们将创建一个分割新行的可读解析器: var offset = 0; process.stdin.on('readable', function () { var buf = process.stdin.read(); if (!buf) return; for (; offset < buf.length; offset++) { if (buf[offset] === 0x0a) { console.dir(buf.slice(0, offset).toString()); buf = buf.slice(offset + 1); offset = 0; process.stdin.unshift(buf); return; } } process.stdin.unshift(buf); }); 代码的运行结果如下所示: $ tail -n +50000 /usr/share/dict/american-english | head -n10 | node lines.js 'hearties' 'heartiest' 'heartily' 'heartiness' 'heartiness\'s' 'heartland' 'heartland\'s' 'heartlands' 'heartless' 'heartlessly' 当然,已经有很多这样的模块比如split来帮助你完成这件事情,你完全不需要自己写一个。 ### writable流 一个writable流指的是只能流进不能流出的流: src.pipe(writableStream) #### 创建一个writable流 只需要定义一个`._write(chunk,enc,next)`函数,你就可以将一个readable流的数据释放到其中: var Writable = require('stream').Writable; var ws = Writable(); ws._write = function (chunk, enc, next) { console.dir(chunk); next(); }; process.stdin.pipe(ws); 代码运行结果如下所示: $ (echo beep; sleep 1; echo boop) | node write0.js <Buffer 62 65 65 70 0a> <Buffer 62 6f 6f 70 0a> 第一个参数,`chunk`代表写进来的数据。 第二个参数`enc`代表编码的字符串,但是只有在`opts.decodeString`为`false`的时候你才可以写一个字符串。 第三个参数,`next(err)`是一个回调函数,使用这个回调函数你可以告诉数据消耗者可以写更多的数据。你可以有选择性的传递一个错误对象`error`,这时会在流实体上触发一个`emit`事件。 在从一个readable流向一个writable流传数据的过程中,数据会自动被转换为`Buffer`对象,除非你在创建writable流的时候制定了`decodeStrings`参数为`false`,`Writable({decodeStrings: false})`。 如果你需要传递对象,需要指定`objectMode`参数为`true`,`Writable({ objectMode: true })`。 #### 向一个writable流中写东西 如果你需要向一个writable流中写东西,只需要调用`.write(data)`即可。 process.stdout.write('beep boop\n'); 为了告诉一个writable流你已经写完毕了,只需要调用`.end()`方法。你也可以使用`.end(data)`在结束前再写一些数据。 var fs = require('fs'); var ws = fs.createWriteStream('message.txt'); ws.write('beep '); setTimeout(function () { ws.end('boop\n'); }, 1000); 运行结果如下所示: $ node writing1.js $ cat message.txt beep boop 如果你在创建writable流时指定了`highWaterMark`参数,那么当没有更多数据写入时,调用`.write()`方法将会返回false。 如果你想要等待缓存情况,可以监听`drain`事件。 ### duplex流 Duplex流是一个可读也可写的流,全双工。 如图:  代码实现上: ```js const Readable = require('_stream_readable'); const Writable = require('_stream_writable'); util.inherits(Duplex, Readable); var keys = Object.keys(Writable.prototype); for (var v = 0; v < keys.length; v++) { var method = keys[v]; if (!Duplex.prototype[method]) Duplex.prototype[method] = Writable.prototype[method]; } ``` `Duplex` 首先继承了 `Readable`, 因为 javascript 没有 C++的多重继承的特性,所以 遍历 `Writable`的原型方法然后赋值到 `Duplex`的原型上。 ### transform流 转换流(Transform streams)是一种输出由输入计算所得的双工流。它同时实现了 Readable 和 Writable 接口。 Node中的转换流有: * zlib streams * crypto streams 你可以将transform流想象成一个流的中间部分,它可以读也可写,但是并不保存数据,它只负责处理流经它的数据。 ### 总结 流式处理的优势: 将功能切分,并通过管道组合。 ### 参考 https://github.com/substack/stream-handbook