knooq/lib/clients/mysql2.js

Generate schema objects intended for use with knex.js

module.exports = function createConnection(config) {
    let client;
    try {
        client = require('mysql2');
    } catch (e) {
        throw new Error('Client dependency (mssql2) not found. Run `npm install mssql2`')
    }
    let connection = client.createConnection(config.connection);

    const schemaName = getSchemaFromConfig(config);

    async function getTableNames() {
        return new Promise((resolve, reject) => {
            connection.query(`SELECT table_name
                        FROM information_schema.tables
                        WHERE table_schema = '${schemaName}'`,
                (err, results) => {
                    if (err) {
                        reject(err);
                    } else {
                        results = results.map(r => r.table_name);
                        resolve(results);
                    }
                });
        });
    }

    async function getColumnNamesForTable(tableName) {
        return new Promise((resolve, reject) => {
            connection.query(`SELECT column_name FROM information_schema.columns
                        WHERE table_schema = '${schemaName}'
                        AND table_name = '${tableName}'`,
                (err, results) => {
                    if (err) {
                        reject(err);
                    } else {
                        results = results.map(r => r.column_name);
                        resolve(results);
                    }
                });
        });
    }

    return {
        connection,
        getTableNames,
        getColumnNamesForTable,
    };
};

function getSchemaFromConfig(config) {
    let a = config.connection.split('/');
    return a[a.length - 1];
}