mirror of https://github.com/Fabio286/antares.git
113 lines
3.2 KiB
TypeScript
113 lines
3.2 KiB
TypeScript
import SSHConfig from '@fabio286/ssh2-promise/lib/sshConfig';
|
|
import * as antares from 'common/interfaces/antares';
|
|
import { ImportOptions } from 'common/interfaces/importer';
|
|
import * as mysql from 'mysql2';
|
|
import * as pg from 'pg';
|
|
|
|
import { MySQLClient } from '../libs/clients/MySQLClient';
|
|
import { PostgreSQLClient } from '../libs/clients/PostgreSQLClient';
|
|
import { ClientsFactory } from '../libs/ClientsFactory';
|
|
import MySQLImporter from '../libs/importers/sql/MySQLlImporter';
|
|
import PostgreSQLImporter from '../libs/importers/sql/PostgreSQLImporter';
|
|
let importer: antares.Importer;
|
|
|
|
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
process.on('message', async ({ type, dbConfig, options }: {
|
|
type: string;
|
|
dbConfig: mysql.ConnectionOptions & { schema: string; ssl?: mysql.SslOptions; ssh?: SSHConfig; readonly: boolean }
|
|
| pg.ClientConfig & { schema: string; ssl?: mysql.SslOptions; ssh?: SSHConfig; readonly: boolean }
|
|
| { databasePath: string; readonly: boolean };
|
|
options: ImportOptions;
|
|
}) => {
|
|
if (type === 'init') {
|
|
try {
|
|
const connection = await ClientsFactory.getClient({
|
|
client: options.type,
|
|
params: {
|
|
...dbConfig,
|
|
schema: options.schema
|
|
},
|
|
poolSize: 1
|
|
}) as MySQLClient | PostgreSQLClient;
|
|
|
|
const pool = await connection.getConnectionPool();
|
|
|
|
switch (options.type) {
|
|
case 'mysql':
|
|
case 'maria':
|
|
importer = new MySQLImporter(pool as unknown as mysql.Pool, options);
|
|
break;
|
|
case 'pg':
|
|
importer = new PostgreSQLImporter(pool as unknown as pg.PoolClient, options);
|
|
break;
|
|
default:
|
|
process.send({
|
|
type: 'error',
|
|
payload: `"${options.type}" importer not aviable`
|
|
});
|
|
return;
|
|
}
|
|
|
|
importer.once('error', err => {
|
|
console.error(err);
|
|
process.send({
|
|
type: 'error',
|
|
payload: err.toString()
|
|
});
|
|
});
|
|
|
|
importer.once('end', () => {
|
|
process.send({
|
|
type: 'end',
|
|
payload: { cancelled: importer.isCancelled }
|
|
});
|
|
});
|
|
|
|
importer.once('cancel', () => {
|
|
process.send({ type: 'cancel' });
|
|
});
|
|
|
|
importer.on('progress', state => {
|
|
process.send({
|
|
type: 'import-progress',
|
|
payload: state
|
|
});
|
|
});
|
|
|
|
importer.on('query-error', state => {
|
|
process.send({
|
|
type: 'query-error',
|
|
payload: state
|
|
});
|
|
});
|
|
|
|
importer.run();
|
|
}
|
|
catch (err) {
|
|
console.error(err);
|
|
process.send({
|
|
type: 'error',
|
|
payload: err.toString()
|
|
});
|
|
}
|
|
}
|
|
else if (type === 'cancel')
|
|
importer.cancel();
|
|
});
|
|
|
|
process.on('uncaughtException', (err) => {
|
|
console.error(err);
|
|
process.send({
|
|
type: 'error',
|
|
payload: err.toString()
|
|
});
|
|
});
|
|
|
|
process.on('unhandledRejection', (err) => {
|
|
console.error(err);
|
|
process.send({
|
|
type: 'error',
|
|
payload: err.toString()
|
|
});
|
|
});
|