koa1.x获取原始body内容
Node版本比较老,koa1.x配合koa-body-parser,默认koa-body-parser会把请求数据转成json对象,
然而有的时候需要获取原始的内容,不要转换,看波koa-body-parser源码,找到办法。
办法一:设置请求头Content-type值为:text/plain
这样ctx.request.body就是一个字符串了.
缺点:要毅端加东西,想到前端的懒惰,较困难,放弃。
继续摸索源码,发现一个办法,代码:
const koa = require('koa'); const router = require('koa-router'); const bodyParser = require('koa-body-parser'); const app = new koa(); const a = new router(); function injectRawBody(rawBodyName) { return function* (next) { const ctx = this; let _rawBody = yield function (done) { let received = ''; function onData(chunk) { received += chunk; } function onEnd() { ctx.req.removeListener('data', onData); ctx.req.removeListener('end', onEnd); done(null, received); } } ctx.request[rawBodyName] = _rawBody; yield next; } } app.use(injectRawBody('rawBody')); app.use(bodyParser(); const server = require('http').createServer(app.callback()); app.use(a.routes()); a.post('/a', function* () { const ctx = this; const body = ctx.request.body; console.log("ctx.request.body:", body); console.log("type of ctx.request.body:", typeof body); console.log("raw body:", Buffer.concat(ctx.request.rawBody).toString()); console.log("type of raw body:", typeof ctx.request.rawBody); yield tt.addOrCreate.apply(ctx); ctx.body = 'success'; }); server.listen(3000);
运行,看输出,成功把原始内容挂在ctx.request.rawBody上,但ctx.request.body却成了undefined,点解?
经过细查,发现在koa-body-parser/index.js有一句判断:
因为在上一个中间件,已经耗尽了req对象的数据,所以ended就是true,那也就不会进入之后的解析了。为了一点小私,导致依赖body的功能不能用,不行不行。
既然我放在body-parser前面不行,那我能放到后面么?像这样:
app.use(bodyParser());
app.use(injectRawBody('rawBody'));
答案是不行,因为在body-parser设置了req.once,而且即使没有设置once,body-parser只有耗尽了req对象(end事件)才会继续下一个中间件,下一个中间件不能再从已经读完的ReadableStream里重新读数据。
再细细看波源码,co、koa-body-parser、co-body、raw-body真没发现什么好方法,这几个库之间的协作太无鏠了,基本上没法hack进去。
想过伪装一个req对象,跟ctx.req有一样的接口,让koa-body-parser里操作的是这个伪装的对象,完了之后再恢复,但没试出来,暂时放弃。
束手无措之际,突然想起晚上在stackoverflow上看到有人问过同样的问题,回忆了一下其中一个回答,当时觉得没怎么样,现在想想真是可以用,代码:
const koa = require('koa'); const router = require('koa-router'); const bodyParser = require('koa-body-parser');
const app = new koa(); const a = new router(); function injectRawBody(rawBodyName) { return function* (next) { const ctx = this; let received = []; function data(chunk) { received.push(chunk); } ctx.req.on('data', data); ctx.request[rawBodyName] = received; yield next; } } app.use(injectRawBody('rawBody')); app.use(bodyParser(); const server = require('http').createServer(app.callback()); app.use(a.routes()); a.post('/a', function* () { const ctx = this; //console.log(ctx.request); const body = ctx.request.body; console.log("ctx.request.body:", body); console.log("type of ctx.request.body:", typeof body); console.log("raw body:", Buffer.concat(ctx.request.rawBody).toString()); console.log("type of raw body:", typeof ctx.request.rawBody); yield tt.addOrCreate.apply(ctx); ctx.body = 'success'; }); server.listen(3000);
在injectRawBody中间件,绑定了data事件,接收数据。而body-parser只有在end事件发生时才会resolve,所有在injectRawBody里,获取到的是完事的数据,只是存的是buffer,使用的时候需要转换为字符串。
优点:没有body-parser那样的限制数据大小,buffer想放多少放多少,不受node堆限制;
简单,容易实现。
缺点:需要转换成字符串,不会转的还可能出现乱码。