Merge pull request #208 from antares-sql/feat/postgre-import-export

feat(PostgreSQL): import/export
antares-sql · Apr 7, 2022 · 40828cb · 40828cb
2 parents d92facf + 6086ca4
commit 40828cb
Show file tree

Hide file tree

Showing 15 changed files with 779 additions and 39 deletions.
diff --git a/package.json b/package.json
@@ -120,6 +120,7 @@
     "moment": "^2.29.1",
     "mysql2": "^2.3.2",
     "pg": "^8.7.1",
+    "pg-query-stream": "^4.2.3",
     "pgsql-ast-parser": "^7.2.1",
     "source-map-support": "^0.5.20",
     "spectre.css": "^0.5.9",

diff --git a/src/common/customizations/defaults.js b/src/common/customizations/defaults.js
@@ -39,6 +39,7 @@ module.exports = {
    schemaEdit: false,
    schemaDrop: false,
    schemaExport: false,
+   exportByChunks: false,
    schemaImport: false,
    tableSettings: false,
    tableOptions: false,

diff --git a/src/common/customizations/mysql.js b/src/common/customizations/mysql.js
@@ -35,6 +35,7 @@ module.exports = {
    schemaEdit: true,
    schemaDrop: true,
    schemaExport: true,
+   exportByChunks: true,
    schemaImport: true,
    tableSettings: true,
    viewSettings: true,

diff --git a/src/common/customizations/postgresql.js b/src/common/customizations/postgresql.js
@@ -31,9 +31,9 @@ module.exports = {
    routineAdd: true,
    functionAdd: true,
    schemaDrop: true,
+   schemaExport: true,
+   schemaImport: true,
    databaseEdit: false,
-   schemaExport: false,
-   schemaImport: false,
    tableSettings: true,
    viewSettings: true,
    triggerSettings: true,

diff --git a/src/main/libs/clients/PostgreSQLClient.js b/src/main/libs/clients/PostgreSQLClient.js
@@ -572,7 +572,7 @@ export class PostgreSQLClient extends AntaresCore {
     * @memberof MySQLClient
     */
    async dropSchema (params) {
-      return await this.raw(`DROP SCHEMA "${params.database}"`);
+      return await this.raw(`DROP SCHEMA "${params.database}" CASCADE`);
    }
 
    /**

diff --git a/src/main/libs/exporters/sql/MysqlExporter.js b/src/main/libs/exporters/sql/MysqlExporter.js
@@ -6,6 +6,12 @@ import moment from 'moment';
 import { lineString, point, polygon } from '@turf/helpers';
 
 export default class MysqlExporter extends SqlExporter {
+   constructor (...args) {
+      super(...args);
+
+      this._commentChar = '#';
+   }
+
    async getSqlHeader () {
       let dump = await super.getSqlHeader();
       dump += `