基于 Promise 的递归函数中的问题

Question

我正在使用 nodejs 环境开发 AWS Lambda。我有一个 API，我在其中使用递归函数来实现某些功能。

实际上大多数人都说避免递归函数，但根据我的功能，我需要这个。我的功能如下。

我有一张表table1，其中有两列 pid 和 cid 被定义为唯一约束。这意味着两列的组合应该是唯一的。

因此，如果我在 table1 中插入任何组合并且该组合已经存在，那么它会给我重复输入错误，这根据我的功能是正确的。

所以为了处理这个重复输入错误，我使用了 try..catch 块。因此，在 catch 块中，我检查了是否发生重复条目错误，然后我正在调用一个递归函数，该函数尝试不同的组合，直到在 table1 中创建新条目。

我的递归函数是基于承诺的。但是当成功创建新条目时，我正在解决承诺。但是 promise 并没有从我第一次调用递归函数的地方返回。并因此发生超时。

所以请有人建议我解决方案，以便我的承诺得到解决，并且我的功能将从我第一次调用我的递归函数的地方继续。所以超时不会到来。

我提供我的代码以供参考。

var mysql = require('mysql');

var con = mysql.createConnection({
    "host": "somehost.com",
    "user": "myusername",
    "password": "mypassword",
    "database": "mydatabase"
});

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

    try {
        con.connect();
    } catch (error) {
        throw error;
        return 0;
    }

    let tableId = '';
    let count = '';

    try {

        var tempUsageData = {
            user_id: userId,
            code: code,
            platform: source,
            some_id: some_id,
            count: count
        };

        dbColumns = 'user_id, code, platform, added_on, count, some_id';
        let usageData = [
            [userId, code, source, new Date(), count, some_id]
        ];
        var tableInsert = await databaseInsert(con, constants.DB_CONSTANTS.DB_USAGE, dbColumns, usageData);
        tableId = tableInsert.insertId;
    } catch (error) {
        console.log('@@ error insert table1 @@', error);

        if (error.errno == 1062) {

            try {
                // calling recursive function here
                let newTableData = await createTableEntry(con, tempUsageData);
                tableId = newTableData.new_usage_id;
                count = newTableData.new_count;

            } catch (error) {
                console.log('Error', error);
                return 0;
            }

        } else {
            return 0;
        }
    };

    console.log('@@ EXECUTION DONE @@');
    return 1;

}

var createTableEntry = (con, dataObject) => {

    return new Promise(async function (resolve, reject) {
        console.log('createTableEntry Called for count', dataObject.count);
        try {
            var newCounter = await getDataFromDatabase(con, dataObject.some_id);

            dbColumns = 'user_id, code, platform, added_on, count, some_id';
            let tableData = [
                [userId, code, source, new Date(), Number(newCounter[0].counter + 1), some_id]
            ];
            var tableInsert = await databaseInsert(con, 'table1', dbColumns, tableData);


            let response = {
                new_table_id: tableInsert.insertId,
                new_count: Number(newCounter[0].counter + 1)
            }
            return resolve(response);

            //function not returning from here once successful entry done and timeout occures

        } catch (error) {
            console.log('@@ ERROR @@', error);

            if (error.errno == 1062) {

                console.log('@@ CALL FUNCTION AGAIN @@');

                dataObject.count = Number(newCounter[0].counter + 1);

                await createTableEntry(con, dataObject);

            } else {
                return reject(error);
            }
        }
    });
};

我的最终输出应该是消息“EXECUTION DONE”应该在执行完成后显示。

请为此建议我一个好的解决方案。提前致谢。

Answer 1

更新你的 catch 块

catch (error) {
            console.log('@@ ERROR @@', error);

            if (error.errno == 1062) {

                console.log('@@ CALL FUNCTION AGAIN @@');

                dataObject.count = Number(newCounter[0].counter + 1);

                let result = await createTableEntry(con, dataObject);
                return resolve(result);

            } else {
                return reject(error);
            }
        }