opensource/sqlchat
1
0
Fork 0
mirror of https://github.com/sqlchat/sqlchat.git synced 2025-07-28 17:53:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: support execute sql

Browse Source
This commit is contained in:
steven
2023-03-28 11:26:30 +08:00
parent c32a971bf2
commit a41f42b32b
8 changed files with 225 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lib/connectors/postgres/index.ts
View File

@ -27,6 +27,14 @@ const testConnection = async (connection: Connection): Promise<boolean> => {
}
};
const execute = async (connection: Connection, _: string, statement: string): Promise<any> => {
const client = newPostgresClient(connection);
await client.connect();
const { rows } = await client.query(statement);
await client.end();
return rows;
};
const getDatabases = async (connection: Connection): Promise<string[]> => {
const client = newPostgresClient(connection);
await client.connect();
@ -75,6 +83,7 @@ const getTableStructure = async (connection: Connection, _: string, tableName: s
const newConnector = (connection: Connection): Connector => {
return {
testConnection: () => testConnection(connection),
execute: (databaseName: string, statement: string) => execute(connection, databaseName, statement),
getDatabases: () => getDatabases(connection),
getTables: (databaseName: string) => getTables(connection, databaseName),
getTableStructure: (databaseName: string, tableName: string) => getTableStructure(connection, databaseName, tableName),