使用 Node.js 同步数据库查询

Question

我有一个 Node.js/Express 应用程序，它查询路由中的 MySQL 数据库并向用户显示结果。我的问题是如何运行查询并阻止直到两个查询都完成后再将用户重定向到他们请求的页面？

在我的示例中，我有 2 个查询需要在渲染页面之前完成。如果我将查询 2 嵌套在查询 1 的“结果”回调中，我可以让查询同步运行。但是，当查询数量增加时，这将变得非常复杂。

如何同步运行多个（在本例中为 2 个）数据库查询，而不将后续查询嵌套在前一个查询的“结果”回调中？

我查看了 Node 模块中的“流程控制/异步好东西”并尝试了 flow-js，但我无法让它与异步查询一起使用。

下面列出了我尝试从“/home”路由执行的 2 个查询。节点专家能否解释执行此操作的“正确”方法？

app.get('/home', function (req,res) {
    var user_array = [];
    var title_array = [];

    // first query
    var sql = 'select user_name from users';
    db.execute(sql)
        .addListener('row', function(r) {
            user_array.push( { user_name: r.user_name } );
        })
        .addListener('result', function(r) {
            req.session.user_array = user_array;
        });

    // second query
    var sql = 'select title from code_samples';
    db.execute(sql)
        .addListener('row', function(r) {
            title_array.push( { title: r.title } );
        })
        .addListener('result', function(r) {
            req.session.title_array = title_array;
        });

        // because the queries are async no data is returned to the user
        res.render('home.ejs', {layout: false, locals: { user_name: user_array, title: title_array }});
});

Answer 1

节点的目标不是关心事情发生的顺序。这可能会使某些场景变得复杂。嵌套回调并不可耻。一旦你习惯了它的外观，你可能会发现你实际上更喜欢这种风格。我愿意;非常清楚将触发什么订单回调。如果有必要，您可以放弃匿名函数以使其不那么冗长。

如果您愿意稍微重构一下代码，则可以使用“典型”嵌套回调方法。如果您想避免回调，有许多异步框架可以尝试帮助您做到这一点。您可能想要查看的一个是 async.js (https://github.com/fjakobs/async.js)。每个示例：

app.get('/home', function (req,res) {
    var lock = 2;
    var result = {};
    result.user_array = [];
    result.title_array = [];

    var finishRequest = function(result) {
        req.session.title_array = result.title_array;
        req.session.user_array = result.user_array;
        res.render('home.ejs', {layout: false, locals: { user_name: result.user_array, title: result.title_array }});
    };

    // first query
    var q1 = function(fn) {
      var sql = 'select user_name from users';
      db.execute(sql)
          .addListener('row', function(r) {
              result.user_array.push( { user_name: r.user_name } );
          })
          .addListener('result', function(r) {
              return fn && fn(null, result);
        });
    };

    // second query
    var q2 = function(fn) {
      var sql = 'select title from code_samples';
      db.execute(sql)
          .addListener('row', function(r) {
              result.title_array.push( { title: r.title } );
          })
          .addListener('result', function(r) {
              return fn && fn(null, result);
          });
    }

    //Standard nested callbacks
    q1(function (err, result) {
      if (err) { return; //do something}

      q2(function (err, result) {
        if (err) { return; //do something}

        finishRequest(result);
      });
    });

    //Using async.js
    async.list([
        q1,
        q2,
    ]).call().end(function(err, result) {
      finishRequest(result);
    });

});

对于一次性，我可能只使用引用计数类型的方法。只需跟踪您想要执行的查询数量，并在它们全部完成后呈现响应即可。

app.get('/home', function (req,res) {
    var lock = 2;
    var user_array = [];
    var title_array = [];

    var finishRequest = function() {
        res.render('home.ejs', {layout: false, locals: { user_name: user_array, title: title_array }});
    }

    // first query
    var sql = 'select user_name from users';
    db.execute(sql)
        .addListener('row', function(r) {
            user_array.push( { user_name: r.user_name } );
        })
        .addListener('result', function(r) {
            req.session.user_array = user_array;
            lock -= 1;

            if (lock === 0) {
              finishRequest();
            }
        });

    // second query
    var sql = 'select title from code_samples';
    db.execute(sql)
        .addListener('row', function(r) {
            title_array.push( { title: r.title } );
        })
        .addListener('result', function(r) {
            req.session.title_array = title_array;
            lock -= 1;

            if (lock === 0) {
              finishRequest();
            }
        });
});

更好的方法是在渲染响应之前在每个“结果”回调中简单地调用 finishRequest() 来检查非空数组。这是否适合您的情况取决于您的要求。

Answer 2

这是处理多个回调的一个非常简单的技巧。

var after = function _after(count, f) {
  var c = 0, results = [];
  return function _callback() {
    switch (arguments.length) {
      case 0: results.push(null); break;
      case 1: results.push(arguments[0]); break;
      default: results.push(Array.prototype.slice.call(arguments)); break;
    }
    if (++c === count) {
      f.apply(this, results);
    }
  };
};

示例

用法：

var handleDatabase = after(2, function (res1, res2) {
  res.render('home.ejs', { locals: { r1: res1, r2: res2 }):
})

db.execute(sql1).on('result', handleDatabase);
db.execute(sql2).on('result', handleDatabase);

所以基本上你需要引用计数。这是这些情况下的标准方法。我实际上使用这个小实用函数而不是流量控制。

如果您想要一个完整的流量控制解决方案，我会推荐 futuresJS

Answer 3

我发现异步库最适合此类事情。 https://github.com/caolan/async#parallel

我无法测试这个或任何东西，所以如果有一些错别字请原谅。我重构了您的查询函数以使其可重用。因此，调用 queryRows 将返回一个与异步模块并行回调函数的格式匹配的函数。两个查询完成后，它将调用最后一个函数并将两个查询的结果作为参数传递，您可以读取该结果并将其传递给模板。

function queryRows(col, table) {
  return function(cb) {
    var rows = [];
    db.execute('SELECT ' + col + ' FROM ' + table)
      .on('row', function(r) {
        rows.push(r)        
      })
      .on('result', function() {
        cb(rows);
      });
  }
}

app.get('/home', function(req, res) {
  async.parallel({
    users: queryRow('user_name', 'users'),
    titles: queryRow('title', 'code_samples')
  },
  function(result) {
    res.render('home.ejs', { 
      layout: false,
      locals: {user_name: result.users, title: result.titles} 
    });
  });
});

Answer 4

这里有一些解决方案，但在我看来，最好的解决方案是以一种非常简单的方式使代码同步。

您可以使用“synchonize”包。

只是

npm 安装同步

然后 varsync = require(synchronize);

使用

sync. Fiber(function() { 将需要同步的逻辑放入 Fiber 中
//把你的逻辑放在这里
}


两个 mysql 查询的示例：

var express = require('express');
var bodyParser = require('body-parser');
var mysql = require('mysql');
var sync = require('synchronize');

var db = mysql.createConnection({
    host     : 'localhost',
    user     : 'user',
    password : 'password',
    database : 'database'
});

db.connect(function(err) {
    if (err) {
        console.error('error connecting: ' + err.stack);
        return;
    }
});

function saveSomething() {
    var post  = {id: newId};
    //no callback here; the result is in "query"
    var query = sync.await(db.query('INSERT INTO mainTable SET ?', post, sync.defer()));
    var newId = query.insertId;
    post  = {foreignKey: newId};
    //this query can be async, because it doesn't matter in this case
    db.query('INSERT INTO subTable SET ?', post, function(err, result) {
        if (err) throw err;
    });
}

当调用“saveSomething()”时，它会在主表中插入一行并接收最后插入的 id。之后将执行下面的代码。不需要嵌套的承诺或类似的东西。

Answer 5

选项一：如果所有查询都相互关联，则创建存储过程，将所有数据逻辑放入其中，并使用单个 db.execute

选项二：如果您的数据库使用一个连接，则命令一个有保证的连接串行执行，您可以将其用作异步助手

db.execute(sql1).on('row', function(r) {
   req.session.user_array.push(r.user);
});
db.execute(sql2)
.on('row', function(r) {
   req.session.title_array.push(r.title);
})
.on('end'), function() {
   // render data from req.session
});

Answer 6

您可以使用 Fiber 与 Node.JS 编写伪同步代码，看看这些 DB 测试 http://alexeypetrushin.github.com/synchronize