titor/drawDB
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix import failing on single statements (#244)

Browse Source
This commit is contained in:
1ilit 2024-09-21 21:24:56 +04:00
parent f635d0774c
commit b5e23fef20
5 changed files with 36 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/utils/importSQL/mariadb.js
View File

@ -23,7 +23,7 @@ export function fromMariaDB(ast, diagramDb = DB.GENERIC) {
const tables = []; const tables = [];
const relationships = []; const relationships = [];
ast.forEach((e) => { const parseSingleStatement = (e) => {
if (e.type === "create") { if (e.type === "create") {
if (e.keyword === "table") { if (e.keyword === "table") {
const table = {}; const table = {};
@ -250,7 +250,13 @@ export function fromMariaDB(ast, diagramDb = DB.GENERIC) {
} }
}); });
} }
}); };
if (Array.isArray(ast)) {
ast.forEach((e) => parseSingleStatement(e));
} else {
parseSingleStatement(ast);
}
relationships.forEach((r, i) => (r.id = i)); relationships.forEach((r, i) => (r.id = i));

2
src/utils/importSQL/mssql.js
View File

@ -9,7 +9,7 @@ const affinity = {
), ),
[DB.GENERIC]: new Proxy( [DB.GENERIC]: new Proxy(
{ {
INT: "INTEGER", INTEGER: "INT",
TINYINT: "SMALLINT", TINYINT: "SMALLINT",
MEDIUMINT: "INTEGER", MEDIUMINT: "INTEGER",
BIT: "BOOLEAN", BIT: "BOOLEAN",

12
src/utils/importSQL/mysql.js
View File

@ -9,7 +9,7 @@ const affinity = {
), ),
[DB.GENERIC]: new Proxy( [DB.GENERIC]: new Proxy(
{ {
INT: "INTEGER", INTEGER: "INT",
TINYINT: "SMALLINT", TINYINT: "SMALLINT",
MEDIUMINT: "INTEGER", MEDIUMINT: "INTEGER",
BIT: "BOOLEAN", BIT: "BOOLEAN",
@ -23,7 +23,7 @@ export function fromMySQL(ast, diagramDb = DB.GENERIC) {
const tables = []; const tables = [];
const relationships = []; const relationships = [];
ast.forEach((e) => { const parseSingleStatement = (e) => {
if (e.type === "create") { if (e.type === "create") {
if (e.keyword === "table") { if (e.keyword === "table") {
const table = {}; const table = {};
@ -250,7 +250,13 @@ export function fromMySQL(ast, diagramDb = DB.GENERIC) {
} }
}); });
} }
}); };
if (Array.isArray(ast)) {
ast.forEach((e) => parseSingleStatement(e));
} else {
parseSingleStatement(ast);
}
relationships.forEach((r, i) => (r.id = i)); relationships.forEach((r, i) => (r.id = i));

12
src/utils/importSQL/postgres.js
View File

@ -9,7 +9,7 @@ const affinity = {
), ),
[DB.GENERIC]: new Proxy( [DB.GENERIC]: new Proxy(
{ {
INT: "INTEGER", INTEGER: "INT",
MEDIUMINT: "INTEGER", MEDIUMINT: "INTEGER",
BIT: "BOOLEAN", BIT: "BOOLEAN",
}, },
@ -23,7 +23,7 @@ export function fromPostgres(ast, diagramDb = DB.GENERIC) {
const types = []; const types = [];
const enums = []; const enums = [];
ast.forEach((e) => { const parseSingleStatement = (e) => {
if (e.type === "create") { if (e.type === "create") {
if (e.keyword === "table") { if (e.keyword === "table") {
const table = {}; const table = {};
@ -315,7 +315,13 @@ export function fromPostgres(ast, diagramDb = DB.GENERIC) {
} }
}); });
} }
}); };
if (Array.isArray(ast)) {
ast.forEach((e) => parseSingleStatement(e));
} else {
parseSingleStatement(ast);
}
relationships.forEach((r, i) => (r.id = i)); relationships.forEach((r, i) => (r.id = i));

12
src/utils/importSQL/sqlite.js
View File

@ -23,7 +23,7 @@ const affinity = {
), ),
[DB.GENERIC]: new Proxy( [DB.GENERIC]: new Proxy(
{ {
INT: "INTEGER", INTEGER: "INT",
TINYINT: "SMALLINT", TINYINT: "SMALLINT",
MEDIUMINT: "INTEGER", MEDIUMINT: "INTEGER",
INT2: "INTEGER", INT2: "INTEGER",
@ -40,7 +40,7 @@ export function fromSQLite(ast, diagramDb = DB.GENERIC) {
const tables = []; const tables = [];
const relationships = []; const relationships = [];
ast.forEach((e) => { const parseSingleStatement = (e) => {
if (e.type === "create") { if (e.type === "create") {
if (e.keyword === "table") { if (e.keyword === "table") {
const table = {}; const table = {};
@ -201,7 +201,13 @@ export function fromSQLite(ast, diagramDb = DB.GENERIC) {
if (found !== -1) tables[found].indices.forEach((i, j) => (i.id = j)); if (found !== -1) tables[found].indices.forEach((i, j) => (i.id = j));
} }
} }
}); };
if (Array.isArray(ast)) {
ast.forEach((e) => parseSingleStatement(e));
} else {
parseSingleStatement(ast);
}
relationships.forEach((r, i) => (r.id = i)); relationships.forEach((r, i) => (r.id = i));