有时候,所见并不是所得,有些包,你需要去翻他的源码才知道为什么会这样。
背景
今天调试一个程序,用到了一个很久之前的NPM包,名为formstream(https://www.npmjs.com/package/formstream),用来将
form
表单数据转换为流的形式进行接口调用时的数据传递。
这是一个几年前的项目,所以使用的是
Generator
+
co
实现的异步流程。
其中有这样一个功能,从某处获取一些图片
URL
,并将
URL
以及一些其他的常规参数组装到一起,调用另外的一个服务,将数据发送过去。
大致是这样的代码:
const co = require('co')
const moment = require('moment')
const urllib = require('urllib')
const Formstream = require('formstream')
function * main () {
const imageUrlList = [
'img1',
'img2',
'img3'
,
]
// 实例化 form 表单对象
const form = new Formstream()
// 常规参数
form.field('timestamp', moment().unix())
// 将图片 URL 拼接到 form 表单中
imageUrlList.forEach(imgUrl => {
form.field('image', imgUrl)
})
const options = {
method: 'POST',
// 生成对应的 headers 参数
headers: form.headers(),
// 告诉 urllib,我们通过流的方式进行传递数据,并指定流对象
stream: form
}
// 发送请求
const result = yield urllib.request(url, options)
// 输出结果
console.log(result)
}
co(main)
也算是一个比较清晰的逻辑,这样的代码也正常运行了一段时间。
如果没有什么意外,这段代码可能还会在这里安静的躺很多年。 但是,现实总是残酷的,因为一些不可抗拒因素,必须要去调整这个逻辑。之前调用接口传递的是图片
URL
地址,现在要改为直接上传二进制数据。
所以需求很简单,就是将之前的
URL
下载,拿到
buffer
,然后将
buffer
传到
formstream
实例中即可。
大致是这样的操作:
- imageUrlList.forEach(imgUrl => {
- form.field(
'image', imgUrl)
- })
+ let imageUrlResults = yield Promise.all(imageUrlList.map(imgUrl =>
+ urllib.request(imgUrl)
+ ))
+
+ imageUrlResults = imageUrlResults.filter(img => img && img.status === 200).map(img => img.data)
+
+ imageUrlResults.forEach(imgBuffer => {
+ form.buffer('image', imgBuffer)
+ })
下载图片 -> 过滤空数据 -> 拼接到
form
中去,代码看起来毫无问题。
不过在执行的时候,却出现了一个令人头大的问题。
最终调用
yield
urllib
.
request
(
url
,
options
)
的时候,提示接口超时了,起初还以为是网络问题,于是多执行了几次,发现还是这样,开始意识到,应该是刚才的代码改动引发的
bug
。
开始 debug
定位引发 bug 的代码
我习惯的调试方式,是先用最原始的方式:
眼
,看有哪些代码修改。
因为代码都有版本控制,所以大多数编辑器都可以很直观的看到有什么代码修改,即使编辑器中无法看到,也可以在命令行中通过
git diff
来查看修改。
这次的改动就是新增的一个批量下载逻辑,以及
URL
改为
Buffer
。
先用最简单粗暴的方式来确认是这些代码影响的,
注释掉新增的代码,还原老代码
。结果果然是可以正常执行了,那么我们就可以断定
bug
就是由这些代码所导致的。
逐步还原错误代码
上边那个方式只是一个
rollback
,帮助确定了大致的范围。
接下来就是要缩小错误代码的范围。
一般代码改动大的时候,会有多个函数的声明,那么就按照顺序逐个解开注释,来查看运行的效果。
这次因为是比较小的逻辑调整,所以直接在一个函数中实现。
那么很简单的,在保证程序正常运行的前提下,我们就按照代码语句一行行的释放。
很幸运,在第一行代码的注释被打开后就复现了
bug
,也就是那一行
yield
Promsie
.
all
(
XXX
)
。
但是这个语句实际上也可以继续进行拆分,为了排除是
urllib
的问题,我将该行代码换为一个最基础的
Promise
对象:
yield
Promise
.
resolve
(
1
)
。
结果令我很吃惊,这么一个简单的
Promise
执行也会导致下边的请求超时。
当前的部分代码状态:
const form = new Formstream()
form.field('timestamp', moment().unix())
yield Promise.resolve(1)
const options = {
method: 'POST',
headers: form.headers(),
stream: form
}
// 超时
const result = yield urllib.request(url, options)
再缩小了范围以后,进一步进行排查。
目前所剩下的代码已经不错了,唯一可能会导致请求超时的情况,可能就是发请求时的那些
options
参数了。
所以将
options
中的
headers
和
stream
都注释掉,再次执行程序后,果然可以正常访问接口(虽说会提示出错,因为必选的参数没有传递)。
那么目前我们可以得到一个结论:
formstream
实例+
Promise
调用会导致这个问题。
冷静、忏悔
接下来要做的就是深呼吸,冷静,让心率恢复平稳再进行下一步的工作。
在我得到上边的结论之后,第一时间是崩溃的,因为导致这个
bug
的环境还是有些复杂的,涉及到了三个第三方包,
co
、
formstream
和
urllib
。
而直观的去看代码,自己写的逻辑其实是很少的,所以难免会在心中开始抱怨,觉得是第三方包在搞我。
但这时候要切记「程序员修炼之道」中的一句话:
"Select" Isn't Broken
“Select” 没有问题
所以一定要在内心告诉自己:“你所用的包都是经过了N久时间的洗礼,一定是一个很稳健的包,这个
bug
一定是你的问题”。
分析问题
当我们达成这个共识以后,就要开始进行问题的分析了。
首先你要了解你所使用的这几个包的作用是什么,如果能知道他们是怎么实现的那就更好了。
对于
co
,就是一个利用
yield
语法特性将
Promise
转换为更直观的写法罢了,没有什么额外的逻辑。
而
urllib
也会在每次调用
request
时创建一个新的
client
(刚开始有想过会不会是因为多次调用
urllib
导致的,不过用简单的
Promise
.
resolve
代替之后,这个念头也打消了)
那么矛头就指向了
formstream
,现在要进一步的了解它,不过通过官方文档进行查阅,并不能得到太多的有效信息。
源码阅读
源码地址:https://github.com/node-modules/formstream/blob/master/lib/formstream.js。
所以为了解决问题,我们需要去阅读它的源码,从你在代码中调用的那些 API 入手:
-
构造函数:https://github.com/node-modules/formstream/blob/master/lib/formstream.js#L38
-
field:https://github.com/node-modules/formstream/blob/master/lib/formstream.js#L133
-
headers:https://github.com/node-modules/formstream/blob/master/lib/formstream.js#L88
构造函数营养并不多,就是一些简单的属性定义,并且看到了它继承自
Stream
,这也是为什么能够在
urllib
的
options
中直接填写它的原因,因为是一个
Stream
的子类。
util.inherits(FormStream, Stream);
然后就要看
field
函数的实现了。
FormStream.prototype.field = function (name, value) {
if (!Buffer.isBuffer(value)) {
// field(String, Number)
// https://github.com/qiniu/nodejs-sdk/issues/123
if (typeof value === 'number') {
value
= String(value);
}
value = new Buffer(value);
}
return this.buffer(name, value);
};
从代码的实现看,
field
也只是一个
Buffer
的封装处理,最终还是调用了
.
buffer
函数。
那么我们就顺藤摸瓜,继续查看buffer(https://github.com/node-modules/formstream/blob/master/lib/formstream.js#L181)函数的实现。
FormStream.prototype.buffer = function (name, buffer, filename, mimeType) {
if (filename && !mimeType) {
mimeType = mime
.lookup(filename);
}
var disposition = { name: name };
if (filename) {
disposition.filename = filename;
}
var leading = this._leading(disposition, mimeType);
this._buffers.push([leading, buffer]);
// plus buffer length to total content-length
this._contentLength += leading.length;
this._contentLength += buffer.length;
this._contentLength += NEW_LINE_BUFFER.length;
process.nextTick(this.resume.bind(this));
return this;
};
代码不算少,不过大多都不是这次需要关心的,大致的逻辑就是将
Buffer
拼接到数组中去暂存,在最后结尾的地方,发现了这样的一句代码:
process.nextTick(this.resume.bind(this))
顿时眼前一亮,重点的是那个
process
.
nextTick
,大家应该都知道,这个是在
Node
中实现微任务的其中一个方式,而另一种实现微任务的方式,就是用
Promise
。
修改代码验证猜想
拿到这样的结果以后,我觉得仿佛找到了突破口,于是尝试性的将前边的代码改为这样:
const form = new Formstream()
form.field('timestamp', moment().unix())
yield
Promise.resolve(1)
const options = {
method: 'POST',
headers: form.headers(),
stream: form
}
process.nextTick(() => {
urllib.request(url, options)
})
发现,果然超时了。
从这里就能大致推断出问题的原因了。
因为看代码可以很清晰的看出,
field
函数在调用后,会注册一个微任务,而我们使用的
yield
或者
process
.
nextTick
也会注册一个微任务,但是
field
的先注册,所以它的一定会先执行。
那么很显而易见,问题就出现在这个
resume
函数中,因为
resume
的执行早于
urllib
.
request
,所以导致其超时。
这时候也可以同步的想一下造成
request
超时的情况会是什么。
只有一种可能性是比较高的,因为我们使用的是
stream
,而这个流的读取是需要事件来触发的,
stream
.
on
(
'data'
)
、
stream
.
on
(
'end'
)
,那么超时很有可能是因为程序没有正确接收到
stream
的事件导致的。
当然了,「程序员修炼之道」还讲过:
Don't Assume it - Prove It
不要假定,要证明
所以为了证实猜测,需要继续阅读
formstream
的源码,查看
resume
函数究竟做了什么。
resume
函数是一个很简单的一次性函数,在第一次被触发时调用
drain
函数。
FormStream.prototype.resume = function () {
this.paused = false;
if (!this._draining) {
this._draining = true;
this.drain();
}
return this;
};
那么继续查看
drain
函数做的是什么事情。
因为上述使用的是
field
,而非
stream
,所以在获取
item
的时候,肯定为空,那么这就意味着会继续调用
_emitEnd
函数。
而
_emitEnd
函数只有简单的两行代码
emit
(
'data'
)
和
emit
(
'end'
)
。
FormStream.prototype.drain = function () {
console.log('start drain')
this._emitBuffers();