node.js合并多个控制流

时间:2014-11-06 08:30:16

标签: mysql node.js

我正在慢慢开始越来越多地使用node.js,因为我发现它有更多的用途。我对它有一个很好的把握,使用express来进行简单的REST服务。我试图用它来编写一个小实用程序脚本。想象一下这个非常简单的例子,它获取mysql行,遍历每一行并更新它们然后打印" done"。请忽略一个事实,你可以用一个mysql查询显然做到这一点:

q=mysql_query("select id from table");
while($r=mysql_fetch_array($q)){
  mysql_query("update table set col1 = 'whatever' where id=".$r['id']);
}
print "done here!";

在node.js中执行此操作更加麻烦;我可以进行查询,遍历所有行并使用以下内容发出更新:

doQuery("select id from table",function(rows){
   for(var i=0;i<rows.length;i++){
      doQuery("update table set col1='whatever' where id="+rows[i]['id'],function(){
         //do I do something here? this will run as many times as rows were returned in the select
      })
   }
})

function doQuery(sql, callback){//utility function to help do the query
    var mysqlconnection = mysql.createConnection({
        host     : 'localhost',
        user     : 'user',
        password : '********',
        database:  'dbname'
    });
    mysqlconnection.connect(function(err1){
        mysqlconnection.query(sql, function(error, results) {
            if(callback)callback(results)
        })
    })
}

因此,对于此示例,我进行单select个查询,然后将控制流拆分为多个流。我如何将其归结为一个控制流程?或者,换句话说,我怎样才能简单地控制.log&#34;完成&#34;当所有updates完成后?

我看过承诺可能会有所帮助,但所有这些基本示例都显示了静态数量的查询,我的问题是动态的,并且基于mysql的结果。

谢谢!

2 个答案:

答案 0 :(得分:3)

您可以尝试使用async这个流行的nodejs模块来处理异步任务。

var async = require("async");

function doQuery(sql, callback){
  var mysqlconnection = mysql.createConnection({
    host     : 'localhost',
    user     : 'user',
    password : '********',
    database:  'dbname'
  });
  mysqlconnection.connect(function(err1){
    mysqlconnection.query(sql, function(error, results) {
      if(callback) callback(results);
    });
  });
}

function doUpdateQueryForEachRow(row, callback) {
  doQuery("update table set col1='whatever' where id=" + row.id, function (result) {
    callback(null, result); // the first parameter should be an error object
  });
}

doQuery("select id from table", function (rows) {
  async.map(rows, doUpdateQueryForEachRow, function (error, results) {
    // do something..
  });
});

答案 1 :(得分:0)

您应该使用迭代函数替换“for”循环:

function iterate (rowId) {
   if (rowId == rows.length - 1){
      // ... all updates done ...
   }

   doQuery("...update...", function () {
      iterate(rowId + 1);});
}(0);