mirrors/dbgate
1
0
Fork 0
mirror of https://github.com/dbgate/dbgate synced 2024-11-07 20:26:23 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refactor

Browse Source
This commit is contained in:
Jan Prochazka 2021-09-09 08:51:38 +02:00
parent eec1840e8a
commit e568adc825
1 changed files with 17 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
packages/tools/src/alterPlan.ts
View File

@ -190,27 +190,28 @@ export class AlterPlan {
} }
} }
_getDependendColumnConstraints(column: ColumnInfo, dependencyDefinition) {
const table = this.db.tables.find(x => x.pureName == column.pureName && x.schemaName == column.schemaName);
const fks = dependencyDefinition?.includes('dependencies')
? table.dependencies.filter(fk => fk.columns.find(col => col.refColumnName == column.columnName))
: [];
const constraints = _.compact([
dependencyDefinition?.includes('primaryKey') ? table.primaryKey : null,
...(dependencyDefinition?.includes('foreignKeys') ? table.foreignKeys : []),
...(dependencyDefinition?.includes('indexes') ? table.indexes : []),
...(dependencyDefinition?.includes('uniques') ? table.uniques : []),
]).filter(cnt => cnt.columns.find(col => col.columnName == column.columnName));
return [...fks, ...constraints];
}
_addLogicalDependencies(): AlterOperation[] { _addLogicalDependencies(): AlterOperation[] {
const lists = this.operations.map(op => { const lists = this.operations.map(op => {
if (op.operationType == 'dropColumn') { if (op.operationType == 'dropColumn') {
const table = this.db.tables.find( const constraints = this._getDependendColumnConstraints(op.oldObject, this.dialect.dropColumnDependencies);
x => x.pureName == op.oldObject.pureName && x.schemaName == op.oldObject.schemaName
);
const deletedFks = this.dialect.dropColumnDependencies?.includes('dependencies')
? table.dependencies.filter(fk => fk.columns.find(col => col.refColumnName == op.oldObject.columnName))
: [];
const deletedConstraints = _.compact([
this.dialect.dropColumnDependencies?.includes('primaryKey') ? table.primaryKey : null,
...(this.dialect.dropColumnDependencies?.includes('foreignKeys') ? table.foreignKeys : []),
...(this.dialect.dropColumnDependencies?.includes('indexes') ? table.indexes : []),
...(this.dialect.dropColumnDependencies?.includes('uniques') ? table.uniques : []),
]).filter(cnt => cnt.columns.find(col => col.columnName == op.oldObject.columnName));
// console.log('deletedConstraints', deletedConstraints);
const res: AlterOperation[] = [ const res: AlterOperation[] = [
...[...deletedFks, ...deletedConstraints].map(oldObject => { ...constraints.map(oldObject => {
const opRes: AlterOperation = { const opRes: AlterOperation = {
operationType: 'dropConstraint', operationType: 'dropConstraint',
oldObject, oldObject,