Вложенные запросы в MYSQL и nodejs Async Issue

Question

Я хочу выполнить SQL-запрос для каждой строки другого запроса, поэтому я написал такую ​​функцию, как

export function getLocations(req, res) {
let appData = [];
const database = new Database();
database.query('select * from districts')
    .then(rows => {
        rows.forEach(row => {
            const new_database = new Database();
            new_database.query(`SELECT locations.id,locations.name
            FROM locations 
            .....
            WHERE locations.district=?`, row.id)
                .then(sub_rows => {
                    let district = row;
                    district["locations"] = sub_rows;
                    appData.push(district);
                    new_database.close();
                }, err => {
                    return new_database.close().then(() => { throw err; })
                })
                .catch(err => {
                    console.log(err);
                    res.status(500).json("Database Error");
                })
    }, err => {
        return database.close().then(() => { throw err; })
    }).then(result => {
        //final result
        res.status(200).json(appData);
        database.close()
    })
    .catch(err => {
        console.log(err);
        res.status(500).json("Database Error");
    })
}

и мой класс базы данных

const mysql = require('mysql');

class Database {
constructor() {
    this.connection = mysql.createConnection({
        host: process.env.DB_HOST,
        user: process.env.DB_USERNAME,
        password: process.env.DB_PASSWORD,
        database: process.env.DB_NAME,
        port: 3306,
        debug: false,
        multipleStatements: false
    });
}
query(sql, args) {
    return new Promise((resolve, reject) => {
        this.connection.query(sql, args, (err, rows) => {
            if (err)
                return reject(err);
            resolve(rows);
        });
    });
}
close() {
    return new Promise((resolve, reject) => {
        this.connection.end(err => {
            if (err)
                return reject(err);
            resolve();
        });
    });
}
}
module.exports = Database;

Это, однако, при выполнении продолжается и выводит конечный результат (пустой массив), не дожидаясь выполнения подзапросов. Я попытался выполнить асинхронное ожидание в строке и new_database.query , но все еще не ожидает завершения выполнения подзапроса. Что я могу сделать здесь?

Answer 1

Я улучшил вашу логику для sub-queries и переписал их, используя Promise.all () :

export function getLocations(req, res) {
    let appData = [];
    const database = new Database();
    database.query('select * from districts')
        .then(rows => {

            // create promise list for all sub-queries
            const promise_list = rows.map(row => new Promise((resolve, reject) => {
                const new_database = new Database();
                return new_database.query(`SELECT locations.id,locations.name FROM locations .... WHERE locations.district=?`, row.id)
                    .then(sub_rows => {
                        let district = row;
                        district["locations"] = sub_rows;
                        appData.push(district);
                        new_database.close();
                        resolve(district);
                    }, err => {
                        return new_database.close().then(() => { throw err; })
                    });
            })
            );

            // execute all sub-queries in parallel and wait for them
            Promise.all(promise_list).then(result => {
                res.status(200).json(appData);
                database.close()
            }).catch(err => {
                console.log(err);
                res.status(500).json("Database Error");
            });

        });
}

PS: Вы можете объявить appData как const, вы просто помещаете в него данные. Не переназначать его.

Answer 2

Используйте JOIN, чтобы попытаться получить желаемый результат. Какие функции доступны, зависит от версии MySQL, поэтому вам может потребоваться GROUP_CONCAT вместо JSON_ARRAYAGG

SELECT districts.*, JSON_ARRAYAGG(locations)
FROM districts
LEFT JOIN locations
   ON locations.district=districts.id
GROUP BY districts.id

Соединения Javascript имеют ужасную производительность, поэтому позвольте базе данных делать то, для чего она предназначена.