AWS Lambda Node.js 环境无法在第一次 API 调用时循环插入到 Mysql

Question

我正在尝试在 AWS Lambda (node.js) 中创建函数，它调用一些 REST API，并将 API 结果插入 MySQL 数据库。

虽然要求很简单，但是我在部署到 AWS Lambda 时遇到了一些问题（在我的本地机器上没有发生），我的第一个 API 调用只插入了 1 个数据，而第二个 API 调用前转，它按预期插入所有 4 个数据。我尝试了各种可用的堆栈溢出解决方案，结果都是一样的。

另一个问题是结果总是{"message": "Internal server error"}，即使数据在第二次 API 呼叫转移

上正确插入

基本上我对Node.js没有太多经验，所以如果有人能帮助我，我将不胜感激。

 'use strict';

const connection = require('serverless-mysql')({
    config: {
      host: 'xxxxxx.xxxxx.ap-southeast-1.rds.amazonaws.com',
      user: 'xxx',
      password: 'xxx',
      database: 'xxx_db'
    }
})
const axios = require('axios');

exports.handler = (event, context) => {

  //Get Data From API
  axios.get('https://xxx.xyz/wp-json/wp/v2/posts')
  .then(res => {
    const headerDate = res.headers && res.headers.date ? res.headers.date : 'no response date';
    console.log('Status Code:', res.status);
    console.log('Date in Response header:', headerDate);

    //this should result 4 data
    const posts = res.data;
      
    posts.forEach(post => {
        var sql = `INSERT INTO tbl_post(news_id, title, excerpt, content, category, image_link, modified_date, show_in_banner_F, show_in_list_F) VALUES ('${post.id}', '${post.title.rendered}', '${post.excerpt.rendered}', '${post.content.rendered}', '', '${post.yoast_head_json.og_image[0].url}', now(), 0, 0)`;  
        console.log(sql);
        let insert_query = connection.query(sql);
    });

    console.log("finished");
    connection.end();
    
    let responseBody = { message: "OK" };
    let response = {
        statusCode: 200,
        headers: {
            "Access-Control-Allow-Headers" : "Content-Type",
            "Access-Control-Allow-Origin": "*",
            "Access-Control-Allow-Methods": "OPTIONS,POST,GET"
        },
        body: JSON.stringify(responseBody)
    };
    return response;

  })
  .catch(err => {
    console.log('Error: ', err.message);
    let responseBody = { message: "Fail" };
    let response = {
        statusCode: 200,
        headers: {
            "Access-Control-Allow-Headers" : "Content-Type",
            "Access-Control-Allow-Origin": "*",
            "Access-Control-Allow-Methods": "OPTIONS,POST,GET"
        },
        body: JSON.stringify(responseBody)
    };
    return response;
  });
}

Answer 1

首先，forEach 循环会多次调用 connection.query(sql) 函数，然后退出循环，而不会实际等待每个查询完成执行，因此您最终会执行随机数量的查询时间你运行这个循环而不是你想要做的是使用 async/await await connection.query(sql) 以便在退出循环之前等待循环内的每个查询完成执行。

此外 forEach 循环不是为异步代码设计的，因此您也必须更改它并改用 for...of。而且您还必须使用 prepared statements 使用 ? 而不是使用 ${variable} 插入值以防止 sql 注入。

for (const post of posts) {
   const sql = "INSERT INTO tbl_post(news_id, title) VALUES (?, ?)";
   const values = [post.id, post.title.rendered];
   console.log(mysql.format(sql,values)); // This would log query after values substitution 
   await connection.execute(sql, values);
 }

因此最终代码将如下所示：

exports.handler = async (event, context) => {
  try {
    //Get Data From API
    const res = await axios.get("https://xxx.xyz/wp-json/wp/v2/posts");
    const headerDate = res.headers && res.headers.date ? res.headers.date : "no response date";
    console.log("Status Code:", res.status);
    console.log("Date in Response header:", headerDate);

    //this should result 4 data
    const posts = res.data;
    for (const post of posts) {
      const sql = "INSERT INTO tbl_post(news_id, title) VALUES (?, ?)";
      const values = [post.id, post.title.rendered];
      console.log(mysql.format(sql, values)); // This would log query after values substitution
      await connection.execute(sql, values); // Execute prepares statement first then executes it.
    }

    console.log("finished");
    await connection.end();

    let responseBody = { message: "OK" };
    let response = {
      statusCode: 200,
      headers: {
        "Access-Control-Allow-Headers": "Content-Type",
        "Access-Control-Allow-Origin": "*",
        "Access-Control-Allow-Methods": "OPTIONS,POST,GET",
      },
      body: JSON.stringify(responseBody),
    };
    return response;
  } catch (err) {
    console.log("Error: ", err.message);
    let responseBody = { message: "Fail" };
    let response = {
      statusCode: 200,
      headers: {
        "Access-Control-Allow-Headers": "Content-Type",
        "Access-Control-Allow-Origin": "*",
        "Access-Control-Allow-Methods": "OPTIONS,POST,GET",
      },
      body: JSON.stringify(responseBody),
    };
    return response;
  }
};

附带说明，如果您想保证循环内的所有查询成功或失败，请考虑使用事务。

专业提示：如果您想同时执行多个异步函数而不是一个接一个地执行，请使用 Promise.all()。

AWS Lambda Node.js 环境无法在第一次 API 调用时循环插入到 Mysql

AWS Lambda Node.js Environment Failed to Loop Insert to Mysql on First API Call

mysql

node.js

aws-lambda

serverless