스크롤 기능 포함
This commit is contained in:
175
backend-node/src/database/MySQLConnector.ts
Normal file
175
backend-node/src/database/MySQLConnector.ts
Normal file
@@ -0,0 +1,175 @@
|
||||
import * as mysql from 'mysql2/promise';
|
||||
import { DatabaseConnector, ConnectionConfig, QueryResult } from '../interfaces/DatabaseConnector';
|
||||
import { ConnectionTestResult, TableInfo } from '../types/externalDbTypes';
|
||||
|
||||
export class MySQLConnector implements DatabaseConnector {
|
||||
private connection: mysql.Connection | null = null;
|
||||
private config: ConnectionConfig;
|
||||
|
||||
constructor(config: ConnectionConfig) {
|
||||
this.config = config;
|
||||
}
|
||||
|
||||
async connect(): Promise<void> {
|
||||
if (this.connection) {
|
||||
throw new Error('이미 연결되어 있습니다.');
|
||||
}
|
||||
|
||||
this.connection = await mysql.createConnection({
|
||||
host: this.config.host,
|
||||
port: this.config.port,
|
||||
database: this.config.database,
|
||||
user: this.config.username,
|
||||
password: this.config.password,
|
||||
connectTimeout: this.config.connectionTimeout || 30000,
|
||||
ssl: this.config.sslEnabled ? { rejectUnauthorized: false } : undefined,
|
||||
});
|
||||
}
|
||||
|
||||
async disconnect(): Promise<void> {
|
||||
if (this.connection) {
|
||||
await this.connection.end();
|
||||
this.connection = null;
|
||||
}
|
||||
}
|
||||
|
||||
async testConnection(): Promise<ConnectionTestResult> {
|
||||
const startTime = Date.now();
|
||||
try {
|
||||
await this.connect();
|
||||
|
||||
const [versionResult] = await this.connection!.query('SELECT VERSION() as version');
|
||||
const [sizeResult] = await this.connection!.query(
|
||||
'SELECT SUM(data_length + index_length) as size FROM information_schema.tables WHERE table_schema = DATABASE()'
|
||||
);
|
||||
|
||||
const responseTime = Date.now() - startTime;
|
||||
|
||||
return {
|
||||
success: true,
|
||||
message: 'MySQL 연결이 성공했습니다.',
|
||||
details: {
|
||||
response_time: responseTime,
|
||||
server_version: (versionResult as any)[0]?.version || '알 수 없음',
|
||||
database_size: this.formatBytes(parseInt((sizeResult as any)[0]?.size || '0')),
|
||||
},
|
||||
};
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
message: 'MySQL 연결에 실패했습니다.',
|
||||
error: {
|
||||
code: 'CONNECTION_FAILED',
|
||||
details: error instanceof Error ? error.message : '알 수 없는 오류',
|
||||
},
|
||||
};
|
||||
} finally {
|
||||
await this.disconnect();
|
||||
}
|
||||
}
|
||||
|
||||
async executeQuery(query: string): Promise<QueryResult> {
|
||||
if (!this.connection) {
|
||||
await this.connect();
|
||||
}
|
||||
|
||||
try {
|
||||
const [rows, fields] = await this.connection!.query(query);
|
||||
return {
|
||||
rows: rows as any[],
|
||||
rowCount: Array.isArray(rows) ? rows.length : 0,
|
||||
fields: (fields as mysql.FieldPacket[]).map(field => ({
|
||||
name: field.name,
|
||||
dataType: field.type.toString(),
|
||||
})),
|
||||
};
|
||||
} finally {
|
||||
await this.disconnect();
|
||||
}
|
||||
}
|
||||
|
||||
async getTables(): Promise<TableInfo[]> {
|
||||
if (!this.connection) {
|
||||
await this.connect();
|
||||
}
|
||||
|
||||
try {
|
||||
const [tables] = await this.connection!.query(`
|
||||
SELECT
|
||||
t.TABLE_NAME as table_name,
|
||||
t.TABLE_COMMENT as table_description
|
||||
FROM information_schema.TABLES t
|
||||
WHERE t.TABLE_SCHEMA = DATABASE()
|
||||
ORDER BY t.TABLE_NAME
|
||||
`);
|
||||
|
||||
const result: TableInfo[] = [];
|
||||
|
||||
for (const table of tables as any[]) {
|
||||
const [columns] = await this.connection!.query(`
|
||||
SELECT
|
||||
COLUMN_NAME as column_name,
|
||||
DATA_TYPE as data_type,
|
||||
IS_NULLABLE as is_nullable,
|
||||
COLUMN_DEFAULT as column_default
|
||||
FROM information_schema.COLUMNS
|
||||
WHERE TABLE_SCHEMA = DATABASE()
|
||||
AND TABLE_NAME = ?
|
||||
ORDER BY ORDINAL_POSITION
|
||||
`, [table.table_name]);
|
||||
|
||||
result.push({
|
||||
table_name: table.table_name,
|
||||
description: table.table_description || null,
|
||||
columns: columns as any[],
|
||||
});
|
||||
}
|
||||
|
||||
return result;
|
||||
} finally {
|
||||
await this.disconnect();
|
||||
}
|
||||
}
|
||||
|
||||
async getColumns(tableName: string): Promise<Array<{
|
||||
name: string;
|
||||
dataType: string;
|
||||
isNullable: boolean;
|
||||
defaultValue?: string;
|
||||
}>> {
|
||||
if (!this.connection) {
|
||||
await this.connect();
|
||||
}
|
||||
|
||||
try {
|
||||
const [columns] = await this.connection!.query(`
|
||||
SELECT
|
||||
COLUMN_NAME as name,
|
||||
DATA_TYPE as dataType,
|
||||
IS_NULLABLE = 'YES' as isNullable,
|
||||
COLUMN_DEFAULT as defaultValue
|
||||
FROM information_schema.COLUMNS
|
||||
WHERE TABLE_SCHEMA = DATABASE()
|
||||
AND TABLE_NAME = ?
|
||||
ORDER BY ORDINAL_POSITION
|
||||
`, [tableName]);
|
||||
|
||||
return (columns as any[]).map(col => ({
|
||||
name: col.name,
|
||||
dataType: col.dataType,
|
||||
isNullable: col.isNullable,
|
||||
defaultValue: col.defaultValue,
|
||||
}));
|
||||
} finally {
|
||||
await this.disconnect();
|
||||
}
|
||||
}
|
||||
|
||||
private formatBytes(bytes: number): string {
|
||||
if (bytes === 0) return '0 Bytes';
|
||||
const k = 1024;
|
||||
const sizes = ['Bytes', 'KB', 'MB', 'GB', 'TB'];
|
||||
const i = Math.floor(Math.log(bytes) / Math.log(k));
|
||||
return parseFloat((bytes / Math.pow(k, i)).toFixed(2)) + ' ' + sizes[i];
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user