sql-agent / src /utils /database.ts
fullstuckdev's picture
Upload 23 files
645ad9d verified
import mysql from 'mysql2/promise';
export async function createConnection(uri: string) {
try {
const connection = await mysql.createConnection(uri);
return connection;
} catch (error) {
throw new Error('Failed to connect to database');
}
}
export async function executeQuery(connection: mysql.Connection, query: string) {
try {
const [rows] = await connection.execute(query);
return rows;
} catch (error) {
throw new Error(`Failed to execute query: ${error}`);
}
}
export async function getTables(connection: mysql.Connection) {
try {
const [rows] = await connection.execute('SHOW TABLES');
return Object.values(rows).map((row: any) => Object.values(row)[0]);
} catch (error) {
throw new Error('Failed to fetch tables');
}
}
export async function getTablesList(connection: mysql.Connection) {
try {
const [rows] = await connection.execute('SHOW TABLES');
return Object.values(rows).map((row: any) => Object.values(row)[0]);
} catch (error) {
throw new Error('Failed to fetch tables list');
}
}