如何重复执行nodejs mysql查询，直到结果出来？

Question

conn.connect(function (err) {
        if (err) throw err;
        var prizeCode = generatePrizeCode();// this method return a 5-digit code
        var query = "SELECT * FROM user WHERE code = ? ";
        var values = [[prizeCode]];
        conn.query(query, [values], function (err, result) {
            if (err) throw err;
            // Here I want to re-execute the above query if the result is not empty 
        });
    });

在上面的代码中，只要结果包含数据，我想执行查询。 但是我不能像下面的伪代码那样使用循环，

// if the result is empty, the generated code does not exist in database. So it can be used.
while(result.length != 0){
    var result = conn.query("SELECT * FROM user WHERE code = abc");
}

我该如何实现？

Answer 1

基本上，您将需要使用async \\ await或promises（几乎相同）。 https://javascript.info/async-await https://developer.mozilla.org/zh-CN/docs/Web/JavaScript/Reference/Global_Objects/Promise

这样，您将只能在获得正确结果时返回。

Answer 2

 var mysql = require('mysql'); var pool = mysql.createPool({ host : 'YOUR_HOST_NAME', user : 'YOUR_USER_NAME', password : 'YOUR_PASSWORD', database : 'YOUR_DATABASE' }); pool.getConnection(function(err, connection) { // connected! (unless `err` is set) }); connection = mysql.createPool var retryLimit=0; var getData = function () { if(retryLimit <= 50) { retryLimit++; connection.query('SELECT * FROM user WHERE code = abc', function (error, results, fields) { if (error) {throw error;} else if(result.length <1) { getData(); //call the same method again. } }); } } getData(); 

您可以添加else语句，然后执行所需的任何操作。

Answer 3

您将必须使用回调或promises / async函数。 这是如何使用回调编写的：

const mysql = require('mysql');

const retryLimit = 50;

const connection = mysql.createConnection({
  database: 'test',
  host: 'localhost',
  password: 'hunter2',
  user: 'dave',
});

function getData(attempts, cb) {
  if (attempts < retryLimit) { // we haven't exausted our attempts yet
    const code = generatePrizeCode();
    connection.query('SELECT * FROM user WHERE code = ?', code, (err, result) => {
      if (err) {
        return cb(err);
      }
      if (result.length > 0) { // code already exists
        getData(attempts + 1, cb); // recurse
      } else { // this is a new code
        cb(null, code); // return the new code via the callback function
      }
    });
  } else { // we have exausted our attempts
    return cb(new Error('retry limit exceeded'));
  }
}

getData(0, (err, code) => {
  // do what you want here, e.g., console.log(code)
  connection.end();
});

请注意，如果递归次数过多并超过最大调用堆栈，这可能会导致节点崩溃。 解决该问题的一种方法是调用setImmediate或setTimeout而不是直接递归。 代替线

getData(attempts + 1, cb); // recurse

采用

setImmediate(() => {
  getData(attempts + 1, cb); // recurse
});

要么

// recurse, using setImmediate every 1000th time
if (attempts % 1000 === 0) {
  setImmediate(() => {
    getData(attempts + 1, cb); // recurse
  });
} else {
  getData(attempts + 1, cb); // recurse
}

使用promise和async / await样式会清理很多东西，并且看起来更像是您习惯的方式（请注意，这使用了promise-mysql库）：

const mysql = require('promise-mysql');

async function getData(connection) {
  const retryLimit = 50;
  for (let i = 0; i < retryLimit; i++) {
    const code = generatePrizeCode();
    const result = await connection.query('SELECT * FROM user WHERE code = ?', code);
    if (result.length === 0) { // this code doesn't exist yet
      return code;
    }
  }
  throw new Error('retry limit exceeded');
}

(async () => {
  try {
    const connection = await mysql.createConnection({
      database: 'test',
      host: 'localhost',
      password: 'hunter2',
      user: 'dave',
    });
    try {
      const code = await getData(connection);
      // do what you want here, e.g., console.log(code)
    } finally {
      connection.end();
    }
  } catch (err) {
    console.log(err);
  }
})();