opensource/sqlchat
1
0
Fork 0
mirror of https://github.com/sqlchat/sqlchat.git synced 2025-08-02 22:58:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: implement postgres connector

Browse Source
This commit is contained in:
steven
2023-03-23 17:51:31 +08:00
parent 39df26a08f
commit 9969c7fa2a
8 changed files with 222 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
lib/connectors/mysql/index.ts
View File

@ -1,7 +1,9 @@
import { Connection } from "@/types";
import mysql, { RowDataPacket } from "mysql2/promise";
import { Connection } from "@/types";
import { Connector } from "..";
const systemDatabases = ["information_schema", "mysql", "performance_schema", "sys"];
const convertToConnectionUrl = (connection: Connection): string => {
// Connection URL format: mysql://USER:PASSWORD@HOST:PORT/DATABASE
return `mysql://${connection.username}:${connection.password}@${connection.host}:${connection.port}/${connection.database ?? ""}`;
@ -21,7 +23,10 @@ const testConnection = async (connection: Connection): Promise<boolean> => {
const getDatabases = async (connection: Connection): Promise<string[]> => {
const connectionUrl = convertToConnectionUrl(connection);
const conn = await mysql.createConnection(connectionUrl);
const [rows] = await conn.query<RowDataPacket[]>("SELECT schema_name as db_name FROM information_schema.schemata;");
const [rows] = await conn.query<RowDataPacket[]>(
`SELECT schema_name as db_name FROM information_schema.schemata WHERE schema_name NOT IN (?);`,
[systemDatabases]
);
conn.destroy();
const databaseList = [];
for (const row of rows) {
@ -36,16 +41,17 @@ const getTables = async (connection: Connection, databaseName: string): Promise<
const connectionUrl = convertToConnectionUrl(connection);
const conn = await mysql.createConnection(connectionUrl);
const [rows] = await conn.query<RowDataPacket[]>(
`SELECT TABLE_NAME as table_name FROM information_schema.tables WHERE TABLE_SCHEMA='${databaseName}' AND TABLE_TYPE='BASE TABLE';`
`SELECT TABLE_NAME as table_name FROM information_schema.tables WHERE TABLE_SCHEMA=? AND TABLE_TYPE='BASE TABLE';`,
[databaseName]
);
conn.destroy();
const databaseList = [];
const tableList = [];
for (const row of rows) {
if (row["table_name"]) {
databaseList.push(row["table_name"]);
tableList.push(row["table_name"]);
}
}
return databaseList;
return tableList;
};
const getTableStructure = async (connection: Connection, databaseName: string, tableName: string): Promise<string> => {