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堆限制；

　　　　　简单，容易实现。

　　缺点：需要转换成字符串，不会转的还可能出现乱码。

巴特西

koa1.x获取原始body内容

最新文章

热门文章