如何从 Node.js API 正确发送响应代码

Question

我有一个简单的基于节点的 API，它需要解析一些 JSON，将一些数据保存到 Postgres，然后发送适当的响应代码（如 http 201）。 我的代码如下所示：

router.route('/customer')

    .post(function(req, res) {
        Customers = req.body;
        var numberOfCustomers = Customers.length;
        for(var i = 0; i < Customers.length; i++){
            Customer = Customers[i];
            console.log(Customer.Name  + "  " + Customer.Address);
            var date = moment(new Date()).unix();

            client.query(
                'INSERT into customer (name, address, date_modified) VALUES($1, $2, $3) RETURNING id',
                [Customer.Name, Customer.Address, date],
                function(err, result) {
                    if (err) {
                        console.log(err);
                        status = 1;
                    } else {
                        console.log('row inserted with id: ' + result.rows[0].id);
                        if(numberOfCustomers === i) {
                            res.status(201).send({ message: "created" });
                        }
                    }
                });
        }
    })

我收到此错误：

_

http_outgoing.js:344
    throw new Error('Can\'t set headers after they are sent.');
    ^

Error: Can't set headers after they are sent.
    at ServerResponse.OutgoingMessage.setHeader (_http_outgoing.js:344:11)

我需要考虑这样一个事实，即我在一个循环中多次执行 Postgres 插入，所以在第一次插入完成后我无法发送响应标头。

我的“POST”处理程序中最适合放置我的res.status(201).send({ message: "created" }); 的位置是什么？ ?

Answer 1

除了架构决策（例如，您可能需要一个单独的模块作为 HTTP 适配器来处理发送响应代码的逻辑，而不是在路由控制器内部执行），您可以使用 Promise 来等待所有插入完成，然后发送一个 single 响应代码。例如，像这样：

var Promise = require('bluebird');
var query = Promise.promisify(client.query);

router.route('/customer')
.post(function(req, res) {
   // all your logic, and then

   return Promise.all(Customers.map(function() {
     return query(sql, [Customer.Name, Customer.Address, date]);
   })
   .then(function() {
     res.status(201).send({ message: 'Created' });
   });
});

查看the bluebird docs 以获取此示例中使用的 API。

我不熟悉 Postgres 的 API，但概念应该类似：您需要等待所有对您的 DB 的请求都被首先解决。

Answer 2

如上所述：是的，诸如 Promises 和 async 之类的异步助手对此类事务很有帮助。但是，我确实相信解决这个问题的“最佳”方法是只使用一个查询。而不是每个查询只执行一次插入，而是将它们全部批处理到一个查询中，如下所示：

INSERT into customer (name, address, date_modified)
VALUES
   ($1, $2, $3),
   ($4, $5, $6),
   ($7, $8, $9),
   ...
RETURNING id'

建议

router.route('/customer').post(function(req, res) {

    //Fetch customers
    var customers = req.body;

    //Store parameters and query inserts for db-query.
    var params = [];
    var inserts = [];

    //For each customer
    // - Add parameters for query
    // - Build insert string
    customers.forEach(function(customer){
       inserts.push(
         [
           "($",
           params.push(customer.Name),
           ", $",
           params.push(customer.Address),
           ", ",
           NOW(), //unnecessary to generate timestamp in js
           ")",
         ].join('')
       )
    });

    //Build query
    var query = 'INSERT into customer (name, address, date_modified) VALUES '+ inserts +' RETURNING id';

    //Query database in a more simple fashion.
    client.query(query, params, function(err, result) {
        if (err) {
            console.log(err);
            status = 1;
        } else {
            res.status(201).send({ message: "created" });
        });
    }
})

如果您使用的是 ES6，则可以通过使用字符串模板来简化字符串构建操作。

customers.forEach(function(customer){
    var query = `($${params.push(customer.Name)}, $${params.push(customer.Address)}, NOW())`
    inserts.push(query);
});

//and

var query = `
    INSERT into customer (name, address, date_modified)
    VALUES ${inserts}
    RETURNING id
`;

Answer 3

正确的做法是，我也建议您查看Async or lodash lib。

router.route('/customer')

.post(function(req, res) {
    var Customers = req.body,
        numberOfCustomers = Customers.length;

    for(var i = 0; i < Customers.length; i++){
        var Customer = Customers[i];
        console.log(Customer.Name  + "  " + Customer.Address);
        var date = moment(new Date()).unix(),
            sql = 'INSERT into customer (name, address, date_modified) VALUES($1, $2, $3) RETURNING id';

        client.query(sql, [Customer.Name, Customer.Address, date],
            function(err, result) {
                if (err) {
                    console.log(err);
                    res.status(500).json({message: "Server Error", err: err});
                } else {
                    console.log('row inserted with id: ' + result.rows[0].id);
                    if (numberOfCustomers === i) {
                        res.status(201).send({ message: "Created" });
                    }
                }
            });
    }
})